@@ -311,29 +311,28 @@ def test_nat_converters(self):
311
311
(['ns' , 'nanoseconds' , 'nanosecond' , 'nano' ,
312
312
'nanos' , 'n' , 'NS' , 'Nanoseconds' ,
313
313
'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 ):
315
316
# validate all units, GH 6855, GH 21762
316
317
for unit in units :
317
318
# array-likes
318
319
expected = TimedeltaIndex ([np .timedelta64 (i , np_unit )
319
320
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 )
325
325
326
326
if unit == 'M' :
327
327
# M is treated as minutes in string repr
328
328
expected = TimedeltaIndex ([np .timedelta64 (i , 'm' )
329
329
for i in np .arange (5 ).tolist ()])
330
330
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 )
337
336
338
337
# scalar
339
338
expected = Timedelta (np .timedelta64 (2 , np_unit ).astype (
0 commit comments