Tests for strided 1D data input for unique functions

Author: ndgrigorian

dpctl/tests/test_usm_ndarray_unique.py

@@ -63,6 +63,12 @@ def test_unique_values_strided():
     uv = dpt.unique_values(inp)
     assert dpt.all(uv == dpt.arange(2, dtype="i4"))
 
+    inp = dpt.reshape(inp, -1)
+    inp = dpt.flip(dpt.reshape(inp, -1))
+
+    uv = dpt.unique_values(inp)
+    assert dpt.all(uv == dpt.arange(2, dtype="i4"))
+
 
 @pytest.mark.parametrize(
     "dtype",
@@ -108,6 +114,12 @@ def test_unique_counts_strided():
     assert dpt.all(uv == dpt.arange(2, dtype="i4"))
     assert dpt.all(uv_counts == dpt.full(2, n / 2 * m, dtype=uv_counts.dtype))
 
+    inp = dpt.flip(dpt.reshape(inp, -1))
+
+    uv, uv_counts = dpt.unique_counts(inp)
+    assert dpt.all(uv == dpt.arange(2, dtype="i4"))
+    assert dpt.all(uv_counts == dpt.full(2, n / 2 * m, dtype=uv_counts.dtype))
+
 
 @pytest.mark.parametrize(
     "dtype",
@@ -155,6 +167,13 @@ def test_unique_inverse_strided():
     assert dpt.all(inp == uv[inv])
     assert inp.shape == inv.shape
 
+    inp = dpt.flip(dpt.reshape(inp, -1))
+
+    uv, inv = dpt.unique_inverse(inp)
+    assert dpt.all(uv == dpt.arange(2, dtype="i4"))
+    assert dpt.all(inp == uv[inv])
+    assert inp.shape == inv.shape
+
 
 @pytest.mark.parametrize(
     "dtype",
@@ -206,6 +225,15 @@ def test_unique_all_strided():
     assert inp.shape == inv.shape
     assert dpt.all(uv_counts == dpt.full(2, n / 2 * m, dtype=uv_counts.dtype))
 
+    inp = dpt.flip(dpt.reshape(inp, -1))
+
+    uv, ind, inv, uv_counts = dpt.unique_all(inp)
+    assert dpt.all(uv == dpt.arange(2, dtype="i4"))
+    assert dpt.all(uv == inp[ind])
+    assert dpt.all(inp == uv[inv])
+    assert inp.shape == inv.shape
+    assert dpt.all(uv_counts == dpt.full(2, n / 2 * m, dtype=uv_counts.dtype))
+
 
 def test_set_functions_empty_input():
     get_queue_or_skip()