def test_push_correctly_sets_values_with_args(self):
# Given
pa = self.pa
h = DeviceHelper(pa)
self.assertEqual(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
self.assertTrue(np.allclose(pa.x, h.x.get()))
self.assertTrue(np.allclose(pa.y, h.y.get()))
self.assertTrue(np.allclose(pa.m, h.m.get()))
self.assertTrue(np.allclose(pa.rho, h.rho.get()))
self.assertTrue(np.allclose(pa.tag, h.tag.get()))
def test_push_correctly_sets_values_with_no_args(self):
# Given
pa = self.pa
h = DeviceHelper(pa)
# 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
self.assertTrue(np.allclose(pa.x, h.x.get()))
self.assertTrue(np.allclose(pa.y, h.y.get()))
self.assertTrue(np.allclose(pa.m, h.m.get()))
self.assertTrue(np.allclose(pa.rho, h.rho.get()))
self.assertTrue(np.allclose(pa.tag, h.tag.get()))