Skip to content

Commit 7a30569

Browse files
committed
test for corrected mask_position and boolean indexing behavior
1 parent 8ed4fdf commit 7a30569

File tree

1 file changed

+13
-0
lines changed

1 file changed

+13
-0
lines changed

dpctl/tests/test_usm_ndarray_indexing.py

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1044,6 +1044,19 @@ def test_extract_all_1d():
10441044
res2 = dpt.extract(sel, x)
10451045
assert (dpt.asnumpy(res2) == expected_res).all()
10461046

1047+
# test strided case
1048+
x = dpt.arange(15, dtype="i4")
1049+
sel_np = np.zeros(15, dtype="?")
1050+
np.put(sel_np, np.random.choice(sel_np.size, size=7), True)
1051+
sel = dpt.asarray(sel_np)
1052+
1053+
res = x[sel[::-1]]
1054+
expected_res = dpt.asnumpy(x)[sel_np[::-1]]
1055+
assert (dpt.asnumpy(res) == expected_res).all()
1056+
1057+
res2 = dpt.extract(sel[::-1], x)
1058+
assert (dpt.asnumpy(res2) == expected_res).all()
1059+
10471060

10481061
def test_extract_all_2d():
10491062
get_queue_or_skip()

0 commit comments

Comments
 (0)