Disallow bool scalar conversation and update tests

Author: vlad-perevezentsev

dpctl/tensor/_usmarray.pyx

@@ -1140,6 +1140,7 @@ cdef class usm_ndarray:
 
     def __bool__(self):
         if self.size == 1:
+            _check_0d_scalar_conversion(self)
             view = _as_zero_dim_ndarray(self)
             return view.__bool__()
 

dpctl/tests/test_usm_ndarray_ctor.py

@@ -286,18 +286,8 @@ def test_properties(dt):
 @pytest.mark.parametrize("shape", [tuple(), (1,), (1, 1), (1, 1, 1)])
 @pytest.mark.parametrize("dtype", ["|b1", "|u2", "|f4", "|i8"])
 class TestCopyScalar:
-    def test_copy_bool_scalar_with_func(self, shape, dtype):
-        try:
-            X = dpt.usm_ndarray(shape, dtype=dtype)
-        except dpctl.SyclDeviceCreationError:
-            pytest.skip("No SYCL devices available")
-        Y = np.arange(1, X.size + 1, dtype=dtype)
-        X.usm_data.copy_from_host(Y.view("|u1"))
-        Y.shape = tuple()
-        assert bool(X) == bool(Y)
-
-    @pytest.mark.parametrize("func", [float, int, complex])
-    def test_copy_numeric_scalar_with_func(self, func, shape, dtype):
+    @pytest.mark.parametrize("func", [bool, float, int, complex])
+    def test_copy_scalar_with_func(self, func, shape, dtype):
         try:
             X = dpt.usm_ndarray(shape, dtype=dtype)
         except dpctl.SyclDeviceCreationError:
@@ -312,18 +302,10 @@ def test_copy_numeric_scalar_with_func(self, func, shape, dtype):
             # 0D arrays are allowed to convert
             assert func(X) == func(Y)
 
-    def test_copy_bool_scalar_with_method(self, shape, dtype):
-        try:
-            X = dpt.usm_ndarray(shape, dtype=dtype)
-        except dpctl.SyclDeviceCreationError:
-            pytest.skip("No SYCL devices available")
-        Y = np.arange(1, X.size + 1, dtype=dtype)
-        X.usm_data.copy_from_host(Y.view("|u1"))
-        Y = Y.reshape(())
-        assert getattr(X, "__bool__")() == getattr(Y, "__bool__")()
-
-    @pytest.mark.parametrize("method", ["__float__", "__int__", "__complex__"])
-    def test_copy_numeric_scalar_with_method(self, method, shape, dtype):
+    @pytest.mark.parametrize(
+        "method", ["__bool__", "__float__", "__int__", "__complex__"]
+    )
+    def test_copy_scalar_with_method(self, method, shape, dtype):
         try:
             X = dpt.usm_ndarray(shape, dtype=dtype)
         except dpctl.SyclDeviceCreationError:

dpctl/tests/test_usm_ndarray_indexing.py

@@ -1430,11 +1430,12 @@ def test_nonzero_f_contig():
     mask = dpt.zeros((5, 5), dtype="?", order="F")
     mask[2, 3] = True
 
-    expected_res = (2, 3)
-    res = dpt.nonzero(mask)
+    expected_res = np.nonzero(dpt.asnumpy(mask))
+    result = dpt.nonzero(mask)
 
-    assert expected_res == res
-    assert mask[res]
+    for exp, res in zip(expected_res, result):
+        assert_array_equal(dpt.asnumpy(res), exp)
+    assert dpt.all(mask[result])
 
 
 def test_nonzero_compacting():
@@ -1448,11 +1449,12 @@ def test_nonzero_compacting():
     mask[3, 2, 1] = True
     mask_view = mask[..., :3]
 
-    expected_res = (3, 2, 1)
-    res = dpt.nonzero(mask_view)
+    expected_res = np.nonzero(dpt.asnumpy(mask_view))
+    result = dpt.nonzero(mask_view)
 
-    assert expected_res == res
-    assert mask_view[res]
+    for exp, res in zip(expected_res, result):
+        assert_array_equal(dpt.asnumpy(res), exp)
+    assert dpt.all(mask_view[result])
 
 
 def test_assign_scalar():

dpctl/tests/test_usm_ndarray_manipulation.py

@@ -1438,11 +1438,11 @@ def test_tile_size_1():
     # test for gh-1627 behavior
     res = dpt.tile(x1, reps)
     assert x1.shape == res.shape
-    assert x1 == res
+    assert_array_equal(dpt.asnumpy(x1), dpt.asnumpy(res))
 
     res = dpt.tile(x2, reps)
     assert x2.shape == res.shape
-    assert x2 == res
+    assert_array_equal(dpt.asnumpy(x2), dpt.asnumpy(res))
 
 
 def test_tile_prepends_axes():

dpctl/tests/test_usm_ndarray_operators.py

@@ -134,9 +134,9 @@ def test_comp_ops(namespace):
         pytest.skip("No SYCL devices available")
     X._set_namespace(namespace)
     assert X.__array_namespace__() is namespace
-    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)
+    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))

dpctl/tests/test_usm_ndarray_sorting.py

@@ -18,6 +18,7 @@
 
 import numpy as np
 import pytest
+from numpy.testing import assert_array_equal
 
 import dpctl.tensor as dpt
 from dpctl.tests.helper import get_queue_or_skip, skip_if_dtype_not_supported
@@ -345,11 +346,11 @@ def test_radix_sort_size_1_axis():
 
     x1 = dpt.ones((), dtype="i1")
     r1 = dpt.sort(x1, kind="radixsort")
-    assert r1 == x1
+    assert_array_equal(dpt.asnumpy(r1), dpt.asnumpy(x1))
 
     x2 = dpt.ones([1], dtype="i1")
     r2 = dpt.sort(x2, kind="radixsort")
-    assert r2 == x2
+    assert_array_equal(dpt.asnumpy(r2), dpt.asnumpy(x2))
 
     x3 = dpt.reshape(dpt.arange(10, dtype="i1"), (10, 1))
     r3 = dpt.sort(x3, kind="radixsort")
@@ -369,7 +370,7 @@ def test_radix_argsort_size_1_axis():
 
     x2 = dpt.ones([1], dtype="i1")
     r2 = dpt.argsort(x2, kind="radixsort")
-    assert r2 == 0
+    assert dpt.all(r2 == 0)
 
     x3 = dpt.reshape(dpt.arange(10, dtype="i1"), (10, 1))
     r3 = dpt.argsort(x3, kind="radixsort")