@@ -525,6 +525,22 @@ def conv(v):
525
525
expected = TimedeltaIndex ([ np .timedelta64 (1 ,'D' ) ]* 5 )
526
526
tm .assert_index_equal (result , expected )
527
527
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
+
528
544
def testit (unit , transform ):
529
545
530
546
# array
@@ -852,6 +868,13 @@ def test_constructor(self):
852
868
pd .offsets .Second (3 )])
853
869
tm .assert_index_equal (result ,expected )
854
870
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
+
855
878
def test_constructor_coverage (self ):
856
879
rng = timedelta_range ('1 days' , periods = 10.5 )
857
880
exp = timedelta_range ('1 days' , periods = 10 )
0 commit comments