def test_pyarrow_cuda_buffer():
nb_arr = numba_cuda_DeviceNDArray_as.random(5)
pa_cbuf = numba_cuda_DeviceNDArray_as.pyarrow_cuda_buffer(nb_arr)
arr1 = numba_cuda_DeviceNDArray_as.numpy_ndarray(nb_arr)
arr2 = pyarrow_cuda_buffer_as.numpy_ndarray(pa_cbuf)
np.testing.assert_array_equal(arr1, arr2)
nb_arr[1] = 99
arr1 = numba_cuda_DeviceNDArray_as.numpy_ndarray(nb_arr)
arr2 = pyarrow_cuda_buffer_as.numpy_ndarray(pa_cbuf)
np.testing.assert_array_equal(arr1, arr2)
assert arr1[1] == 99
def test_pyarrow_cuda_buffer():
xd_arr = xnd_xnd_cuda_as.random(5)
pa_cbuf = xnd_xnd_cuda_as.pyarrow_cuda_buffer(xd_arr)
arr1 = xnd_xnd_cuda_as.numpy_ndarray(xd_arr)
arr2 = pyarrow_cuda_buffer_as.numpy_ndarray(pa_cbuf)
np.testing.assert_array_equal(arr1, arr2)
pa_cbuf.copy_from_host(np.array([99]), 1, 1)
arr1 = xnd_xnd_cuda_as.numpy_ndarray(xd_arr)
arr2 = pyarrow_cuda_buffer_as.numpy_ndarray(pa_cbuf)
np.testing.assert_array_equal(arr1, arr2)
assert arr1[1] == 99
def test_pyarrow_cuda_buffer():
cp_arr = cupy_ndarray_as.random(5)
pa_cbuf = cupy_ndarray_as.pyarrow_cuda_buffer(cp_arr)
arr1 = cupy_ndarray_as.numpy_ndarray(cp_arr)
arr2 = pyarrow_cuda_buffer_as.numpy_ndarray(pa_cbuf)
np.testing.assert_array_equal(arr1, arr2)
cp_arr[1] = 99
arr1 = cupy_ndarray_as.numpy_ndarray(cp_arr)
arr2 = pyarrow_cuda_buffer_as.numpy_ndarray(pa_cbuf)
np.testing.assert_array_equal(arr1, arr2)
assert arr1[1] == 99