@@ -525,6 +525,22 @@ def conv(v):
525525 expected = TimedeltaIndex ([ np .timedelta64 (1 ,'D' ) ]* 5 )
526526 tm .assert_index_equal (result , expected )
527527
528+ # Test with lists as input when box=false
529+ expected = np .array (np .arange (3 )* 1000000000 , dtype = 'timedelta64[ns]' )
530+ result = to_timedelta (range (3 ), unit = 's' , box = False )
531+ tm .assert_numpy_array_equal (expected , result )
532+
533+ result = to_timedelta (np .arange (3 ), unit = 's' , box = False )
534+ tm .assert_numpy_array_equal (expected , result )
535+
536+ result = to_timedelta ([0 , 1 , 2 ], unit = 's' , box = False )
537+ tm .assert_numpy_array_equal (expected , result )
538+
539+ # Tests with fractional seconds as input:
540+ expected = np .array ([0 , 500000000 , 800000000 , 1200000000 ], dtype = 'timedelta64[ns]' )
541+ result = to_timedelta ([0. , 0.5 , 0.8 , 1.2 ], unit = 's' , box = False )
542+ tm .assert_numpy_array_equal (expected , result )
543+
528544 def testit (unit , transform ):
529545
530546 # array
@@ -852,6 +868,13 @@ def test_constructor(self):
852868 pd .offsets .Second (3 )])
853869 tm .assert_index_equal (result ,expected )
854870
871+ expected = TimedeltaIndex (['0 days 00:00:00' , '0 days 00:00:01' , '0 days 00:00:02' ])
872+ tm .assert_index_equal (TimedeltaIndex (range (3 ), unit = 's' ), expected )
873+ expected = TimedeltaIndex (['0 days 00:00:00' , '0 days 00:00:05' , '0 days 00:00:09' ])
874+ tm .assert_index_equal (TimedeltaIndex ([0 , 5 , 9 ], unit = 's' ), expected )
875+ expected = TimedeltaIndex (['0 days 00:00:00.400' , '0 days 00:00:00.450' , '0 days 00:00:01.200' ])
876+ tm .assert_index_equal (TimedeltaIndex ([400 , 450 , 1200 ], unit = 'ms' ), expected )
877+
855878 def test_constructor_coverage (self ):
856879 rng = timedelta_range ('1 days' , periods = 10.5 )
857880 exp = timedelta_range ('1 days' , periods = 10 )
0 commit comments