MAINT: safer copy=False handling (no device transfers)

Author: ev-br

array_api_tests/test_dlpack.py

@@ -47,15 +47,20 @@ def test_dlpack_device(dtype, data):
     data=st.data()
 )
 def test_from_dlpack(x, copy_kw, data):
-    devices = xp.__array_namespace_info__().devices()
-    devices = filter_meta(devices)
+    copy = copy_kw["copy"] if copy_kw else None
+    # TODO: 1. test copy;  2. generate inputs on non-default devices;
+    #       3. test for copy=False cross-device transfers
+    if copy is False:
+        devices = [x.device]
+    else:
+        devices = xp.__array_namespace_info__().devices()
+        devices = filter_meta(devices)
+
     tgt_device_kw = data.draw(
         hh.kwargs(device=st.sampled_from(devices) | st.none())
     )
-    # TODO: 1. test copy;  2. generate inputs on non-default devices
     tgt_device = tgt_device_kw['device'] if tgt_device_kw else None
 
-
     repro_snippet = ph.format_snippet(
         f"y = from_dlpack({x!r}, **tgt_device_kw, **copy_kw) with {tgt_device_kw=} and {copy_kw=}"
     )