def test_pull_correctly_sets_values_with_no_args(self): # Given pa = self.pa h = DeviceHelper(pa) # 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 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_pull_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) 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 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()))