Skip to content

Commit 9df4e75

Browse files
committed
Adds tests for 0D array cases for sorting
1 parent 263b666 commit 9df4e75

File tree

1 file changed

+14
-0
lines changed

1 file changed

+14
-0
lines changed

dpctl/tests/test_usm_ndarray_sorting.py

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -197,3 +197,17 @@ def test_argsort_strided():
197197
idx = dpt.argsort(x_flipped)
198198

199199
assert dpt.all(x_flipped[idx] == x_orig)
200+
201+
202+
def test_sort_0d_array():
203+
get_queue_or_skip()
204+
205+
x = dpt.asarray(1, dtype="i4")
206+
assert dpt.sort(x) == 1
207+
208+
209+
def test_argsort_0d_array():
210+
get_queue_or_skip()
211+
212+
x = dpt.asarray(1, dtype="i4")
213+
assert dpt.argsort(x) == 0

0 commit comments

Comments
 (0)