diff --git a/pandas-stubs/core/frame.pyi b/pandas-stubs/core/frame.pyi index 20879b000..8607343c7 100644 --- a/pandas-stubs/core/frame.pyi +++ b/pandas-stubs/core/frame.pyi @@ -624,6 +624,7 @@ class DataFrame(NDFrame, OpsMixin): "number", "datetime64", "datetime", + "integer", "timedelta", "timedelta64", "datetimetz", diff --git a/tests/test_frame.py b/tests/test_frame.py index 4b68f391c..32f23e240 100644 --- a/tests/test_frame.py +++ b/tests/test_frame.py @@ -3226,6 +3226,7 @@ def test_convert_dtypes_dtype_backend() -> None: def test_select_dtypes() -> None: df = pd.DataFrame({"a": [1, 2] * 3, "b": [True, False] * 3, "c": [1.0, 2.0] * 3}) check(assert_type(df.select_dtypes("number"), pd.DataFrame), pd.DataFrame) + check(assert_type(df.select_dtypes("integer"), pd.DataFrame), pd.DataFrame) check(assert_type(df.select_dtypes(np.number), pd.DataFrame), pd.DataFrame) check(assert_type(df.select_dtypes(object), pd.DataFrame), pd.DataFrame) check(assert_type(df.select_dtypes(include="bool"), pd.DataFrame), pd.DataFrame)