asi8

Author: cmp0xff

pandas-stubs/_libs/interval.pyi

@@ -38,6 +38,7 @@ class _LengthDescriptor:
     def __get__(
         self, instance: Interval[_OrderableTimesT], owner: Any
     ) -> Timedelta: ...
+    @overload
     def __get__(self, instance: IntervalMixin, owner: Any) -> np_ndarray: ...
 
 @type_check_only
@@ -48,6 +49,7 @@ class _MidDescriptor:
     def __get__(
         self, instance: Interval[_OrderableTimesT], owner: Any
     ) -> _OrderableTimesT: ...
+    @overload
     def __get__(self, instance: IntervalMixin, owner: Any) -> np_ndarray: ...
 
 class IntervalMixin:

pandas-stubs/_typing.pyi

@@ -858,6 +858,8 @@ ShapeT = TypeVar("ShapeT", bound=tuple[int, ...], default=tuple[Any, ...])
 np_ndarray: TypeAlias = np.ndarray[ShapeT, np.dtype[GenericT]]
 # Numpy arrays with known shape (Do not use as argument types, only as return types)
 np_1darray: TypeAlias = np.ndarray[tuple[int], np.dtype[GenericT]]
+np_1darray_int64: TypeAlias = np_1darray[np.int64]
+
 np_2darray: TypeAlias = np.ndarray[tuple[int, int], np.dtype[GenericT]]
 
 DtypeNp = TypeVar("DtypeNp", bound=np.dtype[np.generic])

pandas-stubs/core/arrays/datetimelike.pyi

@@ -25,7 +25,7 @@ from pandas._typing import (
     TimeNonexistent,
     TimeUnit,
     np_1darray,
-    np_ndarray_int64,
+    np_1darray_int64,
 )
 
 DTScalarOrNaT: TypeAlias = DatetimeLikeScalar | NaTType
@@ -65,7 +65,7 @@ class DatetimeLikeArrayMixin(ExtensionOpsMixin, ExtensionArray):
     def ravel(self, *args: Any, **kwargs: Any): ...
     def __iter__(self): ...
     @property
