Add tests for the array API creation functions

As with the other array API tests, the tests primarily focus on things that
should error. Working behavior is tested by the official array API test suite.

1 files changed, 103 insertions, 0 deletions

diff --git a/numpy/_array_api/tests/test_creation_functions.py b/numpy/_array_api/tests/test_creation_functions.py
new file mode 100644
index 000000000..654f1d9b3
--- /dev/null
+++ b/numpy/_array_api/tests/test_creation_functions.py
@@ -0,0 +1,103 @@
+from numpy.testing import assert_raises
+import numpy as np
+
+from .. import all
+from .._creation_functions import (asarray, arange, empty, empty_like, eye, from_dlpack, full, full_like, linspace, meshgrid, ones, ones_like, zeros, zeros_like)
+from .._array_object import Array
+from .._dtypes import (_all_dtypes, _boolean_dtypes, _floating_dtypes,
+                       _integer_dtypes, _integer_or_boolean_dtypes,
+                       _numeric_dtypes, int8, int16, int32, int64, uint64)
+
+def test_asarray_errors():
+    # Test various protections against incorrect usage
+    assert_raises(TypeError, lambda: Array([1]))
+    assert_raises(TypeError, lambda: asarray(['a']))
+    assert_raises(ValueError, lambda: asarray([1.], dtype=np.float16))
+    assert_raises(OverflowError, lambda: asarray(2**100))
+    # Preferably this would be OverflowError
+    # assert_raises(OverflowError, lambda: asarray([2**100]))
+    assert_raises(TypeError, lambda: asarray([2**100]))
+    asarray([1], device='cpu') # Doesn't error
+    assert_raises(ValueError, lambda: asarray([1], device='gpu'))
+
+    assert_raises(ValueError, lambda: asarray([1], dtype=int))
+    assert_raises(ValueError, lambda: asarray([1], dtype='i'))
+
+def test_asarray_copy():
+    a = asarray([1])
+    b = asarray(a, copy=True)
+    a[0] = 0
+    assert all(b[0] == 1)
+    assert all(a[0] == 0)
+    # Once copy=False is implemented, replace this with
+    # a = asarray([1])
+    # b = asarray(a, copy=False)
+    # a[0] = 0
+    # assert all(b[0] == 0)
+    assert_raises(NotImplementedError, lambda: asarray(a, copy=False))
+
+def test_arange_errors():
+    arange(1, device='cpu') # Doesn't error
+    assert_raises(ValueError, lambda: arange(1, device='gpu'))
+    assert_raises(ValueError, lambda: arange(1, dtype=int))
+    assert_raises(ValueError, lambda: arange(1, dtype='i'))
+
+def test_empty_errors():
+    empty((1,), device='cpu') # Doesn't error
+    assert_raises(ValueError, lambda: empty((1,), device='gpu'))
+    assert_raises(ValueError, lambda: empty((1,), dtype=int))
+    assert_raises(ValueError, lambda: empty((1,), dtype='i'))
+
+def test_empty_like_errors():
+    empty_like(asarray(1), device='cpu') # Doesn't error
+    assert_raises(ValueError, lambda: empty_like(asarray(1), device='gpu'))
+    assert_raises(ValueError, lambda: empty_like(asarray(1), dtype=int))
+    assert_raises(ValueError, lambda: empty_like(asarray(1), dtype='i'))
+
+def test_eye_errors():
+    eye(1, device='cpu') # Doesn't error
+    assert_raises(ValueError, lambda: eye(1, device='gpu'))
+    assert_raises(ValueError, lambda: eye(1, dtype=int))
+    assert_raises(ValueError, lambda: eye(1, dtype='i'))
+
+def test_full_errors():
+    full((1,), 0, device='cpu') # Doesn't error
+    assert_raises(ValueError, lambda: full((1,), 0, device='gpu'))
+    assert_raises(ValueError, lambda: full((1,), 0, dtype=int))
+    assert_raises(ValueError, lambda: full((1,), 0, dtype='i'))
+
+def test_full_like_errors():
+    full_like(asarray(1), 0, device='cpu') # Doesn't error
+    assert_raises(ValueError, lambda: full_like(asarray(1), 0, device='gpu'))
+    assert_raises(ValueError, lambda: full_like(asarray(1), 0, dtype=int))
+    assert_raises(ValueError, lambda: full_like(asarray(1), 0, dtype='i'))
+
+def test_linspace_errors():
+    linspace(0, 1, 10, device='cpu') # Doesn't error
+    assert_raises(ValueError, lambda: linspace(0, 1, 10, device='gpu'))
+    assert_raises(ValueError, lambda: linspace(0, 1, 10, dtype=float))
+    assert_raises(ValueError, lambda: linspace(0, 1, 10, dtype='f'))
+
+def test_ones_errors():
+    ones((1,), device='cpu') # Doesn't error
+    assert_raises(ValueError, lambda: ones((1,), device='gpu'))
+    assert_raises(ValueError, lambda: ones((1,), dtype=int))
+    assert_raises(ValueError, lambda: ones((1,), dtype='i'))
+
+def test_ones_like_errors():
+    ones_like(asarray(1), device='cpu') # Doesn't error
+    assert_raises(ValueError, lambda: ones_like(asarray(1), device='gpu'))
+    assert_raises(ValueError, lambda: ones_like(asarray(1), dtype=int))
+    assert_raises(ValueError, lambda: ones_like(asarray(1), dtype='i'))
+
+def test_zeros_errors():
+    zeros((1,), device='cpu') # Doesn't error
+    assert_raises(ValueError, lambda: zeros((1,), device='gpu'))
+    assert_raises(ValueError, lambda: zeros((1,), dtype=int))
+    assert_raises(ValueError, lambda: zeros((1,), dtype='i'))
+
+def test_zeros_like_errors():
+    zeros_like(asarray(1), device='cpu') # Doesn't error
+    assert_raises(ValueError, lambda: zeros_like(asarray(1), device='gpu'))
+    assert_raises(ValueError, lambda: zeros_like(asarray(1), dtype=int))
+    assert_raises(ValueError, lambda: zeros_like(asarray(1), dtype='i'))