add Series to Series.between (#531)

* add Series to Series.between

* Add test_types_between

* Code review

* Lint

Author: pavelzw

pandas-stubs/core/series.pyi

@@ -97,6 +97,7 @@ from pandas._typing import (
     JsonSeriesOrient,
     Level,
     ListLike,
+    ListLikeU,
     MaskType,
     NaPosition,
     QuantileInterpolation,
@@ -935,8 +936,8 @@ class Series(IndexOpsMixin, NDFrame, Generic[S1]):
     def isin(self, values: Iterable | Series[S1] | dict) -> Series[_bool]: ...
     def between(
         self,
-        left: Scalar | Sequence,
-        right: Scalar | Sequence,
+        left: Scalar | ListLikeU,
+        right: Scalar | ListLikeU,
         inclusive: Literal["both", "neither", "left", "right"] = ...,
     ) -> Series[_bool]: ...
     def isna(self) -> Series[_bool]: ...

tests/test_series.py

@@ -622,6 +622,19 @@ def test_types_compare() -> None:
     s2.compare(s1, align_axis="columns", keep_shape=True, keep_equal=True)
 
 
+def test_types_between() -> None:
+    s1 = pd.Series([1, 2, 3])
+    s2 = pd.Series([0, 1, 2])
+    s3 = pd.Series([2, 3, 4])
+    check(assert_type(s1.between(0, 2), "pd.Series[bool]"), pd.Series, bool)
+    check(
+        assert_type(s1.between([0, 1, 2], [2, 3, 4]), "pd.Series[bool]"),
+        pd.Series,
+        bool,
+    )
+    check(assert_type(s1.between(s2, s3), "pd.Series[bool]"), pd.Series, bool)
+
+
 def test_types_agg() -> None:
     s = pd.Series([1, 2, 3], index=["col1", "col2", "col3"])
     check(assert_type(s.agg("min"), Any), np.int64)