use keywrod argument

Author: vtavana

dpctl/tests/elementwise/test_abs.py

@@ -23,15 +23,15 @@ def test_abs_out_type(dtype):
         }
         assert dpt.abs(X).dtype == type_map[arg_dt]
 
-        out = dpt.empty_like(X, dtype=type_map[arg_dt])
-        dpt.abs(X, out)
-        assert np.allclose(dpt.asnumpy(out), dpt.asnumpy(dpt.abs(X)))
+        r = dpt.empty_like(X, dtype=type_map[arg_dt])
+        dpt.abs(X, out=r)
+        assert np.allclose(dpt.asnumpy(r), dpt.asnumpy(dpt.abs(X)))
     else:
         assert dpt.abs(X).dtype == arg_dt
 
-        out = dpt.empty_like(X, dtype=arg_dt)
-        dpt.abs(X, out)
-        assert np.allclose(dpt.asnumpy(out), dpt.asnumpy(dpt.abs(X)))
+        r = dpt.empty_like(X, dtype=arg_dt)
+        dpt.abs(X, out=r)
+        assert np.allclose(dpt.asnumpy(r), dpt.asnumpy(dpt.abs(X)))
 
 
 @pytest.mark.parametrize("usm_type", _usm_types)

dpctl/tests/elementwise/test_add.py

@@ -33,9 +33,9 @@ def test_add_dtype_matrix(op1_dtype, op2_dtype):
     assert (dpt.asnumpy(r) == np.full(r.shape, 2, dtype=r.dtype)).all()
     assert r.sycl_queue == ar1.sycl_queue
 
-    out = dpt.empty_like(ar1, dtype=r.dtype)
-    dpt.add(ar1, ar2, out)
-    assert (dpt.asnumpy(out) == np.full(out.shape, 2, dtype=out.dtype)).all()
+    r2 = dpt.empty_like(ar1, dtype=r.dtype)
+    dpt.add(ar1, ar2, out=r2)
+    assert (dpt.asnumpy(r2) == np.full(r2.shape, 2, dtype=r2.dtype)).all()
 
     ar3 = dpt.ones(sz, dtype=op1_dtype)
     ar4 = dpt.ones(2 * sz, dtype=op2_dtype)
@@ -49,9 +49,9 @@ def test_add_dtype_matrix(op1_dtype, op2_dtype):
     assert r.shape == ar3.shape
     assert (dpt.asnumpy(r) == np.full(r.shape, 2, dtype=r.dtype)).all()
 
-    out = dpt.empty_like(ar1, dtype=r.dtype)
-    dpt.add(ar3[::-1], ar4[::2], out)
-    assert (dpt.asnumpy(out) == np.full(out.shape, 2, dtype=out.dtype)).all()
+    r2 = dpt.empty_like(ar1, dtype=r.dtype)
+    dpt.add(ar3[::-1], ar4[::2], out=r2)
+    assert (dpt.asnumpy(r2) == np.full(r2.shape, 2, dtype=r2.dtype)).all()
 
 
 @pytest.mark.parametrize("op1_usm_type", _usm_types)
@@ -131,17 +131,13 @@ def test_add_broadcasting():
     r2 = dpt.add(v, m)
     assert (dpt.asnumpy(r2) == np.arange(1, 6, dtype="i4")[np.newaxis, :]).all()
 
-    out = dpt.empty_like(m)
-    dpt.add(m, v, out)
-    assert (
-        dpt.asnumpy(out) == np.arange(1, 6, dtype="i4")[np.newaxis, :]
-    ).all()
-
-    out2 = dpt.empty_like(m)
-    dpt.add(v, m, out2)
-    assert (
-        dpt.asnumpy(out2) == np.arange(1, 6, dtype="i4")[np.newaxis, :]
-    ).all()
+    r3 = dpt.empty_like(m)
+    dpt.add(m, v, out=r3)
+    assert (dpt.asnumpy(r3) == np.arange(1, 6, dtype="i4")[np.newaxis, :]).all()
+
+    r4 = dpt.empty_like(m)
+    dpt.add(v, m, out=r4)
+    assert (dpt.asnumpy(r4) == np.arange(1, 6, dtype="i4")[np.newaxis, :]).all()
 
 
 def test_add_broadcasting_error():

