We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
2 parents ccdf915 + d70079e commit c716de1Copy full SHA for c716de1
array_api_tests/test_inspection_functions.py
@@ -16,13 +16,14 @@ def test_capabilities(self):
16
17
expected_attr = {"boolean indexing": bool, "data-dependent shapes": bool}
18
if xp.__array_api_version__ >= "2024.12":
19
- expected_attr.update(**{"max dimensions": int})
+ expected_attr.update(**{"max dimensions": type(None) | int})
20
21
for attr, typ in expected_attr.items():
22
assert attr in capabilities, f'capabilites is missing "{attr}".'
23
assert isinstance(capabilities[attr], typ)
24
25
- assert capabilities.get("max dimensions", 100500) > 0
+ max_dims = capabilities.get("max dimensions", 100500)
26
+ assert (max_dims is None) or (max_dims > 0)
27
28
def test_devices(self):
29
out = xp.__array_namespace_info__()
0 commit comments