fix(comment): https://github.com/pandas-dev/pandas-stubs/pull/1275/files#r2211571551

Author: cmp0xff

tests/series/arithmetic/complex/test_add.py

@@ -59,9 +59,28 @@ def test_add_numpy_array() -> None:
     check(assert_type(left + f, "pd.Series[complex]"), pd.Series, np.complex128)
     check(assert_type(left + c, "pd.Series[complex]"), pd.Series, np.complex128)
 
-    # check(assert_type(i + l, "pd.Series[complex]"), pd.Series, np.complex128)
-    # check(assert_type(f + l, "pd.Series[complex]"), pd.Series, np.complex128)
-    # check(assert_type(c + l, "pd.Series[complex]"), pd.Series, np.complex128)
+    # numpy typing gives ndarray instead of `pd.Series[...]` in reality, which we cannot fix
+    check(
+        assert_type(  # type: ignore[assert-type]
+            i + left, "pd.Series[complex]"  # pyright: ignore[reportAssertTypeFailure]
+        ),
+        pd.Series,
+        np.complex128,
+    )
+    check(
+        assert_type(  # type: ignore[assert-type]
+            f + left, "pd.Series[complex]"  # pyright: ignore[reportAssertTypeFailure]
+        ),
+        pd.Series,
+        np.complex128,
+    )
+    check(
+        assert_type(  # type: ignore[assert-type]
+            c + left, "pd.Series[complex]"  # pyright: ignore[reportAssertTypeFailure]
+        ),
+        pd.Series,
+        np.complex128,
+    )
 
     check(assert_type(left.add(i), "pd.Series[complex]"), pd.Series, np.complex128)
     check(assert_type(left.add(f), "pd.Series[complex]"), pd.Series, np.complex128)

tests/series/arithmetic/float/test_add.py

@@ -59,9 +59,28 @@ def test_add_numpy_array() -> None:
     check(assert_type(left + f, "pd.Series[float]"), pd.Series, np.float64)
     check(assert_type(left + c, "pd.Series[complex]"), pd.Series, np.complex128)
 
-    # check(assert_type(i + l, "pd.Series[float]"), pd.Series, np.float64)
-    # check(assert_type(f + l, "pd.Series[float]"), pd.Series, np.float64)
-    # check(assert_type(c + l, "pd.Series[complex]"), pd.Series, np.complex128)
+    # numpy typing gives ndarray instead of `pd.Series[...]` in reality, which we cannot fix
+    check(
+        assert_type(  # type: ignore[assert-type]
+            i + left, "pd.Series[float]"  # pyright: ignore[reportAssertTypeFailure]
+        ),
+        pd.Series,
+        np.float64,
+    )
+    check(
+        assert_type(  # type: ignore[assert-type]
+            f + left, "pd.Series[float]"  # pyright: ignore[reportAssertTypeFailure]
+        ),
+        pd.Series,
+        np.float64,
+    )
+    check(
+        assert_type(  # type: ignore[assert-type]
+            c + left, "pd.Series[complex]"  # pyright: ignore[reportAssertTypeFailure]
+        ),
+        pd.Series,
+        np.complex128,
+    )
 
     check(assert_type(left.add(i), "pd.Series[float]"), pd.Series, np.float64)
     check(assert_type(left.add(f), "pd.Series[float]"), pd.Series, np.float64)

tests/series/arithmetic/int/test_add.py

@@ -59,9 +59,28 @@ def test_add_numpy_array() -> None:
     check(assert_type(left + f, "pd.Series[float]"), pd.Series, np.float64)
     check(assert_type(left + c, "pd.Series[complex]"), pd.Series, np.complex128)
 
-    # check(assert_type(i + l, "pd.Series[int]"), pd.Series, np.int64)
-    # check(assert_type(f + l, "pd.Series[float]"), pd.Series, np.float64)
-    # check(assert_type(c + l, "pd.Series[complex]"), pd.Series, np.complex128)
+    # numpy typing gives ndarray instead of `pd.Series[...]` in reality, which we cannot fix
+    check(
+        assert_type(  # type: ignore[assert-type]
+            i + left, "pd.Series[int]"  # pyright: ignore[reportAssertTypeFailure]
+        ),
+        pd.Series,
+        np.int64,
+    )
+    check(
+        assert_type(  # type: ignore[assert-type]
+            f + left, "pd.Series[float]"  # pyright: ignore[reportAssertTypeFailure]
+        ),
+        pd.Series,
+        np.float64,
+    )
+    check(
+        assert_type(  # type: ignore[assert-type]
+            c + left, "pd.Series[complex]"  # pyright: ignore[reportAssertTypeFailure]
+        ),
+        pd.Series,
+        np.complex128,
+    )
 
     check(assert_type(left.add(i), "pd.Series[int]"), pd.Series, np.int64)
     check(assert_type(left.add(f), "pd.Series[float]"), pd.Series, np.float64)

tests/series/arithmetic/test_add.py

@@ -59,9 +59,25 @@ def test_add_numpy_array() -> None:
     check(assert_type(left + f, pd.Series), pd.Series)
     check(assert_type(left + c, pd.Series), pd.Series)
 
-    # check(assert_type(i + l, pd.Series), pd.Series)
-    # check(assert_type(f + l, pd.Series), pd.Series)
-    # check(assert_type(c + l, pd.Series), pd.Series)
+    # numpy typing gives ndarray instead of `pd.Series[...]` in reality, which we cannot fix
+    check(
+        assert_type(  # type: ignore[assert-type]
+            i + left, pd.Series  # pyright: ignore[reportAssertTypeFailure]
+        ),
+        pd.Series,
+    )
+    check(
+        assert_type(  # type: ignore[assert-type]
+            f + left, pd.Series  # pyright: ignore[reportAssertTypeFailure]
+        ),
+        pd.Series,
+    )
+    check(
+        assert_type(  # type: ignore[assert-type]
+            c + left, pd.Series  # pyright: ignore[reportAssertTypeFailure]
+        ),
+        pd.Series,
+    )
 
     check(assert_type(left.add(i), pd.Series), pd.Series)
     check(assert_type(left.add(f), pd.Series), pd.Series)