Skip to content

Commit 25f8afb

Browse files
Adjusted test per change in implementation
1 parent 03e9001 commit 25f8afb

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

dpctl/tests/test_usm_ndarray_dlpack.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -709,8 +709,10 @@ def test_copy_via_host():
709709
assert y.usm_type == "device"
710710

711711
with pytest.raises(ValueError):
712+
# uncorrect length of tuple
712713
dpt.from_dlpack(x_np, device=(1, 0, 0))
713-
with pytest.raises(BufferError):
714+
with pytest.raises(ValueError):
715+
# only kDLCPU and kDLOneAPI are supported
714716
dpt.from_dlpack(x, device=(2, 0))
715717

716718
num_devs = dpctl.get_num_devices()

0 commit comments

Comments
 (0)