Skip to content

Commit 7105ef7

Browse files
committed
TST: run on cupy
1 parent f55b22b commit 7105ef7

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

array_api_tests/test_array_object.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -287,9 +287,9 @@ def test_getitem_arrays_and_ints(shape, data):
287287

288288
# XXX: how to properly check
289289
import numpy as np
290-
x_np = np.asarray(x)
291-
out_np = np.asarray(out)
292-
key_np = tuple(k if isinstance(k, int) else np.asarray(k) for k in key)
290+
x_np = np.asarray(x.get())
291+
out_np = np.asarray(out.get())
292+
key_np = tuple(k if isinstance(k, int) else np.asarray(k.get()) for k in key)
293293

294294
np.testing.assert_equal(out_np, x_np[key_np])
295295

0 commit comments

Comments
 (0)