Add TestPythonScalarConversion

Author: vladislav.perevezentsev

dpnp/tests/test_ndarray.py

@@ -5,6 +5,7 @@
     assert_allclose,
     assert_array_equal,
     assert_equal,
+    assert_raises,
     assert_raises_regex,
 )
 
@@ -17,6 +18,7 @@
     get_complex_dtypes,
     get_float_dtypes,
     has_support_aspect64,
+    numpy_version,
 )
 from .third_party.cupy import testing
 
@@ -530,34 +532,50 @@ def test_print_dpnp_zero_shape():
     assert result == expected
 
 
-# Numpy will raise an error when converting a.ndim > 0 to a scalar
-# TODO: Discuss dpnp behavior according to these future changes
-@pytest.mark.filterwarnings("ignore::DeprecationWarning")
-@pytest.mark.parametrize("func", [bool, float, int, complex])
-@pytest.mark.parametrize("shape", [tuple(), (1,), (1, 1), (1, 1, 1)])
-@pytest.mark.parametrize(
-    "dtype", get_all_dtypes(no_float16=False, no_complex=True)
-)
-def test_scalar_type_casting(func, shape, dtype):
-    a = numpy.full(shape, 5, dtype=dtype)
-    ia = dpnp.full(shape, 5, dtype=dtype)
-    assert func(a) == func(ia)
+class TestPythonScalarConversion:
+    @pytest.mark.parametrize("shape", [tuple(), (1,), (1, 1), (1, 1, 1)])
+    @pytest.mark.parametrize(
+        "dtype", get_all_dtypes(no_float16=False, no_complex=True)
+    )
+    def test_bool_conversion(shape, dtype):
+        a = numpy.full(shape, 5, dtype=dtype)
+        ia = dpnp.full(shape, 5, dtype=dtype)
+        assert bool(a) == bool(ia)
 
+    @pytest.mark.parametrize("shape", [tuple(), (1,), (1, 1), (1, 1, 1)])
+    @pytest.mark.parametrize(
+        "dtype", get_all_dtypes(no_float16=False, no_complex=True)
+    )
+    def test_bool_method_conversion(shape, dtype):
+        a = numpy.full(shape, 5, dtype=dtype)
+        ia = dpnp.full(shape, 5, dtype=dtype)
+        assert a.__bool__() == ia.__bool__()
 
-# Numpy will raise an error when converting a.ndim > 0 to a scalar
-# TODO: Discuss dpnp behavior according to these future changes
-@pytest.mark.filterwarnings("ignore::DeprecationWarning")
-@pytest.mark.parametrize(
-    "method", ["__bool__", "__float__", "__int__", "__complex__"]
-)
-@pytest.mark.parametrize("shape", [tuple(), (1,), (1, 1), (1, 1, 1)])
-@pytest.mark.parametrize(
-    "dtype", get_all_dtypes(no_float16=False, no_complex=True)
-)
-def test_scalar_type_casting_by_method(method, shape, dtype):
-    a = numpy.full(shape, 4.7, dtype=dtype)
-    ia = dpnp.full(shape, 4.7, dtype=dtype)
-    assert_allclose(getattr(a, method)(), getattr(ia, method)(), rtol=1e-06)
+    @pytest.mark.parametrize("func", [float, int, complex])
+    @pytest.mark.parametrize("shape", [tuple(), (1,), (1, 1), (1, 1, 1)])
+    @pytest.mark.parametrize(
+        "dtype", get_all_dtypes(no_float16=False, no_complex=True)
+    )
+    def test_non_bool_conversion(func, shape, dtype):
+        a = numpy.full(shape, 5, dtype=dtype)
+        ia = dpnp.full(shape, 5, dtype=dtype)
+        assert_raises(TypeError, func(ia))
+
+        if numpy_version() >= "2.4.0":
+            assert_raises(TypeError, func(a))
+
+    @pytest.mark.parametrize("method", ["__float__", "__int__", "__complex__"])
+    @pytest.mark.parametrize("shape", [tuple(), (1,), (1, 1), (1, 1, 1)])
+    @pytest.mark.parametrize(
+        "dtype", get_all_dtypes(no_float16=False, no_complex=True)
+    )
+    def test_non_bool_method_conversion(method, shape, dtype):
+        a = numpy.full(shape, 5, dtype=dtype)
+        ia = dpnp.full(shape, 5, dtype=dtype)
+        assert_raises(TypeError, getattr(ia, method)())
+
+        if numpy_version() >= "2.4.0":
+            assert_raises(TypeError, getattr(a, method)())
 
 
 @pytest.mark.parametrize("shape", [(1,), (1, 1), (1, 1, 1)])