ENH: testing: add dlpack conversion support (#749)

prady0t · lucascolley · web-flow · commit 80a0dc0b9872 · 2026-05-22T15:35:26.000Z
* Adding  dlpack conversion support

Signed-off-by: Pradyot Ranjan &lt;99216956+pradyotRanjan@users.noreply.github.com&gt;

* precommit error fix

Signed-off-by: Pradyot Ranjan &lt;99216956+pradyotRanjan@users.noreply.github.com&gt;

* precommit error fix

Signed-off-by: Pradyot Ranjan &lt;99216956+pradyotRanjan@users.noreply.github.com&gt;

* fix lint

* remove comment

---------

Signed-off-by: Pradyot Ranjan &lt;99216956+pradyotRanjan@users.noreply.github.com&gt;
Co-authored-by: Pradyot Ranjan &lt;99216956+pradyotRanjan@users.noreply.github.com&gt;
Co-authored-by: Lucas Colley &lt;lucas.colley8@gmail.com&gt;
diff --git a/src/array_api_extra/_lib/_testing.py b/src/array_api_extra/_lib/_testing.py
@@ -136,6 +136,12 @@ def as_numpy_array(array: Array, *, xp: ModuleType) -> np.typing.NDArray[Any]:
         cpu = cast(Device, jax.devices("cpu")[0])
         array = to_device(array, cpu)
 
+    if hasattr(array, "__dlpack__"):
+        try:
+            return np.from_dlpack(array)  # type: ignore[arg-type]  # pyright: ignore[reportArgumentType]
+        except (TypeError, BufferError):
+            pass
+
     return np.asarray(array)
 
 
