Modified tests for cbrt, copysign, and rsqrt

Now test more type combinations/output types

Author: ndgrigorian

dpctl/tests/elementwise/test_cbrt.py

@@ -21,10 +21,10 @@
 import dpctl.tensor as dpt
 from dpctl.tests.helper import get_queue_or_skip, skip_if_dtype_not_supported
 
-from .utils import _map_to_device_dtype, _real_fp_dtypes
+from .utils import _map_to_device_dtype, _no_complex_dtypes, _real_fp_dtypes
 
 
-@pytest.mark.parametrize("dtype", _real_fp_dtypes)
+@pytest.mark.parametrize("dtype", _no_complex_dtypes)
 def test_cbrt_out_type(dtype):
     q = get_queue_or_skip()
     skip_if_dtype_not_supported(dtype, q)

dpctl/tests/elementwise/test_copysign.py

@@ -22,11 +22,11 @@
 import dpctl.tensor as dpt
 from dpctl.tests.helper import get_queue_or_skip, skip_if_dtype_not_supported
 
-from .utils import _compare_dtypes, _real_fp_dtypes
+from .utils import _compare_dtypes, _no_complex_dtypes, _real_fp_dtypes
 
 
-@pytest.mark.parametrize("op1_dtype", _real_fp_dtypes)
-@pytest.mark.parametrize("op2_dtype", _real_fp_dtypes)
+@pytest.mark.parametrize("op1_dtype", _no_complex_dtypes)
+@pytest.mark.parametrize("op2_dtype", _no_complex_dtypes)
 def test_copysign_dtype_matrix(op1_dtype, op2_dtype):
     q = get_queue_or_skip()
     skip_if_dtype_not_supported(op1_dtype, q)

dpctl/tests/elementwise/test_rsqrt.py

@@ -21,16 +21,16 @@
 import dpctl.tensor as dpt
 from dpctl.tests.helper import get_queue_or_skip, skip_if_dtype_not_supported
 
-from .utils import _map_to_device_dtype, _real_fp_dtypes
+from .utils import _map_to_device_dtype, _no_complex_dtypes, _real_fp_dtypes
 
 
-@pytest.mark.parametrize("dtype", _real_fp_dtypes)
+@pytest.mark.parametrize("dtype", _no_complex_dtypes)
 def test_rsqrt_out_type(dtype):
     q = get_queue_or_skip()
     skip_if_dtype_not_supported(dtype, q)
 
     x = dpt.asarray(1, dtype=dtype, sycl_queue=q)
-    expected_dtype = np.reciprocal(np.sqrt(1, dtype=dtype)).dtype
+    expected_dtype = np.reciprocal(np.sqrt(np.array(1, dtype=dtype))).dtype
     expected_dtype = _map_to_device_dtype(expected_dtype, q.sycl_device)
     assert dpt.rsqrt(x).dtype == expected_dtype