Skip to content

Commit bcbff63

Browse files
authored
Simplify asserts in dlpack tests (#1310)
1 parent 0bce9e0 commit bcbff63

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

tests/test_sycl_queue.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -862,8 +862,7 @@ def test_from_dlpack(arr_dtype, shape, device):
862862
Y = dpnp.from_dlpack(X)
863863
assert_array_equal(X, Y)
864864
assert X.__dlpack_device__() == Y.__dlpack_device__()
865-
assert X.sycl_device == Y.sycl_device
866-
assert X.sycl_context == Y.sycl_context
865+
assert_sycl_queue_equal(X.sycl_queue, Y.sycl_queue)
867866
assert X.usm_type == Y.usm_type
868867
if Y.ndim:
869868
V = Y[::-1]
@@ -882,6 +881,5 @@ def test_from_dlpack_with_dpt(arr_dtype, device):
882881
assert_array_equal(X, Y)
883882
assert isinstance(Y, dpnp.dpnp_array.dpnp_array)
884883
assert X.__dlpack_device__() == Y.__dlpack_device__()
885-
assert X.sycl_device == Y.sycl_device
886-
assert X.sycl_context == Y.sycl_context
887884
assert X.usm_type == Y.usm_type
885+
assert_sycl_queue_equal(X.sycl_queue, Y.sycl_queue)

0 commit comments

Comments
 (0)