Update cupy buffer protocol copy test

asmeurer · Mar 21, 2024 · a194344 · a194344
1 parent 166c650
commit a194344
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tests/test_common.py b/tests/test_common.py
@@ -170,4 +170,8 @@ def test_asarray_copy(library):
     b = asarray(a, copy=None)
     assert is_lib_func(b)
     a[0] = 0.0
-    assert all(b[0] == 0.0)
+    if library == 'cupy':
+        # A copy is required for libraries where the default device is not CPU
+        assert all(b[0] == 1.0)
+    else:
+        assert all(b[0] == 0.0)