test_timedelta_array

Author: cmp0xff

pandas-stubs/core/algorithms.pyi

@@ -39,18 +39,19 @@ def unique(values: IntervalIndex[IntervalT]) -> IntervalIndex[IntervalT]: ...
 @overload
 def unique(values: PeriodIndex) -> PeriodIndex: ...
 @overload
-# switch to DatetimeIndex after Pandas 3.0
+# TODO: switch to DatetimeIndex after Pandas 3.0 pandas-dev/pandas#57064
 def unique(values: DatetimeIndex) -> np_1darray_dt | DatetimeIndex: ...
 @overload
-# switch to TimedeltaIndex after Pandas 3.0
+# TODO: switch to TimedeltaIndex after Pandas 3.0 pandas-dev/pandas#57064
 def unique(values: TimedeltaIndex) -> np_1darray_td: ...
 @overload
 # switch to Index[int] after Pandas 3.0
 def unique(values: RangeIndex) -> np_1darray_int64: ...
 @overload
 def unique(values: MultiIndex) -> np_ndarray: ...
 @overload
-def unique(values: Index) -> np_1darray | Index: ...  # switch to Index after Pandas 3.0
+# TODO: switch to Index after Pandas 3.0 pandas-dev/pandas#57064
+def unique(values: Index) -> np_1darray | Index: ...
 @overload
 def unique(values: Categorical) -> Categorical: ...
 

pandas-stubs/core/construction.pyi

@@ -98,7 +98,7 @@ def array(  # type: ignore[overload-overlap]
 ) -> FloatingArray: ...
 @overload
 def array(  # type: ignore[overload-overlap]
-    data: (
+    data: (  # TODO: merge the two Sequence's after 3.0 pandas-dev/pandas#57064
         Sequence[datetime | NaTType | None]
         | Sequence[np.datetime64 | NaTType | None]
         | np_ndarray_dt

tests/arrays/test_timedelta_array.py

@@ -0,0 +1,33 @@
+from datetime import timedelta
+
+import numpy as np
+import pandas as pd
+from pandas.core.arrays.timedeltas import TimedeltaArray
+from typing_extensions import assert_type
+
+from tests import check
+
+
+def test_constructor() -> None:
+    td = timedelta(2025, 11, 10)
+    np_dt = np.timedelta64(td)
+    check(assert_type(pd.array([td]), TimedeltaArray), TimedeltaArray)
+    check(
+        assert_type(pd.array([td, pd.Timedelta(td), np_dt]), TimedeltaArray),
+        TimedeltaArray,
+    )
+    check(assert_type(pd.array([td, None]), TimedeltaArray), TimedeltaArray)
+    check(assert_type(pd.array([td, pd.NaT, None]), TimedeltaArray), TimedeltaArray)
+
+    check(
+        assert_type(  # type: ignore[assert-type]  # I do not understand
+            pd.array(np.array([td], np.timedelta64)), TimedeltaArray
+        ),
+        TimedeltaArray,
+    )
+
+    check(assert_type(pd.array(pd.array([td])), TimedeltaArray), TimedeltaArray)
+
+    check(assert_type(pd.array(pd.Index([td])), TimedeltaArray), TimedeltaArray)
+
+    check(assert_type(pd.array(pd.Series([td])), TimedeltaArray), TimedeltaArray)