Merge pull request data-apis#337 from adityagoel4512/skip-unused-kinds-in-default

ev-br · web-flow · commit dad773187be0 · 2025-01-31T09:28:49.000+01:00
Do not require default dtype if no dtypes in kind supported
diff --git a/array_api_tests/dtype_helpers.py b/array_api_tests/dtype_helpers.py
@@ -164,6 +164,11 @@ def _make_dtype_tuple_from_names(names: List[str]) -> Tuple[DataType]:
 }
 
 
+def available_kinds():
+    return {
+        kind for kind, dtypes in kind_to_dtypes.items() if dtypes
+    }
+
 def is_int_dtype(dtype):
     return dtype in all_int_dtypes
 
diff --git a/array_api_tests/test_inspection_functions.py b/array_api_tests/test_inspection_functions.py
@@ -1,5 +1,6 @@
 import pytest
 from hypothesis import given, strategies as st
+from array_api_tests.dtype_helpers import available_kinds
 
 from . import xp
 
@@ -16,7 +17,8 @@ def test_array_namespace_info():
 
     default_dtypes = out.default_dtypes()
     assert isinstance(default_dtypes, dict)
-    assert {"real floating", "complex floating", "integral", "indexing"}.issubset(set(default_dtypes.keys()))
+    expected_subset = {"real floating", "complex floating", "integral"} & available_kinds() | {"indexing"}
+    assert expected_subset.issubset(set(default_dtypes.keys()))
 
     devices = out.devices()
     assert isinstance(devices, list)

Original file line number	Diff line number	Diff line change
`@@ -164,6 +164,11 @@ def _make_dtype_tuple_from_names(names: List[str]) -> Tuple[DataType]:`
`164`	`164`	`}`
`165`	`165`
`166`	`166`
	`167`	`+def available_kinds():`
	`168`	`+ return {`
	`169`	`+ kind for kind, dtypes in kind_to_dtypes.items() if dtypes`
	`170`	`+ }`
	`171`	`+`
`167`	`172`	`def is_int_dtype(dtype):`
`168`	`173`	`return dtype in all_int_dtypes`
`169`	`174`