refactor test_trig_out_type

Author: ndgrigorian

dpctl/tests/elementwise/test_trigonometric.py

@@ -25,7 +25,7 @@
 import dpctl.tensor as dpt
 from dpctl.tests.helper import get_queue_or_skip, skip_if_dtype_not_supported
 
-from .utils import _all_dtypes, _map_to_device_dtype
+from .utils import _all_dtypes
 
 _trig_funcs = [(np.sin, dpt.sin), (np.cos, dpt.cos), (np.tan, dpt.tan)]
 _inv_trig_funcs = [
@@ -37,23 +37,14 @@
 _dpt_funcs = [t[1] for t in _all_funcs]
 
 
-@pytest.mark.parametrize("np_call, dpt_call", _all_funcs)
+@pytest.mark.parametrize("func", _dpt_funcs)
 @pytest.mark.parametrize("dtype", _all_dtypes)
-def test_trig_out_type(np_call, dpt_call, dtype):
+def test_trig_out_type(func, dtype):
     q = get_queue_or_skip()
     skip_if_dtype_not_supported(dtype, q)
 
-    X = dpt.asarray(0, dtype=dtype, sycl_queue=q)
-    expected_dtype = np_call(np.array(0, dtype=dtype)).dtype
-    expected_dtype = _map_to_device_dtype(expected_dtype, q.sycl_device)
-    assert dpt_call(X).dtype == expected_dtype
-
-    X = dpt.asarray(0, dtype=dtype, sycl_queue=q)
-    expected_dtype = np_call(np.array(0, dtype=dtype)).dtype
-    expected_dtype = _map_to_device_dtype(expected_dtype, q.sycl_device)
-    Y = dpt.empty_like(X, dtype=expected_dtype)
-    dpt_call(X, out=Y)
-    assert_allclose(dpt.asnumpy(dpt_call(X)), dpt.asnumpy(Y))
+    x = dpt.asarray(0, dtype=dtype, sycl_queue=q)
+    assert func(x).dtype == x.dtype
 
 
 @pytest.mark.parametrize("np_call, dpt_call", _all_funcs)