def test_set_by_index_src_float64(self):
dest_gpu = gpuarray.to_gpu(np.zeros(3, dtype=np.double))
ind = gpuarray.to_gpu(np.array([0, 2, 4]))
src_gpu = gpuarray.to_gpu(np.arange(5, dtype=np.double))
misc.set_by_index(dest_gpu, ind, src_gpu, 'src')
assert np.allclose(dest_gpu.get(),
np.array([0, 2, 4], dtype=np.double))
dest_gpu = gpuarray.to_gpu(np.arange(5, dtype=np.double))
ind = gpuarray.to_gpu(np.array([], np.int64))
src_gpu = gpuarray.to_gpu(np.array([1, 1, 1], dtype=np.double))
misc.set_by_index(dest_gpu, ind, src_gpu, 'src')
assert np.allclose(dest_gpu.get(),
np.arange(5, dtype=np.double))
def test_set_by_index_dest_float32(self):
dest_gpu = gpuarray.to_gpu(np.arange(5, dtype=np.float32))
ind = gpuarray.to_gpu(np.array([0, 2, 4]))
src_gpu = gpuarray.to_gpu(np.array([1, 1, 1], dtype=np.float32))
misc.set_by_index(dest_gpu, ind, src_gpu, 'dest')
assert np.allclose(dest_gpu.get(),
np.array([1, 1, 1, 3, 1], dtype=np.float32))
dest_gpu = gpuarray.to_gpu(np.arange(5, dtype=np.float32))
ind = gpuarray.to_gpu(np.array([], np.int64))
src_gpu = gpuarray.to_gpu(np.array([1, 1, 1], dtype=np.float32))
misc.set_by_index(dest_gpu, ind, src_gpu, 'dest')
assert np.allclose(dest_gpu.get(),
np.arange(5, dtype=np.float32))
