Skip to content

Commit d507e20

Browse files
committed
Improve test check and fix typo in type check
1 parent 43b63fe commit d507e20

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

dpnp/memory/_memory.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -95,7 +95,7 @@ def create_data(x):
9595
)
9696
usm_data = x.usm_data
9797

98-
if isinstance(usm_data, tuple(dispatch.keys())):
98+
if isinstance(usm_data, tuple(dispatch.values())):
9999
return usm_data
100100

101101
cls = dispatch.get(type(usm_data), None)

dpnp/tests/test_memory.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,4 +30,4 @@ def test_wrong_usm_data(self):
3030
def test_ndarray_from_data(self):
3131
a = dpnp.empty(5)
3232
b = dpnp.ndarray(a.shape, buffer=a.data)
33-
assert b.data is a.data
33+
assert b.data.ptr == a.data.ptr

0 commit comments

Comments
 (0)