1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
| from pandas import (
| TimedeltaIndex,
| timedelta_range,
| )
| import pandas._testing as tm
|
|
| class TestTimedeltaIndexDelete:
| def test_delete(self):
| idx = timedelta_range(start="1 Days", periods=5, freq="D", name="idx")
|
| # preserve freq
| expected_0 = timedelta_range(start="2 Days", periods=4, freq="D", name="idx")
| expected_4 = timedelta_range(start="1 Days", periods=4, freq="D", name="idx")
|
| # reset freq to None
| expected_1 = TimedeltaIndex(
| ["1 day", "3 day", "4 day", "5 day"], freq=None, name="idx"
| )
|
| cases = {
| 0: expected_0,
| -5: expected_0,
| -1: expected_4,
| 4: expected_4,
| 1: expected_1,
| }
| for n, expected in cases.items():
| result = idx.delete(n)
| tm.assert_index_equal(result, expected)
| assert result.name == expected.name
| assert result.freq == expected.freq
|
| with tm.external_error_raised((IndexError, ValueError)):
| # either depending on numpy version
| idx.delete(5)
|
| def test_delete_slice(self):
| idx = timedelta_range(start="1 days", periods=10, freq="D", name="idx")
|
| # preserve freq
| expected_0_2 = timedelta_range(start="4 days", periods=7, freq="D", name="idx")
| expected_7_9 = timedelta_range(start="1 days", periods=7, freq="D", name="idx")
|
| # reset freq to None
| expected_3_5 = TimedeltaIndex(
| ["1 d", "2 d", "3 d", "7 d", "8 d", "9 d", "10d"], freq=None, name="idx"
| )
|
| cases = {
| (0, 1, 2): expected_0_2,
| (7, 8, 9): expected_7_9,
| (3, 4, 5): expected_3_5,
| }
| for n, expected in cases.items():
| result = idx.delete(n)
| tm.assert_index_equal(result, expected)
| assert result.name == expected.name
| assert result.freq == expected.freq
|
| result = idx.delete(slice(n[0], n[-1] + 1))
| tm.assert_index_equal(result, expected)
| assert result.name == expected.name
| assert result.freq == expected.freq
|
| def test_delete_doesnt_infer_freq(self):
| # GH#30655 behavior matches DatetimeIndex
|
| tdi = TimedeltaIndex(["1 Day", "2 Days", None, "3 Days", "4 Days"])
| result = tdi.delete(2)
| assert result.freq is None
|
|
