Skip to content

Commit 4fbe729

Browse files
committed
Move nullable int64 dtypes to separate test and isort
1 parent 99c6b6d commit 4fbe729

File tree

2 files changed

+13
-13
lines changed

2 files changed

+13
-13
lines changed

pandas/core/arrays/timedeltas.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@
2929

3030
from pandas.core import nanops
3131
from pandas.core.algorithms import checked_add_with_arr
32-
from pandas.core.arrays import datetimelike as dtl, IntegerArray
32+
from pandas.core.arrays import IntegerArray, datetimelike as dtl
3333
from pandas.core.arrays._ranges import generate_regular_range
3434
import pandas.core.common as com
3535
from pandas.core.construction import extract_array

pandas/tests/tools/test_to_timedelta.py

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)