def xtest8(self):
instance = SmallNInterface()
instance.initialize_code()
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 test9(self):
print "Test SmallNInterface evolve_model"
instance = SmallNInterface()
self.assertEquals(0, instance.initialize_code())
self.assertEquals(0, instance.set_eta(0.001))
self.assertEquals(0, instance.commit_parameters())
# Set up an equal-mass binary on a circular orbit:
self.assertEquals([1, 0], instance.new_particle(0.5, 0.5, 0, 0, 0, 0.5, 0, 0.01).values())
self.assertEquals([2, 0], instance.new_particle(0.5, -0.5, 0, 0, 0,-0.5, 0, 0.01).values())
self.assertEquals(0, instance.commit_particles())
self.assertEquals(0, instance.evolve_model(math.pi))
for result, expected in zip(instance.get_position(1).values(), [-0.5, 0.0, 0.0, 0]):
self.assertAlmostEquals(result, expected, 3)
for result, expected in zip(instance.get_position(2).values(), [0.5, 0.0, 0.0, 0]):
self.assertAlmostEquals(result, expected, 3)
self.assertEquals(0, instance.evolve_model(2 * math.pi))
#print instance.get_time()
#print instance.get_position(1), instance.get_velocity(1)
#print instance.get_position(2)
#for result, expected in zip(instance.get_position(1).values(), [0.5, 0.0, 0.0, 0]):
# self.assertAlmostEquals(result, expected, 3)
#for result, expected in zip(instance.get_position(2).values(), [-0.5, 0.0, 0.0, 0]):
# self.assertAlmostEquals(result, expected, 3)
self.assertEquals(0, instance.cleanup_code())
instance.stop()
def xtest8(self):
instance = SmallNInterface()
instance.initialize_code()
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.assertEquals(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.stop()
self.assertAlmostRelativeEquals(total_potential, numpy.sum(potentials * [10.0, 1.0]) / 2.0)
def test9(self):
print "Test SmallNInterface evolve_model"
instance = SmallNInterface()
self.assertEquals(0, instance.initialize_code())
self.assertEquals(0, instance.set_eta(0.001))
self.assertEquals(0, instance.commit_parameters())
# Set up an equal-mass binary on a circular orbit:
self.assertEquals([1, 0],
instance.new_particle(0.5, 0.5, 0, 0, 0, 0.5, 0,
0.01).values())
self.assertEquals([2, 0],
instance.new_particle(0.5, -0.5, 0, 0, 0, -0.5, 0,
0.01).values())
self.assertEquals(0, instance.commit_particles())
self.assertEquals(0, instance.evolve_model(math.pi))
for result, expected in zip(
instance.get_position(1).values(), [-0.5, 0.0, 0.0, 0]):
self.assertAlmostEquals(result, expected, 3)
for result, expected in zip(
instance.get_position(2).values(), [0.5, 0.0, 0.0, 0]):
self.assertAlmostEquals(result, expected, 3)
self.assertEquals(0, instance.evolve_model(2 * math.pi))
#print instance.get_time()
#print instance.get_position(1), instance.get_velocity(1)
#print instance.get_position(2)
#for result, expected in zip(instance.get_position(1).values(), [0.5, 0.0, 0.0, 0]):
# self.assertAlmostEquals(result, expected, 3)
#for result, expected in zip(instance.get_position(2).values(), [-0.5, 0.0, 0.0, 0]):
# self.assertAlmostEquals(result, expected, 3)
self.assertEquals(0, instance.cleanup_code())
instance.stop()
