From de44f114a9d9195dd251208ae54c8908fe31d0fb Mon Sep 17 00:00:00 2001 From: Anton Volkov Date: Mon, 20 Feb 2023 09:23:39 -0600 Subject: [PATCH] Simplify asserts in dlpack tests --- tests/test_sycl_queue.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/tests/test_sycl_queue.py b/tests/test_sycl_queue.py index 1bffa18111b8..ea03e21b4335 100644 --- a/tests/test_sycl_queue.py +++ b/tests/test_sycl_queue.py @@ -848,8 +848,7 @@ def test_from_dlpack(arr_dtype, shape, device): Y = dpnp.from_dlpack(X) assert_array_equal(X, Y) assert X.__dlpack_device__() == Y.__dlpack_device__() - assert X.sycl_device == Y.sycl_device - assert X.sycl_context == Y.sycl_context + assert_sycl_queue_equal(X.sycl_queue, Y.sycl_queue) assert X.usm_type == Y.usm_type if Y.ndim: V = Y[::-1] @@ -868,6 +867,5 @@ def test_from_dlpack_with_dpt(arr_dtype, device): assert_array_equal(X, Y) assert isinstance(Y, dpnp.dpnp_array.dpnp_array) assert X.__dlpack_device__() == Y.__dlpack_device__() - assert X.sycl_device == Y.sycl_device - assert X.sycl_context == Y.sycl_context assert X.usm_type == Y.usm_type + assert_sycl_queue_equal(X.sycl_queue, Y.sycl_queue)