Add a test that vendoring works

Author: asmeurer

tests/__init__.py

@@ -0,0 +1,7 @@
+"""
+Basic tests for the compat library
+
+This only tests basic things like that vendoring works. The extensive tests
+are done by the array API test suite https://github.com/data-apis/array-api-tests
+
+"""

tests/test_vendoring.py

@@ -0,0 +1,15 @@
+from pytest import skip
+
+def test_vendoring_numpy():
+    from vendor_test import uses_numpy
+    uses_numpy._test_numpy()
+
+
+def test_vendoring_cupy():
+    try:
+        import cupy
+    except ImportError:
+        skip("CuPy is not installed")
+
+    from vendor_test import uses_cupy
+    uses_cupy._test_cupy()

tests/vendor_test/__init__.py

@@ -0,0 +1,18 @@
+# Basic test that vendoring works
+
+from .vendored._compat import cupy as cp_compat
+
+import cupy as cp
+
+def _test_cupy():
+    a = cp_compat.asarray([1., 2., 3.])
+    b = cp_compat.arange(3, dtype=cp_compat.float32)
+
+    # cp.pow does not exist. Update this to use something else if it is added
+    res = cp_compat.pow(a, b)
+    assert res.dtype == cp_compat.float64 == cp.float64
+    assert isinstance(a, cp.ndarray)
+    assert isinstance(b, cp.ndarray)
+    assert isinstance(res, cp.ndarray)
+
+    cp.testing.assert_allclose(res, [1., 2., 9.])

tests/vendor_test/uses_cupy.py

@@ -0,0 +1,18 @@
+# Basic test that vendoring works
+
+from .vendored._compat import numpy as np_compat
+
+import numpy as np
+
+def _test_numpy():
+    a = np_compat.asarray([1., 2., 3.])
+    b = np_compat.arange(3, dtype=np_compat.float32)
+
+    # np.pow does not exist. Update this to use something else if it is added
+    res = np_compat.pow(a, b)
+    assert res.dtype == np_compat.float64 == np.float64
+    assert isinstance(a, np.ndarray)
+    assert isinstance(b, np.ndarray)
+    assert isinstance(res, np.ndarray)
+
+    np.testing.assert_allclose(res, [1., 2., 9.])

tests/vendor_test/uses_numpy.py

@@ -0,0 +1 @@
+../../../array_api_compat/