Skip to content

Commit 7e2c43c

Browse files
Add test based on example provided in gh-1738
1 parent 3e6c9b6 commit 7e2c43c

File tree

1 file changed

+22
-0
lines changed

1 file changed

+22
-0
lines changed

dpctl/tests/test_usm_ndarray_unique.py

Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -321,3 +321,25 @@ def test_set_functions_compute_follows_data():
321321
assert ind.sycl_queue == q
322322
assert inv_ind.sycl_queue == q
323323
assert uc.sycl_queue == q
324+
325+
326+
def test_gh_1738():
327+
get_queue_or_skip()
328+
329+
ones = dpt.ones(10, dtype="i8")
330+
iota = dpt.arange(10, dtype="i8")
331+
332+
assert ones.device == iota.device
333+
334+
dpt_info = dpt.__array_namespace_info__()
335+
ind_dt = dpt_info.default_dtypes(device=ones.device)["indexing"]
336+
337+
dt = dpt.unique_inverse(ones).inverse_indices.dtype
338+
assert dt == ind_dt
339+
dt = dpt.unique_all(ones).inverse_indices.dtype
340+
assert dt == ind_dt
341+
342+
dt = dpt.unique_inverse(iota).inverse_indices.dtype
343+
assert dt == ind_dt
344+
dt = dpt.unique_all(iota).inverse_indices.dtype
345+
assert dt == ind_dt

0 commit comments

Comments
 (0)