GH1089 Fix import

Author: loicdiridollou

pandas-stubs/core/series.pyi

@@ -1954,7 +1954,7 @@ class Series(IndexOpsMixin[S1], NDFrame):
         level: Level | None = ...,
         fill_value: float | None = ...,
         axis: AxisIndex | None = ...,
-    ) -> Series[S1]: ...
+    ) -> Self: ...
     def ge(
         self,
         other: Scalar | Series[S1],
@@ -2065,6 +2065,14 @@ class Series(IndexOpsMixin[S1], NDFrame):
         axis: AxisIndex | None = ...,
     ) -> Series[S1]: ...
     @overload
+    def mul(  # type: ignore[overload-overlap]
+        self,
+        other: Series[S1] | Self,
+        level: Level | None = ...,
+        fill_value: int | None = ...,
+        axis: AxisIndex | None = ...,
+    ) -> Self: ...
+    @overload
     def mul(  # pyright: ignore[reportOverlappingOverload]
         self: Series[int],
         other: Series[int] | int,
@@ -2103,7 +2111,7 @@ class Series(IndexOpsMixin[S1], NDFrame):
         level: Level | None = ...,
         fill_value: float | None = ...,
         axis: AxisIndex | None = ...,
-    ) -> Series: ...
+    ) -> Self: ...
     def multiply(
         self,
         other: num | _ListLike | Series[S1],

tests/test_series_arithmetic.py

@@ -1,5 +1,7 @@
 """Test module for arithmetic operations on Series."""
 
+from typing import cast
+
 import numpy as np
 import pandas as pd
 from typing_extensions import assert_type
@@ -188,3 +190,21 @@ def test_element_wise_float_float() -> None:
     check(
         assert_type(divmod(s, s2), tuple["pd.Series[float]", "pd.Series[float]"]), tuple
     )
+
+
+def test_element_wise_int_unknown() -> None:
+    s = cast(pd.Series, pd.Series([7, -5, 10]))
+    s2 = pd.Series([0, 1, -105])
+
+    check(assert_type(s + s2, pd.Series), pd.Series)
+    check(assert_type(s.add(s2, fill_value=0), "pd.Series[float]"), pd.Series)
+
+    check(assert_type(s - s2, pd.Series), pd.Series)
+    check(assert_type(s.sub(s2, fill_value=0), pd.Series), pd.Series)
+
+    check(assert_type(s * s2, pd.Series), pd.Series)
+    check(assert_type(s.mul(s2, fill_value=0), pd.Series), pd.Series)
+
+    # GH1089 should be the following
+    check(assert_type(s / s2, "pd.Series[float]"), pd.Series)
+    check(assert_type(s.div(s2, fill_value=0), "pd.Series[float]"), pd.Series)