adapt to Windows default precision of numpy ints

jcmgray · May 13, 2024 · bbc72d4 · bbc72d4
1 parent 14af78c
commit bbc72d4
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/test_autoray.py b/tests/test_autoray.py
@@ -880,9 +880,9 @@ def test_zeros_passes_dtype_device(self, backend, dtype):
 ]
 
 creation_builtins = [
-    (float, np.float64),
-    (int, np.int64),
-    (complex, np.complex128),
+    (float, [np.float64]),
+    (int, [np.int32, np.int64]), # np.int32 on Windows and np.int64 else
+    (complex, [np.complex128]),
 ]
 
 
@@ -891,4 +891,4 @@ def test_zeros_passes_dtype_device(self, backend, dtype):
 def test_creation_with_builtins(fn, args, dtype, expected):
     x = dtype(4)
     y = ar.do(fn, *args, like=x)
-    assert y.dtype == expected
+    assert y.dtype in expected