Skip to content

Commit 885c635

Browse files
committed
address review comments
1 parent 02dd391 commit 885c635

File tree

2 files changed

+12
-4
lines changed

2 files changed

+12
-4
lines changed

dpnp/dpnp_iface_logic.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1302,7 +1302,7 @@ def isin(
13021302
)
13031303
usm_element = dpnp.get_usm_ndarray(element)
13041304
usm_test = dpnp.get_usm_ndarray(test_elements)
1305-
return dpnp.get_result_array(
1305+
return dpnp_array._create_from_usm_ndarray(
13061306
dpt.isin(
13071307
usm_element,
13081308
usm_test,

dpnp/tests/test_logic.py

Lines changed: 11 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,7 @@
1+
import dpctl
12
import numpy
23
import pytest
4+
from dpctl.utils import ExecutionPlacementError
35
from numpy.testing import (
46
assert_allclose,
57
assert_array_equal,
@@ -880,12 +882,18 @@ def test_isin_empty(self):
880882
assert_equal(result, expected)
881883

882884
def test_isin_errors(self):
883-
a = dpnp.arange(5)
884-
b = dpnp.arange(3)
885+
q1 = dpctl.SyclQueue()
886+
q2 = dpctl.SyclQueue()
887+
888+
a = dpnp.arange(5, sycl_queue=q1)
889+
b = dpnp.arange(3, sycl_queue=q2)
885890

886891
# unsupported type for elements or test_elements
887892
with pytest.raises(TypeError):
888-
dpnp.isin(dict(), b)
893+
dpnp.isin(dict(), a)
889894

890895
with pytest.raises(TypeError):
891896
dpnp.isin(a, dict())
897+
898+
with pytest.raises(ExecutionPlacementError):
899+
dpnp.isin(a, b)

0 commit comments

Comments
 (0)