Added ArrayLike In Series.clip (#443)

* update

* changed the tests to only assert_type

* added check in tests

* Added check to tests and updated them

Author: ramvikrams

pandas-stubs/core/series.pyi

@@ -74,6 +74,7 @@ from pandas._typing import (
     AggFuncTypeBase,
     AggFuncTypeDictFrame,
     AggFuncTypeSeriesToFrame,
+    AnyArrayLike,
     ArrayLike,
     Axes,
     Axis,
@@ -1065,8 +1066,8 @@ class Series(IndexOpsMixin, NDFrame, Generic[S1]):
     ) -> Scalar | Series[S1]: ...
     def clip(
         self,
-        lower: float | None = ...,
-        upper: float | None = ...,
+        lower: AnyArrayLike | float | None = ...,
+        upper: AnyArrayLike | float | None = ...,
         axis: SeriesAxisType | None = ...,
         inplace: _bool = ...,
         *args,

tests/test_series.py

@@ -1330,3 +1330,11 @@ def test_logical_operators() -> None:
         pd.Series,
         bool,
     )
+
+
+def test_AnyArrayLike_and_clip() -> None:
+    ser = pd.Series([1, 2, 3])
+    s1 = ser.clip(lower=ser)
+    s2 = ser.clip(upper=ser)
+    check(assert_type(s1, pd.Series), pd.Series)
+    check(assert_type(s2, pd.Series), pd.Series)