Skip to content

Commit 04f3ceb

Browse files
committed
fix tests
1 parent 789e9aa commit 04f3ceb

File tree

1 file changed

+11
-12
lines changed

1 file changed

+11
-12
lines changed

pandas/tests/scalar/timedelta/test_timedelta.py

+11-12
Original file line numberDiff line numberDiff line change
@@ -311,29 +311,28 @@ def test_nat_converters(self):
311311
(['ns', 'nanoseconds', 'nanosecond', 'nano',
312312
'nanos', 'n', 'NS', 'Nanoseconds',
313313
'Nanosecond', 'Nano', 'Nanos', 'N'], 'ns')])
314-
def test_unit_parser(self, units, np_unit):
314+
@pytest.mark.parametrize('wrapper', [np.array, list, pd.Index])
315+
def test_unit_parser(self, units, np_unit, wrapper):
315316
# validate all units, GH 6855, GH 21762
316317
for unit in units:
317318
# array-likes
318319
expected = TimedeltaIndex([np.timedelta64(i, np_unit)
319320
for i in np.arange(5).tolist()])
320-
for wrapper in [np.array, list, pd.Index]:
321-
result = to_timedelta(wrapper(range(5)), unit=unit)
322-
tm.assert_index_equal(result, expected)
323-
result = TimedeltaIndex(wrapper(range(5)), unit=unit)
324-
tm.assert_index_equal(result, expected)
321+
result = to_timedelta(wrapper(range(5)), unit=unit)
322+
tm.assert_index_equal(result, expected)
323+
result = TimedeltaIndex(wrapper(range(5)), unit=unit)
324+
tm.assert_index_equal(result, expected)
325325

326326
if unit == 'M':
327327
# M is treated as minutes in string repr
328328
expected = TimedeltaIndex([np.timedelta64(i, 'm')
329329
for i in np.arange(5).tolist()])
330330

331-
for wrapper in [np.array, list, pd.Index]:
332-
str_repr = ['{}{}'.format(x, unit) for x in np.arange(5)]
333-
result = to_timedelta(wrapper(str_repr))
334-
tm.assert_index_equal(result, expected)
335-
result = TimedeltaIndex(wrapper(str_repr))
336-
tm.assert_index_equal(result, expected)
331+
str_repr = ['{}{}'.format(x, unit) for x in np.arange(5)]
332+
result = to_timedelta(wrapper(str_repr))
333+
tm.assert_index_equal(result, expected)
334+
result = TimedeltaIndex(wrapper(str_repr))
335+
tm.assert_index_equal(result, expected)
337336

338337
# scalar
339338
expected = Timedelta(np.timedelta64(2, np_unit).astype(

0 commit comments

Comments
 (0)