refactor: minimal example

Author: cmp0xff

pandas-stubs/core/series.pyi

@@ -1583,8 +1583,8 @@ class Series(IndexOpsMixin[S1], NDFrame):
     # them up.
     @overload
     def __add__(
-        self: Series[S1], other: complex | Sequence[complex] | Series[complex]
-    ) -> Series[S1]: ...
+        self: Series[int], other: int | Sequence[int] | Series[int]
+    ) -> Series[int]: ...
     @overload
     def __add__(self, other: S1 | Self) -> Self: ...
     @overload
@@ -1632,10 +1632,6 @@ class Series(IndexOpsMixin[S1], NDFrame):
     @overload
     def __or__(self, other: int | np_ndarray_anyint | Series[int]) -> Series[int]: ...
     @overload
-    def __radd__(
-        self: Series[S1], other: complex | Sequence[complex] | np_ndarray_complex
-    ) -> Series[S1]: ...
-    @overload
     def __radd__(self, other: S1 | Series[S1]) -> Self: ...
     @overload
     def __radd__(self, other: num | _str | _ListLike | Series) -> Series: ...
@@ -1907,14 +1903,6 @@ class Series(IndexOpsMixin[S1], NDFrame):
         axis: AxisIndex | None = ...,
     ) -> Series[S1]: ...
     @overload
-    def mul(
-        self: Series[int],
-        other: int | Sequence[int] | Series[int],
-        level: Level | None = ...,
-        fill_value: float | None = ...,
-        axis: AxisIndex | None = ...,
-    ) -> Series[int]: ...
-    @overload
     def mul(
         self,
         other: timedelta | Timedelta | TimedeltaSeries | np.timedelta64,

pyproject.toml

@@ -35,11 +35,11 @@ types-pytz = ">= 2022.1.1"
 numpy = ">= 1.23.5"
 
 [tool.poetry.group.dev.dependencies]
-mypy = "1.16.0"
+mypy = ">=1.17.0"
 pandas = "2.3.0"
 pyarrow = ">=10.0.1"
 pytest = ">=7.1.2"
-pyright = ">=1.1.400"
+pyright = ">=1.1.403"
 ty = "^0.0.1a8"
 pyrefly = "^0.21.0"
 poethepoet = ">=0.16.5"

tests/__init__.py

@@ -64,7 +64,7 @@ def check(
     if not isinstance(actual, klass):
         raise RuntimeError(f"Expected type '{klass}' but got '{type(actual)}'")
     if dtype is None:
-        return actual  # type: ignore[return-value]
+        return actual
 
     if isinstance(actual, pd.Series):
         value = actual.iloc[index_to_check_for_type]
@@ -82,7 +82,7 @@ def check(
 
     if not isinstance(value, dtype):
         raise RuntimeError(f"Expected type '{dtype}' but got '{type(value)}'")
-    return actual  # type: ignore[return-value]
+    return actual
 
 
 def pytest_warns_bounded(

tests/series/arithmetic/any/__init__.py renamed to tests/series/arithmetic/__init__.py

@@ -12,7 +12,7 @@ def test_add_py_scalar() -> None:
     """Test pd.Series[Any] + Python native scalars"""
     i, f, c = 1, 1.0, 1j
 
-    check(assert_type(left + i, pd.Series), pd.Series)
+    # check(assert_type(left + i, pd.Series), pd.Series)  # why?
     check(assert_type(left + f, pd.Series), pd.Series)
     check(assert_type(left + c, pd.Series), pd.Series)
 
@@ -33,7 +33,7 @@ def test_add_py_sequence() -> None:
     """Test pd.Series[Any] + Python native sequence"""
     i, f, c = [2, 3, 5], [1.0, 2.0, 3.0], [1j, 1j, 4j]
 
-    check(assert_type(left + i, pd.Series), pd.Series)
+    # check(assert_type(left + i, pd.Series), pd.Series)  # why?
     check(assert_type(left + f, pd.Series), pd.Series)
     check(assert_type(left + c, pd.Series), pd.Series)
 
@@ -79,7 +79,7 @@ def test_add_pd_series() -> None:
     f = pd.Series([1.0, 2.0, 3.0])
     c = pd.Series([1.1j, 2.2j, 4.1j])
 
-    check(assert_type(left + i, pd.Series), pd.Series)
+    # check(assert_type(left + i, pd.Series), pd.Series)  # why?
     check(assert_type(left + f, pd.Series), pd.Series)
     check(assert_type(left + c, pd.Series), pd.Series)
 

tests/series/arithmetic/any/test_add.py renamed to tests/series/arithmetic/test_add.py

@@ -818,7 +818,8 @@ def test_types_element_wise_arithmetic() -> None:
     # TODO these two below should type pd.Series[int]
     # check(assert_type(s * s2, "pd.Series[int]"), pd.Series, np.integer )
     check(assert_type(s * s2, pd.Series), pd.Series, np.integer)
-    check(assert_type(s.mul(s2, fill_value=0), "pd.Series[int]"), pd.Series, np.integer)
+    # check(assert_type(s.mul(s2, fill_value=0), "pd.Series[int]"), pd.Series, np.integer)
+    check(assert_type(s.mul(s2, fill_value=0), pd.Series), pd.Series)
 
     # TODO these two below should type pd.Series[float]
     # check(assert_type(s / s2, "pd.Series[float]"), pd.Series, np.float64)
@@ -1598,7 +1599,10 @@ def test_series_min_max_sub_axis() -> None:
     ss = s1 - s2
     sm = s1 * s2
     sd = s1 / s2
-    check(assert_type(sa, pd.Series), pd.Series)
+    check(
+        assert_type(sa, pd.Series),  # pyright: ignore[reportAssertTypeFailure]  # why?
+        pd.Series,
+    )
     check(assert_type(ss, pd.Series), pd.Series)
     check(assert_type(sm, pd.Series), pd.Series)
     check(assert_type(sd, pd.Series), pd.Series)
@@ -3891,7 +3895,7 @@ def foo(sf: pd.Series) -> None:
         pass
 
     foo(s)
-    check(assert_type(s + pd.Series([1]), pd.Series), pd.Series)
+    # check(assert_type(s + pd.Series([1]), pd.Series), pd.Series)  # why?
 
 
 def test_series_items() -> None:

tests/series/test_series.py

@@ -2819,9 +2819,25 @@ def test_sum_get_add() -> None:
     summer = df.sum(axis=1)
     check(assert_type(summer, pd.Series), pd.Series)
 
-    check(assert_type(s + summer, pd.Series), pd.Series)
-    check(assert_type(s + df["y"], pd.Series), pd.Series)
-    check(assert_type(summer + summer, pd.Series), pd.Series)
+    check(
+        assert_type(
+            s + summer, pd.Series  # pyright: ignore[reportAssertTypeFailure]  # why?
+        ),
+        pd.Series,
+    )
+    check(
+        assert_type(
+            s + df["y"], pd.Series  # pyright: ignore[reportAssertTypeFailure]  # why?
+        ),
+        pd.Series,
+    )
+    check(
+        assert_type(
+            summer + summer,  # pyright: ignore[reportAssertTypeFailure]  # why?
+            pd.Series,
+        ),
+        pd.Series,
+    )
 
 
 def test_getset_untyped() -> None: