WIP

Author: twoertwein

pandas-stubs/core/arrays/base.pyi

@@ -10,11 +10,11 @@ from pandas._typing import (
     npt,
 )
 
-from pandas.core.dtypes.dtypes import ExtensionDtype as ExtensionDtype
+from pandas.core.dtypes.dtypes import ExtensionDtype
 
 class ExtensionArray:
     def __getitem__(self, item) -> Any: ...
-    def __setitem__(self, key: int | slice | np.ndarray, value) -> None: ...
+    def __setitem__(self, key: int | slice | np.ndarray, value: Self) -> None: ...
     def __len__(self) -> int: ...
     def __iter__(self): ...
     def __contains__(self, item: object) -> bool | np.bool_: ...

pandas-stubs/core/arrays/interval.pyi

@@ -12,13 +12,10 @@ from pandas import (
     Index,
     Series,
 )
-from pandas.core.arrays.base import ExtensionArray as ExtensionArray
+from pandas.core.arrays.base import ExtensionArray
 from typing_extensions import Self
 
-from pandas._libs.interval import (
-    Interval as Interval,
-    IntervalMixin as IntervalMixin,
-)
+from pandas._libs.interval import Interval, IntervalMixin
 from pandas._typing import (
     Axis,
     IntervalT,
@@ -77,7 +74,6 @@ class IntervalArray(IntervalMixin, ExtensionArray, Generic[IntervalT]):
     def __iter__(self) -> IntervalT: ...
     def __len__(self) -> int: ...
     def __getitem__(self, value: IntervalT): ...
-    def __setitem__(self, key: int, value: IntervalT) -> None: ...
     def __eq__(self, other): ...
     def __ne__(self, other): ...
     def fillna(self, value=..., method=..., limit=...): ...

pandas-stubs/core/arrays/string_.pyi

@@ -48,7 +48,7 @@ class StringArray(PandasArray):
         | npt.NDArray[np.string_]
         | StringArray,
     ) -> None: ...
-    @overload  # type: ignore[override]
+    @overload
     def __getitem__(self, item: int) -> str | NAType: ...
     @overload
     def __getitem__(self, item: _ArrayKey) -> StringArray: ...

tests/test_arrays.py

@@ -439,11 +439,11 @@ def test_interval_array_construction() -> None:
 
 def test_integer_array_attrib_props() -> None:
     ia = IntervalArray([pd.Interval(0, 1), pd.Interval(1, 2)])
-    check(assert_type(ia.left, pd.Index), pd.Int64Index)
-    check(assert_type(ia.right, pd.Index), pd.Int64Index)
+    check(assert_type(ia.left, pd.Index), pd.Index, np.intp)
+    check(assert_type(ia.right, pd.Index), pd.Index, np.intp)
     check(assert_type(ia.closed, str), str)
-    check(assert_type(ia.mid, pd.Index), pd.Float64Index)
-    check(assert_type(ia.length, pd.Index), pd.Int64Index)
+    check(assert_type(ia.mid, pd.Index), pd.Index, float)
+    check(assert_type(ia.length, pd.Index), pd.Index, np.intp)
     check(assert_type(ia.is_empty, npt.NDArray[np.bool_]), np.ndarray, np.bool_)
     check(assert_type(ia.is_non_overlapping_monotonic, bool), bool)
 
@@ -473,9 +473,9 @@ def test_integer_array_attrib_props() -> None:
     check(assert_type(ia.to_tuples(False), npt.NDArray[np.object_]), np.ndarray, tuple)
 
     ia_float = IntervalArray([pd.Interval(0, 1.5), pd.Interval(1, 2)])
-    check(assert_type(ia_float.left, pd.Index), pd.Float64Index)
-    check(assert_type(ia_float.right, pd.Index), pd.Float64Index)
-    check(assert_type(ia_float.length, pd.Index), pd.Float64Index)
+    check(assert_type(ia_float.left, pd.Index), pd.Index, float)
+    check(assert_type(ia_float.right, pd.Index), pd.Index, float)
+    check(assert_type(ia_float.length, pd.Index), pd.Index, float)
 
     ia_ts = IntervalArray(
         [