GH1264 Version 2.0 cleanup

Author: loicdiridollou

pandas-stubs/core/indexes/base.pyi

@@ -347,11 +347,16 @@ class Index(IndexOpsMixin[S1]):
     def __neg__(self) -> Self: ...
     def __nonzero__(self) -> None: ...
     __bool__ = ...
-    def union(self, other: list[HashableT] | Index, sort=...) -> Index: ...
-    def intersection(self, other: list[S1] | Self, sort: bool = ...) -> Self: ...
+    def union(
+        self, other: list[HashableT] | Index, sort: bool | None = ...
+    ) -> Index: ...
+    def intersection(self, other: list[S1] | Self, sort: bool | None = ...) -> Self: ...
     def difference(self, other: list | Index, sort: bool | None = None) -> Self: ...
     def symmetric_difference(
-        self, other: list[S1] | Self, result_name: Hashable = ..., sort=...
+        self,
+        other: list[S1] | Self,
+        result_name: Hashable = ...,
+        sort: bool | None = ...,
     ) -> Self: ...
     def get_loc(
         self,
@@ -472,5 +477,6 @@ class Index(IndexOpsMixin[S1]):
             | Sequence[float]
         ),
     ) -> Self: ...
+    def infer_objects(self, copy: bool = ...) -> Self: ...
 
 UnknownIndex: TypeAlias = Index[Any]

pandas-stubs/core/indexes/multi.pyi

@@ -155,7 +155,7 @@ class MultiIndex(Index):
     def equal_levels(self, other): ...
     def union(self, other, sort=...): ...  # pyrefly: ignore
     def intersection(  # pyright: ignore[reportIncompatibleMethodOverride]
-        self, other: list | Self, sort: bool = ...
+        self, other: list | Self, sort: bool | None = ...
     ): ...
     def difference(self, other, sort=...): ...
     def astype(self, dtype: DtypeArg, copy: bool = ...) -> Self: ...

pandas-stubs/core/series.pyi

@@ -2324,6 +2324,13 @@ class TimedeltaSeries(Series[Timedelta]):
         *args: Any,
         **kwargs: Any,
     ) -> TimedeltaSeries: ...
+    def cumprod(
+        self,
+        axis: AxisIndex | None = ...,
+        skipna: _bool = ...,
+        *args: Any,
+        **kwargs: Any,
+    ) -> Never: ...
 
 class PeriodSeries(Series[Period]):
     @property

tests/test_indexes.py

@@ -172,6 +172,10 @@ def test_difference_none() -> None:
     # GH 253
     check(assert_type(ind.difference([1]), "pd.Index[int]"), pd.Index)
 
+    # check with sort parameter
+    check(assert_type(ind.difference([1, None], sort=False), "pd.Index[int]"), pd.Index)
+    check(assert_type(ind.difference([1], sort=True), "pd.Index[int]"), pd.Index)
+
 
 def test_str_split() -> None:
     # GH 194
@@ -312,6 +316,20 @@ def test_range_index_union():
         ),
         pd.Index,
     )
+    check(
+        assert_type(
+            pd.RangeIndex(0, 10).union(["a", "b", "c"], sort=True),
+            Union[pd.Index, "pd.Index[int]", pd.RangeIndex],
+        ),
+        pd.Index,
+    )
+    check(
+        assert_type(
+            pd.RangeIndex(0, 10).union(["a", "b", "c"], sort=False),
+            Union[pd.Index, "pd.Index[int]", pd.RangeIndex],
+        ),
+        pd.Index,
+    )
 
 
 def test_range_index_start_stop_step():
@@ -1361,3 +1379,10 @@ def test_index_dict() -> None:
         ),
         pd.TimedeltaIndex,
     )
+
+
+def test_index_infer_objects() -> None:
+    """Test infer_objects method on Index."""
+    df = pd.DataFrame({"A": ["a", 1, 2, 3]})
+    idx = df.set_index("A").index[1:]
+    check(assert_type(idx.infer_objects(), pd.Index), pd.Index)

tests/test_series.py

@@ -3941,3 +3941,8 @@ def test_series_index_type() -> None:
 
     if TYPE_CHECKING_INVALID_USAGE:
         t = pd.Series([1, 2], index="ab")  # type: ignore[call-overload] # pyright: ignore[reportCallIssue, reportArgumentType]
+
+
+def test_timedelta_index_cumsum() -> None:
+    if TYPE_CHECKING_INVALID_USAGE:
+        assert_type(pd.Series([pd.Timedelta(0), pd.Timedelta(1)]).cumprod(), Never)