@@ -24,18 +24,6 @@ def test_to_timedelta(self):
2424 result = to_timedelta (Series (["1d" , "1days 00:00:01" ]))
2525 tm .assert_series_equal (result , expected )
2626
27- # IntegerArray Series
28- expected = Series ([timedelta (days = 1 ), timedelta (days = 2 )])
29- result = to_timedelta (Series ([1 , 2 ], dtype = "Int64" ), unit = "days" )
30-
31- tm .assert_series_equal (result , expected )
32-
33- # IntegerArray Series with nulls
34- expected = Series ([timedelta (days = 1 ), None ])
35- result = to_timedelta (Series ([1 , None ], dtype = "Int64" ), unit = "days" )
36-
37- tm .assert_series_equal (result , expected )
38-
3927 # with units
4028 result = TimedeltaIndex (
4129 [np .timedelta64 (0 , "ns" ), np .timedelta64 (10 , "s" ).astype ("m8[ns]" )]
@@ -178,3 +166,15 @@ def test_to_timedelta_ignore_strings_unit(self):
178166 arr = np .array ([1 , 2 , "error" ], dtype = object )
179167 result = pd .to_timedelta (arr , unit = "ns" , errors = "ignore" )
180168 tm .assert_numpy_array_equal (result , arr )
169+
170+ def test_to_timedelta_nullable_int64_dtype (self ):
171+ expected = Series ([timedelta (days = 1 ), timedelta (days = 2 )])
172+ result = to_timedelta (Series ([1 , 2 ], dtype = "Int64" ), unit = "days" )
173+
174+ tm .assert_series_equal (result , expected )
175+
176+ # IntegerArray Series with nulls
177+ expected = Series ([timedelta (days = 1 ), None ])
178+ result = to_timedelta (Series ([1 , None ], dtype = "Int64" ), unit = "days" )
179+
180+ tm .assert_series_equal (result , expected )
0 commit comments