Skip to content

Commit ce5425b

Browse files
Added tests based on failing examples from PR review
1 parent efa348a commit ce5425b

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

dpctl/tests/test_tensor_asarray.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -337,6 +337,10 @@ def test_asarray_seq_of_arrays_on_different_queues():
337337
assert res.sycl_queue == q3
338338
assert dpt.isdtype(res.dtype, "integral")
339339

340+
res = dpt.asarray([m, [w, range(w.shape[0])]], sycl_queue=q3)
341+
assert res.sycl_queue == q3
342+
assert dpt.isdtype(res.dtype, "integral")
343+
340344
res = dpt.asarray([m, [w, w]], sycl_queue=q)
341345
assert res.sycl_queue == q
342346
assert dpt.isdtype(res.dtype, "integral")
@@ -345,5 +349,9 @@ def test_asarray_seq_of_arrays_on_different_queues():
345349
assert res.sycl_queue == q2
346350
assert dpt.isdtype(res.dtype, "integral")
347351

352+
res = dpt.asarray([w, dpt.asnumpy(w)])
353+
assert res.sycl_queue == w.sycl_queue
354+
assert dpt.isdtype(res.dtype, "integral")
355+
348356
with pytest.raises(dpctl.utils.ExecutionPlacementError):
349357
dpt.asarray([m, [w, py_seq]])

0 commit comments

Comments
 (0)