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
def test_append_parray(self, backend): check_import(backend) self.setup() # Given pa1 = self.pa pa2 = get_particle_array(name='s', x=[0.0, 1.0], m=1.0, rho=2.0) h = DeviceHelper(pa1, backend=backend) pa1.set_device_helper(h) # When h.append_parray(pa2) # Then assert h.get_number_of_particles() == 4
def test_extend(self, backend): check_import(backend) self.setup() # Given pa = self.pa h = DeviceHelper(pa, backend=backend) # When pa.set_device_helper(h) h.extend(4) # Then assert h.get_number_of_particles() == 6