Skip to content

Commit fb11c66

Browse files
committed
update test
1 parent d9fa426 commit fb11c66

File tree

1 file changed

+26
-8
lines changed

1 file changed

+26
-8
lines changed

pandas/tests/frame/indexing/test_getitem.py

Lines changed: 26 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -176,7 +176,10 @@ 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, freq=DateOffset(days=1)))
179+
df = DataFrame(
180+
list(range(10)),
181+
index=date_range("01-01-2022", periods=10, freq=DateOffset(days=1)),
182+
)
180183
result = df.loc["2022-01-01":"2022-01-03"]
181184
expected = DataFrame(
182185
[0, 1, 2],
@@ -188,7 +191,12 @@ def test_getitem_iloc_dateoffset_days(self):
188191
)
189192
tm.assert_frame_equal(result, expected)
190193

191-
df = DataFrame(list(range(10)), index=date_range("01-01-2022", periods=10, freq=DateOffset(days=1, hours=2)))
194+
df = DataFrame(
195+
list(range(10)),
196+
index=date_range(
197+
"01-01-2022", periods=10, freq=DateOffset(days=1, hours=2)
198+
),
199+
)
192200
result = df.loc["2022-01-01":"2022-01-03"]
193201
expected = DataFrame(
194202
[0, 1, 2],
@@ -200,16 +208,26 @@ def test_getitem_iloc_dateoffset_days(self):
200208
)
201209
tm.assert_frame_equal(result, expected)
202210

203-
204-
df = DataFrame(list(range(10)), index=date_range("01-01-2022", periods=10, freq=DateOffset(minutes=3)))
211+
df = DataFrame(
212+
list(range(10)),
213+
index=date_range("01-01-2022", periods=10, freq=DateOffset(minutes=3)),
214+
)
205215
result = df.loc["2022-01-01":"2022-01-03"]
206216
expected = DataFrame(
207217
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9],
208218
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"],
219+
[
220+
"2022-01-01 00:00:00",
221+
"2022-01-01 00:03:00",
222+
"2022-01-01 00:06:00",
223+
"2022-01-01 00:09:00",
224+
"2022-01-01 00:12:00",
225+
"2022-01-01 00:15:00",
226+
"2022-01-01 00:18:00",
227+
"2022-01-01 00:21:00",
228+
"2022-01-01 00:24:00",
229+
"2022-01-01 00:27:00",
230+
],
213231
dtype="datetime64[ns]",
214232
freq=DateOffset(minutes=3),
215233
),

0 commit comments

Comments
 (0)