def test_align_particles(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))
h.tag.set(np.array([0, 0, 1, 0, 1], h.tag.dtype))
h.align_particles()
# Then
x = h.x.get()
assert np.all(np.sort(x[:-2]) == np.array([2., 3., 5.]))
def test_get_number_of_particles(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))
h.tag.set(np.array([0, 0, 1, 0, 1], h.tag.dtype))
h.align_particles()
# Then
assert h.get_number_of_particles() == 5
assert h.get_number_of_particles(real=True) == 3
