diff --git a/test/test_array.py b/test/test_array.py
index 2e71bbdaaf22e2ac15cd7a89d8f3d36bcc2d8b0c..6be8366cae52b76ba5a11b3055a9e65d287671fb 100644
--- a/test/test_array.py
+++ b/test/test_array.py
@@ -289,8 +289,7 @@ def test_nan_arithmetic(ctx_getter):
     ab = a*b
     ab_gpu = (a_gpu*b_gpu).get()
 
-    for i in range(size):
-        assert np.isnan(ab[i]) == np.isnan(ab_gpu[i])
+    assert (np.isnan(ab) == np.isnan(ab_gpu)).all()