def test8(self):
instance = HermiteInterface()
instance.initialize_code()
instance.set_eps2(0)
instance.commit_parameters()
id1, errorcode = instance.new_particle(mass=10.0,
radius=1.0,
x=0.0,
y=0.0,
z=0.0,
vx=0.0,
vy=0.0,
vz=0.0)
id2, errorcode = instance.new_particle(mass=1.0,
radius=1.0,
x=2.0,
y=0.0,
z=0.0,
vx=10.0,
vy=0.0,
vz=0.0)
instance.commit_particles()
potential, errorcode = instance.get_potential(id1)
self.assertEqual(errorcode, 0)
self.assertAlmostRelativeEquals(potential, -1.0 / numpy.sqrt(2.0**2),
8)
total_potential, errorcode = instance.get_potential_energy()
potentials, errorcode = instance.get_potential([id1, id2])
instance.cleanup_code()
instance.stop()
self.assertAlmostRelativeEquals(
total_potential,
numpy.sum(potentials * [10.0, 1.0]) / 2.0)
def test3(self):
hermite = HermiteInterface()
self.assertEqual(0, hermite.set_eps2(0.101))
self.assertEqual([0.101, 0], list(hermite.get_eps2().values()))
self.assertEqual(0, hermite.set_eps2(0.2))
self.assertEqual([0.2, 0], list(hermite.get_eps2().values()))
hermite.cleanup_code()
hermite.stop()
