Use asnumpy().all() in tests to avoid additional kernel launches

Author: vlad-perevezentsev

dpctl/tests/test_usm_ndarray_indexing.py

@@ -1435,7 +1435,7 @@ def test_nonzero_f_contig():
 
     for exp, res in zip(expected_res, result):
         assert_array_equal(dpt.asnumpy(res), exp)
-    assert dpt.all(mask[result])
+    assert dpt.asnumpy(mask[result]).all()
 
 
 def test_nonzero_compacting():
@@ -1454,7 +1454,7 @@ def test_nonzero_compacting():
 
     for exp, res in zip(expected_res, result):
         assert_array_equal(dpt.asnumpy(res), exp)
-    assert dpt.all(mask_view[result])
+    assert dpt.asnumpy(mask_view[result]).all()
 
 
 def test_assign_scalar():

dpctl/tests/test_usm_ndarray_operators.py

@@ -129,14 +129,14 @@ def test_mat_ops(namespace):
 @pytest.mark.parametrize("namespace", [dpt, Dummy()])
 def test_comp_ops(namespace):
     try:
-        X = dpt.ones(1, dtype="u8")
+        X = dpt.asarray(1, dtype="u8")
     except dpctl.SyclDeviceCreationError:
         pytest.skip("No SYCL devices available")
     X._set_namespace(namespace)
     assert X.__array_namespace__() is namespace
-    assert dpt.all(X.__gt__(-1))
-    assert dpt.all(X.__ge__(-1))
-    assert not dpt.all(X.__lt__(-1))
-    assert not dpt.all(X.__le__(-1))
-    assert not dpt.all(X.__eq__(-1))
-    assert dpt.all(X.__ne__(-1))
+    assert X.__gt__(-1)
+    assert X.__ge__(-1)
+    assert not X.__lt__(-1)
+    assert not X.__le__(-1)
+    assert not X.__eq__(-1)
+    assert X.__ne__(-1)

dpctl/tests/test_usm_ndarray_sorting.py

@@ -354,11 +354,11 @@ def test_radix_sort_size_1_axis():
 
     x3 = dpt.reshape(dpt.arange(10, dtype="i1"), (10, 1))
     r3 = dpt.sort(x3, kind="radixsort")
-    assert dpt.all(r3 == x3)
+    assert dpt.asnumpy(r3 == x3).all()
 
     x4 = dpt.reshape(dpt.arange(10, dtype="i1"), (1, 10))
     r4 = dpt.sort(x4, axis=0, kind="radixsort")
-    assert dpt.all(r4 == x4)
+    assert dpt.asnumpy(r4 == x4).all()
 
 
 def test_radix_argsort_size_1_axis():
@@ -370,12 +370,12 @@ def test_radix_argsort_size_1_axis():
 
     x2 = dpt.ones([1], dtype="i1")
     r2 = dpt.argsort(x2, kind="radixsort")
-    assert dpt.all(r2 == 0)
+    assert dpt.asnumpy(r2 == 0).all()
 
     x3 = dpt.reshape(dpt.arange(10, dtype="i1"), (10, 1))
     r3 = dpt.argsort(x3, kind="radixsort")
-    assert dpt.all(r3 == 0)
+    assert dpt.asnumpy(r3 == 0).all()
 
     x4 = dpt.reshape(dpt.arange(10, dtype="i1"), (1, 10))
     r4 = dpt.argsort(x4, axis=0, kind="radixsort")
-    assert dpt.all(r4 == 0)
+    assert dpt.asnumpy(r4 == 0).all()