diff --git a/pandas-stubs/core/frame.pyi b/pandas-stubs/core/frame.pyi index 7da149106..fe3fef2f7 100644 --- a/pandas-stubs/core/frame.pyi +++ b/pandas-stubs/core/frame.pyi @@ -570,7 +570,12 @@ class DataFrame(NDFrame, OpsMixin): storage_options: StorageOptions = ..., ) -> str: ... def info( - self, verbose=..., buf=..., max_cols=..., memory_usage=..., null_counts=... + self, + verbose: bool | None = ..., + buf: WriteBuffer[str] = ..., + max_cols: int | None = ..., + memory_usage: bool | Literal["deep"] | None = ..., + show_counts: bool | None = ..., ) -> None: ... def memory_usage(self, index: _bool = ..., deep: _bool = ...) -> Series: ... def transpose(self, *args, copy: _bool = ...) -> DataFrame: ... diff --git a/tests/test_frame.py b/tests/test_frame.py index 93c72656a..0eab81915 100644 --- a/tests/test_frame.py +++ b/tests/test_frame.py @@ -3147,3 +3147,25 @@ def test_get() -> None: pd.DataFrame, ) check(assert_type(df.get(["z"], default=1), Union[pd.DataFrame, int]), int) + + +def test_info() -> None: + df = pd.DataFrame() + + check(assert_type(df.info(verbose=True), None), type(None)) + check(assert_type(df.info(verbose=False), None), type(None)) + check(assert_type(df.info(verbose=None), None), type(None)) + + check(assert_type(df.info(buf=io.StringIO()), None), type(None)) + + check(assert_type(df.info(max_cols=1), None), type(None)) + check(assert_type(df.info(max_cols=None), None), type(None)) + + check(assert_type(df.info(memory_usage=True), None), type(None)) + check(assert_type(df.info(memory_usage=False), None), type(None)) + check(assert_type(df.info(memory_usage="deep"), None), type(None)) + check(assert_type(df.info(memory_usage=None), None), type(None)) + + check(assert_type(df.info(show_counts=True), None), type(None)) + check(assert_type(df.info(show_counts=False), None), type(None)) + check(assert_type(df.info(show_counts=None), None), type(None))