-    def asi8(self) -> np_ndarray_int64: ...
+    def asi8(self) -> np_1darray_int64: ...
     @property
     def nbytes(self): ...
     def __array__(

pandas-stubs/core/construction.pyi

@@ -1,18 +1,42 @@
 from collections.abc import Sequence
+from datetime import (
+    datetime,
+    timedelta,
+)
 from typing import overload
 
 import numpy as np
 from pandas.core.arrays.base import ExtensionArray
+from pandas.core.arrays.datetimes import DatetimeArray
 from pandas.core.arrays.integer import IntegerArray
+from pandas.core.arrays.timedeltas import TimedeltaArray
 
 from pandas._libs.missing import NAType
+from pandas._typing import (
+    IntDtypeArg,
+    TimedeltaDtypeArg,
+    TimestampDtypeArg,
+    UIntDtypeArg,
+)
 
 from pandas.core.dtypes.dtypes import ExtensionDtype
 
+@overload
+def array(  # type: ignore[overload-overlap] # pyright: ignore[reportOverlappingOverload]
+    data: Sequence[timedelta | NAType],
+    dtype: TimedeltaDtypeArg | None = None,
+    copy: bool = True,
+) -> TimedeltaArray: ...
+@overload
+def array(  # type: ignore[overload-overlap]
+    data: Sequence[datetime | NAType],
+    dtype: TimestampDtypeArg | None = None,
+    copy: bool = True,
+) -> DatetimeArray: ...
 @overload
 def array(
-    data: Sequence[int] | Sequence[int | NAType],
-    dtype: str | np.dtype | ExtensionDtype | None = None,
+    data: Sequence[int | NAType],
+    dtype: IntDtypeArg | UIntDtypeArg | None = None,
     copy: bool = True,
 ) -> IntegerArray: ...
 @overload

pandas-stubs/core/indexes/datetimelike.pyi

@@ -13,6 +13,7 @@ from pandas._typing import (
     AxisIndex,
     GenericT_co,
     TimeUnit,
+    np_1darray_int64,
     np_ndarray_complex,
 )
 
@@ -62,3 +63,5 @@ class DatetimeTimedeltaMixin(DatetimeIndexOpsMixin[S1, GenericT_co]):
     @property
     def unit(self) -> TimeUnit: ...
     def as_unit(self, unit: TimeUnit) -> Self: ...
+    @property
+    def asi8(self) -> np_1darray_int64: ...

tests/__init__.py

@@ -45,6 +45,7 @@
         UIntDtypeArg as UIntDtypeArg,
         VoidDtypeArg as VoidDtypeArg,
         np_1darray as np_1darray,
+        np_1darray_int64 as np_1darray_int64,
         np_2darray as np_2darray,
         np_ndarray as np_ndarray,
         np_ndarray_bool as np_ndarray_bool,
@@ -55,10 +56,12 @@
     _S = TypeVar("_S", bound=tuple[int, ...])
     # Separately define here so pytest works
     np_1darray: TypeAlias = np.ndarray[tuple[int], np.dtype[_G]]
+    np_1darray_int64: TypeAlias = np_1darray[np.int64]
     np_2darray: TypeAlias = np.ndarray[tuple[int, int], np.dtype[_G]]
     np_ndarray: TypeAlias = np.ndarray[_S, np.dtype[_G]]
     np_ndarray_bool: TypeAlias = npt.NDArray[np.bool_]
     np_ndarray_int: TypeAlias = npt.NDArray[np.signedinteger]
+    np_ndarray_int64: TypeAlias = npt.NDArray[np.int64]
 
 TYPE_CHECKING_INVALID_USAGE: Final = TYPE_CHECKING
 WINDOWS = os.name == "nt" or "cygwin" in platform.system().lower()

tests/arrays/test_datetime_arrays.py

@@ -0,0 +1,24 @@
+from __future__ import annotations
+
+from datetime import datetime
+
+import numpy as np
+import pandas as pd
+from pandas.core.arrays.datetimes import DatetimeArray
+import pytest
+from typing_extensions import assert_type
+
+from tests import (
+    check,
+    np_1darray_int64,
+)
+
+
+@pytest.fixture
+def arr() -> DatetimeArray:
+    a = pd.array([datetime(2025, 11, 7), datetime(2025, 11, 8)])
+    return check(assert_type(a, DatetimeArray), DatetimeArray, pd.Timestamp)
+
+
+def test_timedelta_index_properties(arr: DatetimeArray) -> None:
+    check(assert_type(arr.asi8, np_1darray_int64), np_1darray_int64, np.integer)

tests/arrays/test_timedelta_arrays.py

@@ -0,0 +1,24 @@
+from __future__ import annotations
+
+from datetime import timedelta
+
+import numpy as np
+import pandas as pd
+from pandas.core.arrays.timedeltas import TimedeltaArray
+import pytest
+from typing_extensions import assert_type
+
+from tests import (
+    check,
+    np_1darray_int64,
+)
+
+
+@pytest.fixture
+def arr() -> TimedeltaArray:
+    a = pd.array([timedelta(seconds=1), timedelta(seconds=2)])
+    return check(assert_type(a, TimedeltaArray), TimedeltaArray, pd.Timedelta)
+
+
+def test_timedelta_index_properties(arr: TimedeltaArray) -> None:
+    check(assert_type(arr.asi8, np_1darray_int64), np_1darray_int64, np.integer)

tests/indexes/test_datetime_index.py

@@ -11,6 +11,7 @@
 from tests import (
     check,
     np_1darray,
+    np_1darray_int64,
 )
 
 
@@ -130,3 +131,8 @@ def test_datetimeindex_snap() -> None:
         ),
         pd.DatetimeIndex,
     )
+
+
+def test_datetimeindex_properties() -> None:
+    dti = pd.date_range("2023-01-01", "2023-02-01")
+    check(assert_type(dti.asi8, np_1darray_int64), np_1darray_int64, np.integer)

tests/indexes/test_timedelta_indexes.py

@@ -0,0 +1,23 @@
+from __future__ import annotations
+
+import numpy as np
+import pandas as pd
+import pytest
+from typing_extensions import (
+    assert_type,
+)
+
+from tests import (
+    check,
+    np_1darray_int64,
+)
+
+
+@pytest.fixture
+def tdi() -> pd.TimedeltaIndex:
+    idx = pd.timedelta_range("1 days", periods=3, freq="D")
+    return check(assert_type(idx, pd.TimedeltaIndex), pd.TimedeltaIndex, pd.Timedelta)
+
+
+def test_timedelta_index_properties(tdi: pd.TimedeltaIndex) -> None:
+    check(assert_type(tdi.asi8, np_1darray_int64), np_1darray_int64, np.integer)