(arr, expected_arr)
| 34 | |
| 35 | |
| 36 | def check_dlpack_export(arr, expected_arr): |
| 37 | DLTensor = arr.__dlpack__() |
| 38 | assert PyCapsule_IsValid(DLTensor, b"dltensor") is True |
| 39 | |
| 40 | result = np.from_dlpack(arr) |
| 41 | np.testing.assert_array_equal(result, expected_arr, strict=True) |
| 42 | |
| 43 | assert arr.__dlpack_device__() == (1, 0) |
| 44 | |
| 45 | |
| 46 | def check_bytes_allocated(f): |
no test coverage detected