Skip to content

Commit 65cfa04

Browse files
committed
Add compute follows data test for isin
1 parent 59d4a94 commit 65cfa04

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

dpctl/tests/test_tensor_isin.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -246,3 +246,13 @@ def test_isin_py_scalars(dt):
246246
assert isinstance(r1, dpt.usm_ndarray)
247247
r2 = dpt.isin(sc, x)
248248
assert isinstance(r2, dpt.usm_ndarray)
249+
250+
251+
def test_isin_compute_follows_data():
252+
q1 = get_queue_or_skip()
253+
q2 = get_queue_or_skip()
254+
255+
x = dpt.ones(10, sycl_queue=q1)
256+
test = dpt.ones_like(x, sycl_queue=q2)
257+
with pytest.raises(ExecutionPlacementError):
258+
dpt.isin(x, test)

0 commit comments

Comments
 (0)