Skip to content

Commit 759a27e

Browse files
Add test for legacy container holding numpy's array.
1 parent 9929463 commit 759a27e

File tree

1 file changed

+19
-0
lines changed

1 file changed

+19
-0
lines changed

dpctl/tests/test_usm_ndarray_dlpack.py

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -788,6 +788,25 @@ def test_generic_container_legacy():
788788
assert Z.device == X.device
789789

790790

791+
def test_generic_container_legacy_np():
792+
get_queue_or_skip()
793+
C = LegacyContainer(np.linspace(0, 100, num=20, dtype="int16"))
794+
795+
X = dpt.from_dlpack(C)
796+
assert isinstance(X, np.ndarray)
797+
assert X.ctypes.data == C._array.ctypes.data
798+
assert X.dtype == C._array.dtype
799+
800+
Y = dpt.from_dlpack(C, device=(dpt.DLDeviceType.kDLCPU, 0))
801+
assert isinstance(Y, np.ndarray)
802+
assert Y.dtype == X.dtype
803+
804+
dev = dpt.Device.create_device()
805+
Z = dpt.from_dlpack(C, device=dev)
806+
assert isinstance(Z, dpt.usm_ndarray)
807+
assert Z.device == dev
808+
809+
791810
def test_generic_container():
792811
get_queue_or_skip()
793812
C = Container(dpt.linspace(0, 100, num=20, dtype="int16"))

0 commit comments

Comments
 (0)