def test_pull_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)
h.x.set(np.array([2.0, 3.0], h.x.dtype))
h.rho[0] = 1.0
h.tag[:] = 1
h.pull('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())
def test_pull_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)
h.x[:] = 1.0
h.rho[:] = 1.0
h.m[:] = 1.0
h.tag[:] = np.array([1, 2], h.tag.dtype)
h.pull()
# 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())
