Test series pipe typing

Author: paw-lu

tests/test_series.py

@@ -2847,3 +2847,114 @@ def test_round() -> None:
 def test_series_new_empty() -> None:
     # GH 826
     check(assert_type(pd.Series(), "pd.Series[Any]"), pd.Series)
+
+
+def test_pipe() -> None:
+    ser = pd.Series(range(10))
+
+    def first_arg_series(
+        ser: pd.Series,
+        positional_only: int,
+        /,
+        argument_1: list[float],
+        argument_2: str,
+        *,
+        keyword_only: tuple[int, int],
+    ) -> pd.Series:
+        return ser
+
+    check(
+        assert_type(
+            ser.pipe(
+                first_arg_series,
+                1,
+                [1.0, 2.0],
+                argument_2="hi",
+                keyword_only=(1, 2),
+            ),
+            pd.Series,
+        ),
+        pd.Series,
+    )
+
+    if TYPE_CHECKING_INVALID_USAGE:
+        ser.pipe(
+            first_arg_series,  # type: ignore[arg-type]
+            "a",  # pyright: ignore[reportGeneralTypeIssues]
+            [1.0, 2.0],
+            argument_2="hi",
+            keyword_only=(1, 2),
+        )
+        ser.pipe(
+            first_arg_series,  # type: ignore[arg-type]
+            1,
+            [1.0, "b"],  # pyright: ignore[reportGeneralTypeIssues]
+            argument_2="hi",
+            keyword_only=(1, 2),
+        )
+        ser.pipe(
+            first_arg_series,  # type: ignore[arg-type]
+            1,
+            [1.0, 2.0],
+            argument_2=11,  # pyright: ignore[reportGeneralTypeIssues]
+            keyword_only=(1, 2),
+        )
+        ser.pipe(
+            first_arg_series,  # type: ignore[arg-type]
+            1,
+            [1.0, 2.0],
+            argument_2="hi",
+            keyword_only=(1,),  # pyright: ignore[reportGeneralTypeIssues]
+        )
+        ser.pipe(
+            first_arg_series,  # type: ignore[arg-type]
+            1,
+            [1.0, 2.0],
+            argument_3="hi",  # pyright: ignore[reportGeneralTypeIssues]
+            keyword_only=(1, 2),
+        )
+        ser.pipe(
+            first_arg_series,  # type: ignore[arg-type]
+            1,
+            [1.0, 2.0],
+            11,
+            (1, 2),  # pyright: ignore[reportGeneralTypeIssues]
+        )
+        ser.pipe(
+            first_arg_series,  # type: ignore[arg-type]
+            positional_only=1,  # pyright: ignore[reportGeneralTypeIssues]
+            argument_1=[1.0, 2.0],
+            argument_2=11,
+            keyword_only=(1, 2),
+        )
+
+    def first_arg_not_series(argument_1: int, ser: pd.Series) -> pd.Series:
+        return ser
+
+    # pyright does not like that the paramspec is not specified
+    check(
+        assert_type(
+            ser.pipe(
+                (first_arg_not_series, "ser"),
+                1,  # pyright: ignore[reportGeneralTypeIssues]
+            ),
+            pd.Series,
+        ),
+        pd.Series,
+    )
+
+    if TYPE_CHECKING_INVALID_USAGE:
+        ser.pipe(
+            (
+                first_arg_not_series,  # type: ignore[arg-type]
+                1,  # pyright: ignore[reportGeneralTypeIssues]
+            ),
+            1,  # pyright: ignore[reportGeneralTypeIssues]
+        )
+        ser.pipe(
+            (
+                1,  # type: ignore[arg-type]
+                "df",
+            ),  # pyright: ignore[reportGeneralTypeIssues]
+            1,  # pyright: ignore[reportGeneralTypeIssues]
+        )