ENH: add minimal dlpack tests

Author: ev-br

array_api_tests/test_dlpack.py

@@ -0,0 +1,63 @@
+from enum import Enum
+
+from hypothesis import given, strategies as st
+from . import _array_module as xp
+from . import pytest_helpers as ph
+from . import hypothesis_helpers as hh
+
+# dlpack Enum values,
+# https://data-apis.org/array-api/latest/API_specification/generated/array_api.array.__dlpack_device__.html
+
+class DLPackDeviceEnum(Enum):
+    CPU = 1
+    CUDA = 2
+    CPU_PINNED = 3
+    OPENCL = 4
+    VULKAN = 7
+    METAL = 8
+    VPI = 9
+    ROCM = 10
+    CUDA_MANAGED = 13
+    ONE_API = 14
+
+
+@given(dtype=hh.all_dtypes, data=st.data())
+def test_dlpack_device(dtype, data):
+    """Test the array object __dlpack_device__ method."""
+    # TODO: 1. generate inputs on non-default devices
+    x = xp.empty(3, dtype=dtype)
+    device_type, device_id  = x.__dlpack_device__()
+
+    assert device_type in DLPackDeviceEnum
+    assert isinstance(device_id, int)
+
+
+@given(
+    x=hh.arrays(dtype=hh.all_dtypes, shape=hh.shapes(min_dims=1, max_side=2)),
+    copy_kw=hh.kwargs(copy=st.booleans()),
+    data=st.data()
+)
+def test_from_dlpack(x, copy_kw, data):
+    devices = xp.__array_namespace_info__().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=}"
+    )
+    try:
+        y = xp.from_dlpack(x, **tgt_device_kw, **copy_kw)
+
+        if tgt_device is None:
+            assert y.device == x.device
+            assert xp.all(y == x)
+        else:
+            assert y.device == tgt_device
+
+    except Exception as exc:
+        ph.add_note(exc, repro_snippet)
+        raise