test_isdtype

Author: honno

array_api_tests/_array_module.py

@@ -62,6 +62,7 @@ def __repr__(self):
 ]
 _constants = ["e", "inf", "nan", "pi"]
 _funcs = [f.__name__ for funcs in stubs.category_to_funcs.values() for f in funcs]
+_funcs += ["isdtype"]  # TODO: bump spec and update array-api-tests to new spec layout
 _top_level_attrs = _dtypes + _constants + _funcs + stubs.EXTENSIONS
 
 for attr in _top_level_attrs:

array_api_tests/dtype_helpers.py

@@ -23,6 +23,7 @@
     "bool_and_all_int_dtypes",
     "dtype_to_name",
     "dtype_to_scalars",
+    "kind_to_dtypes",
     "is_int_dtype",
     "is_float_dtype",
     "get_scalar_type",
@@ -125,6 +126,17 @@ def __repr__(self):
 )
 
 
+kind_to_dtypes = {
+    "bool": [xp.bool],
+    "signed integer": int_dtypes,
+    "unsigned integer": uint_dtypes,
+    "integral": all_int_dtypes,
+    "real floating": float_dtypes,
+    "complex floating": complex_dtypes,
+    "numeric": numeric_dtypes,
+}
+
+
 def is_int_dtype(dtype):
     return dtype in all_int_dtypes
 

array_api_tests/test_data_type_functions.py

@@ -6,6 +6,7 @@
 from hypothesis import strategies as st
 
 from . import _array_module as xp
+from . import api_version
 from . import dtype_helpers as dh
 from . import hypothesis_helpers as hh
 from . import pytest_helpers as ph
@@ -174,6 +175,46 @@ def test_iinfo(dtype):
     # TODO: test values
 
 
+if api_version >= "2022.12":
+
+    def atomic_kinds() -> st.SearchStrategy[Union[DataType, str]]:
+        return st.one_of(
+            xps.scalar_dtypes(),
+            st.sampled_from(
+                [
+                    "bool",
+                    "signed integer",
+                    "unsigned integer",
+                    "integral",
+                    "real floating",
+                    "complex floating",
+                    "numeric",
+                ]
+            ),
+        )
+
+    @given(
+        dtype=xps.scalar_dtypes(),
+        kind=atomic_kinds() | st.lists(atomic_kinds(), min_size=1).map(tuple),
+    )
+    def test_isdtype(dtype, kind):
+        out = xp.isdtype(dtype, kind)
+
+        assert isinstance(out, bool), f"{type(out)=}, but should be bool [isdtype()]"
+        _kinds = kind if isinstance(kind, tuple) else (kind,)
+        expected = False
+        for _kind in _kinds:
+            if isinstance(_kind, str):
+                if dtype in dh.kind_to_dtypes[_kind]:
+                    expected = True
+                    break
+            else:
+                if dtype == _kind:
+                    expected = True
+                    break
+        assert out == expected, f"{out=}, but should be {expected} [isdtype()]"
+
+
 @given(hh.mutually_promotable_dtypes(None))
 def test_result_type(dtypes):
     out = xp.result_type(*dtypes)