test_square.py::test_sqrt_out_overlap -> test_square.py::test_square_out_overlap

Author: oleksandr-pavlyk

dpctl/tests/elementwise/test_square.py

@@ -108,14 +108,18 @@ def test_square_out_overlap(dtype):
     X = dpt.reshape(X, (3, 5, 4))
 
     Xnp = dpt.asnumpy(X)
-    Ynp = np.sqrt(Xnp, out=Xnp)
+    Ynp = np.square(Xnp, out=Xnp)
 
-    Y = dpt.sqrt(X, out=X)
+    Y = dpt.square(X, out=X)
     assert Y is X
     assert np.allclose(dpt.asnumpy(X), Xnp)
 
-    Ynp = np.sqrt(Xnp, out=Xnp[::-1])
-    Y = dpt.sqrt(X, out=X[::-1])
+    X = dpt.linspace(0, 35, 60, dtype=dtype, sycl_queue=q)
+    X = dpt.reshape(X, (3, 5, 4))
+    Xnp = dpt.asnumpy(X)
+
+    Ynp = np.square(Xnp, out=Xnp[::-1])
+    Y = dpt.square(X, out=X[::-1])
     assert Y is not X
     assert np.allclose(dpt.asnumpy(X), Xnp)
     assert np.allclose(dpt.asnumpy(Y), Ynp)