def test_align(backend):
check_import(backend)
# Given
dev_array = Array(np.int32, backend=backend)
orig_array = array.arange(0, 16, 1, dtype=np.int32, backend=backend)
dev_array.set_data(orig_array)
indices = array.arange(15, -1, -1, dtype=np.int32, backend=backend)
# When
dev_array.align(indices)
# Then
assert np.all(dev_array.get() == indices.get())
def test_remove(backend):
check_import(backend)
# Given
dev_array = Array(np.int32, backend=backend)
orig_array = array.arange(0, 16, 1, dtype=np.int32, backend=backend)
dev_array.set_data(orig_array)
indices = array.arange(0, 8, 1, dtype=np.int32, backend=backend)
# When
dev_array.remove(indices)
# Then
assert np.all(dev_array.get() == (8 + indices).get())
def test_copy_values(backend):
check_import(backend)
# Given
dev_array = make_dev_array(backend)
dev_array.fill(2)
dest = array.empty(8, dtype=np.int32, backend=backend)
indices = array.arange(0, 8, 1, dtype=np.int32, backend=backend)
# When
dev_array.copy_values(indices, dest)
# Then
assert np.all(dev_array[:len(indices)].get() == dest.get())
def test_align(self, backend):
check_import(backend)
self.setup()
# Given
pa = self.pa
h = DeviceHelper(pa, backend=backend)
# When
pa.set_device_helper(h)
h.resize(5)
h.x.set(np.array([2.0, 3.0, 4.0, 5.0, 6.0], h.x.dtype))
indices = array.arange(4, -1, -1, dtype=np.int32, backend=backend)
h.align(indices)
# Then
assert np.all(h.x.get() == np.array([6., 5., 4., 3., 2.]))
