GH1441 Allow for axis argument in DataFrame.pct_change

Author: loicdiridollou

pandas-stubs/core/frame.pyi

@@ -2225,6 +2225,7 @@ class DataFrame(NDFrame, OpsMixin, _GetItemHack):
         fill_method: None = None,
         freq: Frequency | dt.timedelta | None = ...,
         fill_value: Scalar | NAType | None = ...,
+        axis: Axis | None = ...,
     ) -> Self: ...
     def pop(self, item: _str) -> Series: ...
     def pow(

tests/test_frame.py

@@ -2820,6 +2820,11 @@ def test_dataframe_pct_change() -> None:
         pd.DataFrame,
     )
     check(assert_type(df.pct_change(fill_value=0), pd.DataFrame), pd.DataFrame)
+    check(assert_type(df.pct_change(axis=0), pd.DataFrame), pd.DataFrame)
+    check(assert_type(df.pct_change(axis=1), pd.DataFrame), pd.DataFrame)
+    check(assert_type(df.pct_change(axis="columns"), pd.DataFrame), pd.DataFrame)
+    check(assert_type(df.pct_change(axis="index"), pd.DataFrame), pd.DataFrame)
+
 
 
 def test_indexslice_setitem() -> None: