Add a new func to helper and fix some tests

Author: vlad-perevezentsev

tests/helper.py

@@ -1,12 +1,11 @@
 import dpctl
 import dpnp
+import pytest
 
 
-def get_all_dtypes(no_bool=False,
-                   no_float16=True,
-                   no_complex=False,
-                   no_none=False,
-                   device=None):
+def get_all_dtypes(
+    no_bool=False, no_float16=True, no_complex=False, no_none=False, device=None
+):
     """
     Build a list of types supported by DPNP based on input flags and device capabilities.
     """
@@ -37,3 +36,29 @@ def get_all_dtypes(no_bool=False,
     if not no_none:
         dtypes.append(None)
     return dtypes
+
+
+def skip_or_change_if_dtype_not_supported(dtype, device=None, change_dtype=False):
+    """
+    The function to check input type supported in DPNP based on the device capabilities.
+    """
+
+    dev = dpctl.select_default_device() if device is None else device
+    dev_has_dp = dev.has_aspect_fp64
+    if dtype in [dpnp.float32, dpnp.float64] and dev_has_dp is False:
+        if change_dtype:
+            return dpnp.float32
+        else:
+            pytest.skip(
+                f"{dev.name} does not support double precision floating point types"
+            )
+    dev_has_hp = dev.has_aspect_fp16
+    if dtype in [dpnp.complex64, dpnp.complex128] and dev_has_hp is False:
+        if change_dtype:
+            return dpnp.complex64
+        else:
+            pytest.skip(
+                f"{dev.name} does not support double precision floating point types"
+            )
+
+    return dtype

tests/test_absolute.py

@@ -6,9 +6,7 @@
 import numpy
 
 
-@pytest.mark.parametrize(
-    "dtype", get_all_dtypes(no_bool=True, no_none=True, no_complex=True)
-)
+@pytest.mark.parametrize("dtype", get_all_dtypes(no_bool=True, no_complex=True))
 def test_abs(dtype):
     a = numpy.array([1, 0, 2, -3, -1, 2, 21, -9], dtype=dtype)
     ia = inp.array(a)

tests/test_arithmetic.py

@@ -1,14 +1,15 @@
 import unittest
 import pytest
+from .helper import skip_or_change_if_dtype_not_supported
 
 from tests.third_party.cupy import testing
 
 
 class TestArithmetic(unittest.TestCase):
-
     @testing.for_float_dtypes()
     @testing.numpy_cupy_allclose()
     def test_modf_part1(self, xp, dtype):
+        dtype = skip_or_change_if_dtype_not_supported(dtype, change_dtype=True)
         a = xp.array([-2.5, -1.5, -0.5, 0, 0.5, 1.5, 2.5], dtype=dtype)
         b, _ = xp.modf(a)
 
@@ -17,6 +18,7 @@ def test_modf_part1(self, xp, dtype):
     @testing.for_float_dtypes()
     @testing.numpy_cupy_allclose()
     def test_modf_part2(self, xp, dtype):
+        dtype = skip_or_change_if_dtype_not_supported(dtype, change_dtype=True)
         a = xp.array([-2.5, -1.5, -0.5, 0, 0.5, 1.5, 2.5], dtype=dtype)
         _, c = xp.modf(a)
 
@@ -26,19 +28,22 @@ def test_modf_part2(self, xp, dtype):
     @testing.for_float_dtypes()
     @testing.numpy_cupy_allclose()
     def test_nanprod(self, xp, dtype):
+        dtype = skip_or_change_if_dtype_not_supported(dtype, change_dtype=True)
         a = xp.array([-2.5, -1.5, xp.nan, 10.5, 1.5, xp.nan], dtype=dtype)
         return xp.nanprod(a)
 
     @pytest.mark.usefixtures("allow_fall_back_on_numpy")
     @testing.for_float_dtypes()
     @testing.numpy_cupy_allclose()
     def test_nansum(self, xp, dtype):
+        dtype = skip_or_change_if_dtype_not_supported(dtype, change_dtype=True)
         a = xp.array([-2.5, -1.5, xp.nan, 10.5, 1.5, xp.nan], dtype=dtype)
         return xp.nansum(a)
 
     @testing.for_float_dtypes()
     @testing.numpy_cupy_allclose()
     def test_remainder(self, xp, dtype):
+        dtype = skip_or_change_if_dtype_not_supported(dtype, change_dtype=True)
         a = xp.array([5, -3, -2, -1, -5], dtype=dtype)
         b = xp.full(a.size, 3, dtype=dtype)