def test_push_correctly_sets_values_with_no_args(self, backend): check_import(backend) self.setup() # Given pa = self.pa h = DeviceHelper(pa, backend=backend) # When pa.set_device_helper(h) pa.x[:] = 1.0 pa.rho[:] = 1.0 pa.m[:] = 1.0 pa.tag[:] = [1, 2] h.push() # Then assert np.allclose(pa.x, h.x.get()) assert np.allclose(pa.y, h.y.get()) assert np.allclose(pa.m, h.m.get()) assert np.allclose(pa.rho, h.rho.get()) assert np.allclose(pa.tag, h.tag.get())
def test_push_correctly_sets_values_with_args(self, backend): check_import(backend) self.setup() # Given pa = self.pa h = DeviceHelper(pa, backend=backend) assert pa.tag[0] == 0 # When pa.set_device_helper(h) pa.x[:] = [2.0, 3.0] pa.rho[0] = 1.0 pa.tag[:] = 1 h.push('x', 'rho', 'tag') # Then assert np.allclose(pa.x, h.x.get()) assert np.allclose(pa.y, h.y.get()) assert np.allclose(pa.m, h.m.get()) assert np.allclose(pa.rho, h.rho.get()) assert np.allclose(pa.tag, h.tag.get())