File tree 1 file changed +7
-4
lines changed
1 file changed +7
-4
lines changed Original file line number Diff line number Diff line change @@ -31,7 +31,10 @@ def test_devices(self):
31
31
assert hasattr (out , "default_device" )
32
32
33
33
assert isinstance (out .devices (), list )
34
- assert out .default_device () in out .devices ()
34
+ if out .default_device () is not None :
35
+ # Per https://github.com/data-apis/array-api/issues/923
36
+ # default_device() can return None. Otherwise, it must be a valid device.
37
+ assert out .default_device () in out .devices ()
35
38
36
39
def test_default_dtypes (self ):
37
40
out = xp .__array_namespace_info__ ()
@@ -40,13 +43,13 @@ def test_default_dtypes(self):
40
43
default_dtypes = out .default_dtypes (device = device )
41
44
assert isinstance (default_dtypes , dict )
42
45
expected_subset = (
43
- {"real floating" , "complex floating" , "integral" }
44
- & available_kinds ()
46
+ {"real floating" , "complex floating" , "integral" }
47
+ & available_kinds ()
45
48
| {"indexing" }
46
49
)
47
50
assert expected_subset .issubset (set (default_dtypes .keys ()))
48
51
49
-
52
+
50
53
atomic_kinds = [
51
54
"bool" ,
52
55
"signed integer" ,
You can’t perform that action at this time.
0 commit comments