72 lines
2.3 KiB
Python
72 lines
2.3 KiB
Python
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
|