def test_flow_field(self):
h = np.array([.8, .8, .8])
box = np.array([10., 10., 10.])
gamma = 2.6
field_data = constraints.FlowField.field_from_fn(box, h, self.force)
F = constraints.FlowField(
field=field_data,
grid_spacing=h,
gamma=gamma)
p = self.system.part.add(pos=[0, 0, 0], v=[1, 2, 3])
self.system.constraints.add(F)
with self.assertRaisesRegex(RuntimeError, 'Parameter gamma is read-only'):
F.gamma = 2.0
for i in itertools.product(*map(range, 3 * [10])):
x = (h * i)
f_val = np.array(F.call_method("_eval_field", x=x))
np.testing.assert_allclose(f_val, self.force(x))
p.pos = x
self.system.integrator.run(0)
np.testing.assert_allclose(
-gamma * (p.v - f_val), np.copy(p.f), atol=1e-12)
def test_flow_field(self):
h = np.array([.2, .2, .2])
box = np.array([10., 10., 10.])
gamma = 2.6
field_data = constraints.FlowField.field_from_fn(box, h, self.force)
F = constraints.FlowField(field=field_data,
grid_spacing=h,
gamma=gamma)
p = self.system.part.add(pos=[0, 0, 0], v=[1, 2, 3])
self.system.constraints.add(F)
for i in product(*map(range, 3 * [10])):
x = (h * i)
f_val = np.array(F.call_method("_eval_field", x=x))
np.testing.assert_allclose(f_val, self.force(x))
p.pos = x
self.system.integrator.run(0)
np.testing.assert_allclose(
-gamma * (p.v - f_val), np.copy(p.f), atol=1e-12)