Add a test for valid call on 0d input array

Author: oleksandr-pavlyk

dpctl/tests/test_usm_ndarray_top_k.py

@@ -266,6 +266,18 @@ def test_top_k_2d_smallest(dtype, n):
     assert dpt.all(dpt.sort(r.values, axis=1) == dpt.sort(x[:, :k], axis=1))
 
 
+def test_top_k_0d():
+    get_queue_or_skip()
+
+    a = dpt.ones(tuple(), dtype="i4")
+    assert a.ndim == 0
+    assert a.size == 1
+
+    r = dpt.top_k(a, 1)
+    assert r.values == a
+    assert r.indices == dpt.zeros_like(a, dtype=r.indices.dtype)
+
+
 def test_top_k_noncontig():
     get_queue_or_skip()