test_vendor

Author: crusaderky

vendor_tests/test_vendor.py

@@ -2,7 +2,7 @@
 
 from typing import Any
 
-import array_api_strict as xp
+import numpy as xp
 from numpy.testing import assert_array_equal
 
 
@@ -31,19 +31,19 @@ def test_vendor_compat():
     )
 
     x = xp.asarray([1, 2, 3])
-    assert array_namespace(x) is xp
+    assert "array_api_compat.numpy" in array_namespace(x).__name__
     to_device(x, device(x))
     assert is_array_api_obj(x)
-    assert is_array_api_strict_namespace(xp)
+    assert not is_array_api_strict_namespace(xp)
     assert not is_cupy_array(x)
     assert not is_cupy_namespace(xp)
     assert not is_dask_array(x)
     assert not is_dask_namespace(xp)
     assert not is_jax_array(x)
     assert not is_jax_namespace(xp)
     assert not is_lazy_array(x)
-    assert not is_numpy_array(x)
-    assert not is_numpy_namespace(xp)
+    assert is_numpy_array(x)
+    assert is_numpy_namespace(xp)
     assert not is_pydata_sparse_array(x)
     assert not is_pydata_sparse_namespace(xp)
     assert not is_torch_array(x)