Skip to content

Commit d9fa426

Browse files
committed
update tests
1 parent 62abb79 commit d9fa426

File tree

1 file changed

+29
-2
lines changed

1 file changed

+29
-2
lines changed

pandas/tests/frame/indexing/test_getitem.py

Lines changed: 29 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -176,8 +176,7 @@ def test_getitem_iloc_two_dimensional_generator(self):
176176

177177
def test_getitem_iloc_dateoffset_days(self):
178178
# GH 46671
179-
df = DataFrame(list(range(10)), index=date_range("01-01-2022", periods=10))
180-
df = df.asfreq(DateOffset(days=1))
179+
df = DataFrame(list(range(10)), index=date_range("01-01-2022", periods=10, freq=DateOffset(days=1)))
181180
result = df.loc["2022-01-01":"2022-01-03"]
182181
expected = DataFrame(
183182
[0, 1, 2],
@@ -189,6 +188,34 @@ def test_getitem_iloc_dateoffset_days(self):
189188
)
190189
tm.assert_frame_equal(result, expected)
191190

191+
df = DataFrame(list(range(10)), index=date_range("01-01-2022", periods=10, freq=DateOffset(days=1, hours=2)))
192+
result = df.loc["2022-01-01":"2022-01-03"]
193+
expected = DataFrame(
194+
[0, 1, 2],
195+
index=DatetimeIndex(
196+
["2022-01-01 00:00:00", "2022-01-02 02:00:00", "2022-01-03 04:00:00"],
197+
dtype="datetime64[ns]",
198+
freq=DateOffset(days=1, hours=2),
199+
),
200+
)
201+
tm.assert_frame_equal(result, expected)
202+
203+
204+
df = DataFrame(list(range(10)), index=date_range("01-01-2022", periods=10, freq=DateOffset(minutes=3)))
205+
result = df.loc["2022-01-01":"2022-01-03"]
206+
expected = DataFrame(
207+
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9],
208+
index=DatetimeIndex(
209+
["2022-01-01 00:00:00", "2022-01-01 00:03:00", "2022-01-01 00:06:00",
210+
"2022-01-01 00:09:00", "2022-01-01 00:12:00", "2022-01-01 00:15:00",
211+
"2022-01-01 00:18:00", "2022-01-01 00:21:00", "2022-01-01 00:24:00",
212+
"2022-01-01 00:27:00"],
213+
dtype="datetime64[ns]",
214+
freq=DateOffset(minutes=3),
215+
),
216+
)
217+
tm.assert_frame_equal(result, expected)
218+
192219

193220
class TestGetitemCallable:
194221
def test_getitem_callable(self, float_frame):

0 commit comments

Comments
 (0)