Added tests for abs/not_equal with not-aligned start of array

Author: oleksandr-pavlyk

dpctl/tests/elementwise/test_abs.py

@@ -187,3 +187,18 @@ def test_abs_complex_fp_special_values(dtype):
     tol = dpt.finfo(r.dtype).resolution
 
     assert dpt.allclose(r, expected, atol=tol, rtol=tol, equal_nan=True)
+
+
+@pytest.mark.parametrize("dtype", _all_dtypes)
+def test_abs_alignment(dtype):
+    q = get_queue_or_skip()
+    skip_if_dtype_not_supported(dtype, q)
+
+    x = dpt.ones(512, dtype=dtype)
+    r = dpt.abs(x)
+
+    r2 = dpt.abs(x[1:])
+    assert np.allclose(dpt.asnumpy(r[1:]), dpt.asnumpy(r2))
+
+    dpt.abs(x[:-1], out=r[1:])
+    assert np.allclose(dpt.asnumpy(r[1:]), dpt.asnumpy(r2))

dpctl/tests/elementwise/test_not_equal.py

@@ -188,3 +188,21 @@ def __sycl_usm_array_interface__(self):
     c = Canary()
     with pytest.raises(ValueError):
         dpt.not_equal(a, c)
+
+
+@pytest.mark.parametrize("dtype", _all_dtypes)
+def test_not_equal_alignment(dtype):
+    q = get_queue_or_skip()
+    skip_if_dtype_not_supported(dtype, q)
+
+    n = 256
+    s = dpt.concat((dpt.zeros(n, dtype=dtype), dpt.zeros(n, dtype=dtype)))
+
+    mask = s[:-1] != s[1:]
+    (pos,) = dpt.nonzero(mask)
+    assert dpt.all(pos == n)
+
+    out_arr = dpt.zeros(2 * n, dtype=mask.dtype)
+    dpt.not_equal(s[:-1], s[1:], out=out_arr[1:])
+    (pos,) = dpt.nonzero(mask)
+    assert dpt.all(pos == (n + 1))