dpctl/tests/elementwise/test_cos.py

@@ -25,7 +25,7 @@ def test_cos_out_type(dtype):
     expected_dtype = np.cos(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.cos(X, Y)
+    dpt.cos(X, out=Y)
     np.testing.assert_allclose(dpt.asnumpy(dpt.cos(X)), dpt.asnumpy(Y))
 
 
@@ -48,7 +48,7 @@ def test_cos_output(dtype):
     )
 
     Z = dpt.empty_like(X, dtype=dtype)
-    dpt.cos(X, Z)
+    dpt.cos(X, out=Z)
 
     np.testing.assert_allclose(
         dpt.asnumpy(Z), np.repeat(np.cos(Xnp), n_rep), atol=tol, rtol=tol

dpctl/tests/elementwise/test_isfinite.py

@@ -41,9 +41,9 @@ def test_isfinite_complex(dtype):
     Y = dpt.asarray(Ynp, sycl_queue=q)
     assert np.array_equal(dpt.asnumpy(dpt.isfinite(Y)), np.isfinite(Ynp))
 
-    out = dpt.empty_like(Y, dtype="bool")
-    dpt.isfinite(Y, out)
-    assert np.array_equal(dpt.asnumpy(out)[()], np.isfinite(Ynp))
+    r = dpt.empty_like(Y, dtype="bool")
+    dpt.isfinite(Y, out=r)
+    assert np.array_equal(dpt.asnumpy(r)[()], np.isfinite(Ynp))
 
 
 @pytest.mark.parametrize("dtype", ["f2", "f4", "f8"])
@@ -60,9 +60,9 @@ def test_isfinite_floats(dtype):
         Y = dpt.asarray(Ynp, sycl_queue=q)
         assert np.array_equal(dpt.asnumpy(dpt.isfinite(Y)), np.isfinite(Ynp))
 
-        out = dpt.empty_like(Y, dtype="bool")
-        dpt.isfinite(Y, out)
-        assert np.array_equal(dpt.asnumpy(out)[()], np.isfinite(Ynp))
+        r = dpt.empty_like(Y, dtype="bool")
+        dpt.isfinite(Y, out=r)
+        assert np.array_equal(dpt.asnumpy(r)[()], np.isfinite(Ynp))
 
 
 @pytest.mark.parametrize("dtype", _all_dtypes)

dpctl/tests/elementwise/test_isnan.py

@@ -41,9 +41,9 @@ def test_isnan_complex(dtype):
     Y = dpt.asarray(Ynp, sycl_queue=q)
     assert np.array_equal(dpt.asnumpy(dpt.isnan(Y)), np.isnan(Ynp))
 
-    out = dpt.empty_like(Y, dtype="bool")
-    dpt.isnan(Y, out)
-    assert np.array_equal(dpt.asnumpy(out)[()], np.isnan(Ynp))
+    r = dpt.empty_like(Y, dtype="bool")
+    dpt.isnan(Y, out=r)
+    assert np.array_equal(dpt.asnumpy(r)[()], np.isnan(Ynp))
 
 
 @pytest.mark.parametrize("dtype", ["f2", "f4", "f8"])
@@ -60,9 +60,9 @@ def test_isnan_floats(dtype):
         Y = dpt.asarray(Ynp, sycl_queue=q)
         assert np.array_equal(dpt.asnumpy(dpt.isnan(Y)), np.isnan(Ynp))
 
-        out = dpt.empty_like(Y, dtype="bool")
-        dpt.isnan(Y, out)
-        assert np.array_equal(dpt.asnumpy(out)[()], np.isnan(Ynp))
+        r = dpt.empty_like(Y, dtype="bool")
+        dpt.isnan(Y, out=r)
+        assert np.array_equal(dpt.asnumpy(r)[()], np.isnan(Ynp))
 
 
 @pytest.mark.parametrize("dtype", _all_dtypes)