Update impacted tests with dpnp.__dlpack__

Author: antonwolfy

dpnp/tests/test_dlpack.py

@@ -1,6 +1,7 @@
+import dpctl
 import numpy
 import pytest
-from numpy.testing import assert_array_equal
+from numpy.testing import assert_array_equal, assert_raises
 
 import dpnp
 
@@ -10,11 +11,20 @@
 
 
 class TestDLPack:
-    @pytest.mark.parametrize("stream", [None, 1])
+    @pytest.mark.parametrize("stream", [None, dpctl.SyclQueue()])
     def test_stream(self, stream):
         x = dpnp.arange(5)
         x.__dlpack__(stream=stream)
 
+    @pytest.mark.parametrize(
+        "stream",
+        [1, dict(), dpctl.SyclDevice()],
+        ids=["scalar", "dictionary", "device"],
+    )
+    def test_invaid_stream(self, stream):
+        x = dpnp.arange(5)
+        assert_raises(TypeError, x.__dlpack__, stream=stream)
+
     @pytest.mark.parametrize("copy", [True, None, False])
     def test_copy(self, copy):
         x = dpnp.arange(5)

dpnp/tests/third_party/cupy/core_tests/test_dlpack.py

@@ -206,7 +206,13 @@ def test_stream(self):
         for src_s in [self._get_stream(s) for s in allowed_streams]:
             for dst_s in [self._get_stream(s) for s in allowed_streams]:
                 orig_array = _gen_array(cupy.float32, alloc_q=src_s)
-                dltensor = orig_array.__dlpack__(stream=orig_array)
+
+                q = dpctl.SyclQueue(
+                    orig_array.sycl_context,
+                    orig_array.sycl_device,
+                    property="enable_profiling",
+                )
+                dltensor = orig_array.__dlpack__(stream=q)
 
                 out_array = dlp.from_dlpack_capsule(dltensor)
                 out_array = cupy.from_dlpack(out_array, device=dst_s)