Update tests

Author: vlad-perevezentsev

tests/test_mathematical.py

@@ -1168,6 +1168,13 @@ def test_errors(self):
         i_neginf = dpnp.array(1)
         assert_raises(TypeError, dpnp.nan_to_num, ia, neginf=i_neginf)
 
+    @pytest.mark.parametrize("kwarg", ["nan", "posinf", "neginf"])
+    @pytest.mark.parametrize("value", [True, 1 - 0j, [1, 2]])
+    def test_errors_diff_types(self, kwarg, value):
+        ia = dpnp.array([0, 1, dpnp.nan, dpnp.inf, -dpnp.inf])
+        with pytest.raises(TypeError):
+            dpnp.nan_to_num(ia, **{kwarg: value})
+
 
 class TestNextafter:
     @pytest.mark.parametrize("dt", get_float_dtypes())

tests/test_sycl_queue.py

@@ -2342,7 +2342,4 @@ def test_nan_to_num(copy, device):
     result = dpnp.nan_to_num(a, copy=copy)
 
     assert_sycl_queue_equal(result.sycl_queue, a.sycl_queue)
-    if copy:
-        assert result is not a
-    else:
-        assert result is a
+    assert copy == (result is not a)

tests/test_usm_type.py

@@ -1363,7 +1363,4 @@ def test_nan_to_num(copy, usm_type_a):
     result = dp.nan_to_num(a, copy=copy)
 
     assert result.usm_type == usm_type_a
-    if copy:
-        assert result is not a
-    else:
-        assert result is a
+    assert copy == (result is not a)

tests/third_party/cupy/math_tests/test_misc.py

@@ -281,11 +281,8 @@ def test_nan_to_num_broadcast(self, kwarg):
             y = xp.zeros((2, 4), dtype=cupy.default_float_type())
             with pytest.raises(TypeError):
                 xp.nan_to_num(x, **{kwarg: y})
-            # dpnp.nan_to_num() doesn`t support a scalar as an input
-            # convert 0.0 to 0-ndim array
             with pytest.raises(TypeError):
-                x_ndim_0 = xp.array(0.0)
-                xp.nan_to_num(x_ndim_0, **{kwarg: y})
+                xp.nan_to_num(0.0, **{kwarg: y})
 
     @testing.for_all_dtypes(no_bool=True, no_complex=True)
     @testing.numpy_cupy_array_equal()