Merge pull request #108 from lucascolley/dlpack-device

BUG: `from_dlpack`: fix default device

Author: betatim

array_api_strict/_creation_functions.py

@@ -226,6 +226,8 @@ def from_dlpack(
     # Going to wait for upstream numpy support
     if device is not _default:
         _check_device(device)
+    else:
+        device = None
     if copy not in [_default, None]:
         raise NotImplementedError("The copy argument to from_dlpack is not yet implemented")
 

array_api_strict/tests/test_creation_functions.py

@@ -236,3 +236,10 @@ def from_dlpack_2023_12(api_version):
     pytest.raises(exception, lambda: from_dlpack(capsule, copy=False))
     pytest.raises(exception, lambda: from_dlpack(capsule, copy=True))
     pytest.raises(exception, lambda: from_dlpack(capsule, copy=None))
+
+
+def test_from_dlpack_default_device():
+    x = asarray([1, 2, 3])
+    y = from_dlpack(x)
+    z = from_dlpack(np.asarray([1, 2, 3]))
+    assert x.device == y.device == z.device == CPU_DEVICE