Add a test for non-contiguous input

Author: oleksandr-pavlyk

dpctl/tests/test_usm_ndarray_top_k.py

@@ -264,3 +264,15 @@ def test_top_k_2d_smallest(dtype, n):
         dpt.sort(r.indices, axis=1) == dpt.sort(expected_inds, axis=1)
     )
     assert dpt.all(dpt.sort(r.values, axis=1) == dpt.sort(x[:, :k], axis=1))
+
+
+def test_top_k_noncontig():
+    get_queue_or_skip()
+
+    a = dpt.arange(256, dtype=dpt.int32)[::2]
+    r = dpt.top_k(a, 3)
+
+    assert dpt.all(dpt.sort(r.values) == dpt.asarray([250, 252, 254])), r.values
+    assert dpt.all(
+        dpt.sort(r.indices) == dpt.asarray([125, 126, 127])
+    ), r.indices