TST: use previous array test

data-apis · May 4, 2023 · 9507a42 · 9507a42
1 parent 4eb945c
commit 9507a42
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/tests/test_array_namespace.py b/tests/test_array_namespace.py
@@ -22,18 +22,22 @@ def test_array_namespace(library, api_version):
 
 def test_array_namespace_errors():
     pytest.raises(TypeError, lambda: array_namespace([1]))
-    pytest.raises(TypeError, lambda: array_namespace([1, 2]))
     pytest.raises(TypeError, lambda: array_namespace())
 
     import numpy as np
-    import torch
     x = np.asarray([1, 2])
+
+    pytest.raises(TypeError, lambda: array_namespace((x, x)))
+    pytest.raises(TypeError, lambda: array_namespace(x, (x, x)))
+
+    import torch
     y = torch.asarray([1, 2])
 
     pytest.raises(TypeError, lambda: array_namespace(x, y))
 
     pytest.raises(ValueError, lambda: array_namespace(x, api_version='2022.12'))
 
+
 def test_get_namespace():
     # Backwards compatible wrapper
     assert array_api_compat.get_namespace is array_api_compat.array_namespace