Add check assert_type to tests for frame.py and add __new__ method for DatetimeIndex

pandas-stubs/core/frame.pyi

@@ -45,6 +45,8 @@ from pandas.core.indexes.multi import MultiIndex
 from pandas.core.indexes.period import PeriodIndex
 from pandas.core.indexes.timedeltas import TimedeltaIndex
 from pandas.core.indexing import (
+    _AtIndexer,
+    _iAtIndexer,
     _iLocIndexer,
     _IndexSliceTuple,
     _LocIndexer,
@@ -285,6 +287,44 @@ class _LocIndexerFrame(_LocIndexer, Generic[_T]):
         value: Scalar | NAType | NaTType | ArrayLike | Series | list | dict | None,
     ) -> None: ...
 
+class _iAtIndexerFrame(_iAtIndexer, Generic[_T]):
+    def __getitem__(self, idx: tuple[int, int]) -> Scalar: ...
+    def __setitem__(
+        self,
+        idx: tuple[int, int],
+        value: Scalar | NAType | NaTType | None,
+    ) -> None: ...
+
+class _AtIndexerFrame(_AtIndexer, Generic[_T]):
+    def __getitem__(
+        self,
+        idx: tuple[
+            int
+            | StrLike
+            | Timestamp
+            | tuple[Scalar, ...]
+            | Callable[[DataFrame], ScalarT],
+            int | StrLike | tuple[Scalar, ...],
+        ],
+    ) -> Scalar: ...
+    def __setitem__(
+        self,
+        idx: (
+            MaskType | StrLike | _IndexSliceTuple | list[ScalarT] | IndexingInt | slice
+        ),
+        value: (
+            Scalar
+            | NAType
+            | NaTType
+            | ArrayLike
+            | Series
+            | DataFrame
+            | list
+            | Mapping[Hashable, Scalar | NAType | NaTType]
+            | None
+        ),
+    ) -> None: ...
+
 # With mypy 1.14.1 and python 3.12, the second overload needs a type-ignore statement
 if sys.version_info >= (3, 12):
     class _GetItemHack:
@@ -1591,13 +1631,13 @@ class DataFrame(NDFrame, OpsMixin, _GetItemHack):
         axis: Axis = 0,
         skipna: _bool = True,
         numeric_only: _bool = False,
-    ) -> Series: ...
+    ) -> Series[int]: ...
     def idxmin(
         self,
         axis: Axis = 0,
         skipna: _bool = True,
         numeric_only: _bool = False,
-    ) -> Series: ...
+    ) -> Series[int]: ...
     def mode(
         self,
         axis: Axis = 0,
@@ -1683,7 +1723,9 @@ class DataFrame(NDFrame, OpsMixin, _GetItemHack):
     def __iter__(self) -> Iterator[Hashable]: ...
     # properties
     @property
-    def at(self): ...  # Not sure what to do with this yet; look at source
+    def at(
+        self,
+    ) -> _AtIndexerFrame: ...  # Not sure what to do with this yet; look at source
     @property
     def columns(self) -> Index[str]: ...
     @columns.setter  # setter needs to be right next to getter; otherwise mypy complains
@@ -1695,7 +1737,9 @@ class DataFrame(NDFrame, OpsMixin, _GetItemHack):
     @property
     def empty(self) -> _bool: ...
     @property
-    def iat(self): ...  # Not sure what to do with this yet; look at source
+    def iat(
+        self,
+    ) -> _iAtIndexerFrame: ...  # Not sure what to do with this yet; look at source
     @property
     def iloc(self) -> _iLocIndexerFrame[Self]: ...
     @property

pandas-stubs/core/indexes/datetimes.pyi

@@ -23,7 +23,6 @@ from pandas import (
 from pandas.core.indexes.accessors import DatetimeIndexProperties
 from pandas.core.indexes.datetimelike import DatetimeTimedeltaMixin
 from pandas.core.series import (
-    Series,
     TimedeltaSeries,
     TimestampSeries,
 )
@@ -45,22 +44,9 @@ from pandas.core.dtypes.dtypes import DatetimeTZDtype
 from pandas.tseries.offsets import BaseOffset
 
 class DatetimeIndex(DatetimeTimedeltaMixin[Timestamp], DatetimeIndexProperties):
-    def __init__(
-        self,
-        data: AxesData,
-        freq: Frequency = ...,
-        tz: TimeZones = ...,
-        ambiguous: str = ...,
-        dayfirst: bool = ...,
-        yearfirst: bool = ...,
-        dtype: Dtype = ...,
-        copy: bool = ...,
-        name: Hashable = ...,
-    ) -> None: ...
-    @classmethod
     def __new__(
         cls,
-        data: AxesData | Series,
+        data: AxesData,
         freq: Frequency = ...,
         tz: TimeZones = ...,
         ambiguous: str = ...,