diff --git a/pandas-stubs/api/types/__init__.pyi b/pandas-stubs/api/types/__init__.pyi index 3a5292a18..bfdaa105b 100644 --- a/pandas-stubs/api/types/__init__.pyi +++ b/pandas-stubs/api/types/__init__.pyi @@ -1,6 +1,7 @@ from pandas._libs.lib import infer_dtype as infer_dtype from pandas.core.dtypes.api import ( + is_any_real_numeric_dtype as is_any_real_numeric_dtype, is_array_like as is_array_like, is_bool as is_bool, is_bool_dtype as is_bool_dtype, diff --git a/pandas-stubs/core/dtypes/api.pyi b/pandas-stubs/core/dtypes/api.pyi index 89fc968de..e52ace018 100644 --- a/pandas-stubs/core/dtypes/api.pyi +++ b/pandas-stubs/core/dtypes/api.pyi @@ -1,4 +1,5 @@ from pandas.core.dtypes.common import ( + is_any_real_numeric_dtype as is_any_real_numeric_dtype, is_array_like as is_array_like, is_bool as is_bool, is_bool_dtype as is_bool_dtype, diff --git a/pandas-stubs/core/dtypes/common.pyi b/pandas-stubs/core/dtypes/common.pyi index 5147c2752..8a0a95edc 100644 --- a/pandas-stubs/core/dtypes/common.pyi +++ b/pandas-stubs/core/dtypes/common.pyi @@ -53,4 +53,5 @@ def is_float_dtype(arr_or_dtype: _ArrayOrDtype) -> bool: ... def is_bool_dtype(arr_or_dtype: _ArrayOrDtype) -> bool: ... def is_extension_array_dtype(arr_or_dtype: _ArrayOrDtype) -> bool: ... def is_complex_dtype(arr_or_dtype: _ArrayOrDtype) -> bool: ... +def is_any_real_numeric_dtype(arr_or_dtype: _ArrayOrDtype) -> bool: ... def pandas_dtype(dtype: object) -> DtypeObj: ... diff --git a/tests/test_dtypes.py b/tests/test_dtypes.py index 53c8d81eb..68c8f98be 100644 --- a/tests/test_dtypes.py +++ b/tests/test_dtypes.py @@ -13,6 +13,7 @@ import numpy as np import pandas as pd +from pandas.api.types import is_any_real_numeric_dtype from pandas.core.arrays import BooleanArray # noqa: F401 from pandas.core.arrays import IntegerArray # noqa: F401 import pyarrow as pa @@ -144,3 +145,9 @@ def test_arrow_dtype() -> None: a_dt = pd.ArrowDtype(pa.int64()) check(assert_type(a_dt, pd.ArrowDtype), pd.ArrowDtype) check(assert_type(a_dt.pyarrow_dtype, pa.DataType), pa.DataType) + + +def test_is_any_real_numeric_dtype() -> None: + check(assert_type(is_any_real_numeric_dtype(np.array([1, 2])), bool), bool) + check(assert_type(is_any_real_numeric_dtype(int), bool), bool) + check(assert_type(is_any_real_numeric_dtype(float), bool), bool)