diff --git a/pandas-stubs/core/frame.pyi b/pandas-stubs/core/frame.pyi index 7e4f5c2ff..3595e24d2 100644 --- a/pandas-stubs/core/frame.pyi +++ b/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( diff --git a/tests/test_frame.py b/tests/test_frame.py index 5ebed2bc2..6dece6176 100644 --- a/tests/test_frame.py +++ b/tests/test_frame.py @@ -2820,6 +2820,10 @@ 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: