def test3(self):
convert_nbody = nbody_system.nbody_to_si(1.0 | units.MSun,
1.0 | units.yr / (2.0 * pi))
instance = Mikkola(convert_nbody)
instance.parameters.lightspeed = constants.c
stars = self.new_system_of_Hulse_Taylor_pulsar()
instance.particles.add_particles(stars)
Hulse = stars[0]
Taylor = stars[1]
postion_at_start = Taylor.position.value_in(units.AU)[0]
#orbital period
#see http://www.johnstonsarchive.net/relativity/binpulsar.html
period_HTpulsar = 7.75 | units.hour
#period_HTpulsar = 77.51939106 | units.hour
# period for abseidal motion
# period_HTpulsar = 85.0 | units.yr #4.2degrees/year
#period_HTpulsar = 1.0 | units.yr
instance.evolve_model(period_HTpulsar)
channel = instance.particles.new_channel_to(stars)
channel.copy()
postion_after_full_rotation = Taylor.position.value_in(units.AU)[0]
print("Time=", instance.model_time, period_HTpulsar)
self.assertAlmostEqual(postion_at_start, postion_after_full_rotation,
3)
instance.stop()
del instance
def test16(self):
particles = datamodel.Particles(3)
particles.mass = [0.1, 0.1, 0.0001] | nbody_system.mass
particles.radius = [0.3, 0.3, 0.1] | nbody_system.length
particles.x = [-0.5, 0.5, 10] | nbody_system.length
particles.y = 0 | nbody_system.length
particles.z = 0 | nbody_system.length
particles.velocity = [0, 0, 0] | nbody_system.speed
particles[-1].vy = 10 | nbody_system.speed
instance = Mikkola()
instance.initialize_code()
instance.parameters.set_defaults()
instance.parameters.lightspeed = 0 | nbody_system.speed
instance.particles.add_particles(particles)
collisions = instance.stopping_conditions.collision_detection
collisions.enable()
instance.evolve_model(10.0 | nbody_system.time)
print(instance.particles.position)
self.assertTrue(collisions.is_set())
print(instance.model_time)
print(len(collisions.particles(0)))
self.assertTrue(instance.model_time < 2 | nbody_system.time)
print(collisions.particles(0).key)
print(collisions.particles(1).key)
self.assertEqual(len(collisions.particles(0)), 1)
self.assertEqual(len(collisions.particles(1)), 1)
instance.stop()
def test3(self):
convert_nbody=nbody_system.nbody_to_si(1.0|units.MSun, 1.0|units.yr/(2.0*pi))
instance = Mikkola(convert_nbody)
instance.parameters.lightspeed = constants.c
stars = self.new_system_of_Hulse_Taylor_pulsar()
instance.particles.add_particles(stars)
Hulse = stars[0]
Taylor = stars[1]
postion_at_start = Taylor.position.value_in(units.AU)[0]
#orbital period
#see http://www.johnstonsarchive.net/relativity/binpulsar.html
period_HTpulsar = 7.75 | units.hour
#period_HTpulsar = 77.51939106 | units.hour
# period for abseidal motion
# period_HTpulsar = 85.0 | units.yr #4.2degrees/year
#period_HTpulsar = 1.0 | units.yr
instance.evolve_model(period_HTpulsar)
channel = instance.particles.new_channel_to(stars)
channel.copy()
postion_after_full_rotation = Taylor.position.value_in(units.AU)[0]
print "Time=", instance.model_time, period_HTpulsar
self.assertAlmostEqual(postion_at_start, postion_after_full_rotation, 3)
instance.stop()
del instance
def test14(self):
convert_nbody = nbody_system.nbody_to_si(1.0 | units.MSun,
1.0 | units.yr / (2.0 * pi))
stars = datamodel.Stars(3)
stars[0].mass = 1.0 | units.MSun
stars[0].radius = 1.0 | units.RSun
stars[0].position = [0.0, 0, 0] | units.km
stars[0].velocity = [0.0, 0.0, 0.0] | units.km / units.s
stars[1].mass = 1.0 | units.MSun
stars[1].radius = 1.0 | units.RSun
stars[1].position = [0.1, 0, 0] | units.RSun
stars[1].velocity = [-0.4, 0.0, 0.0] | units.km / units.s
stars[2].mass = 1e-9 | units.MSun
stars[2].radius = 1.0 | units.RSun
stars[2].position = [10000, 0, 0] | units.RSun
stars[2].velocity = [0.0, -100, 0.0] | units.km / units.s
instance = Mikkola(convert_nbody)
instance.particles.add_particles(stars)
instance.evolve_model(0.3 | units.yr)
pos11 = quantities.as_vector_quantity(instance.get_position(1))
pos12 = quantities.as_vector_quantity(instance.get_position(2))
instance.stop()
convert_nbody = nbody_system.nbody_to_si(1.0 | units.MSun,
1.0 | units.yr / (2.0 * pi))
instance = Mikkola(convert_nbody)
stars = datamodel.Stars(2)
stars[0].mass = 1.0 | units.MSun
stars[0].radius = 1.0 | units.RSun
stars[0].position = [0.0, 0, 0] | units.km
stars[0].velocity = [0.0, 0.0, 0.0] | units.km / units.s
stars[1].mass = 1.0 | units.MSun
stars[1].radius = 1.0 | units.RSun
stars[1].position = [0.1, 0, 0] | units.RSun
stars[1].velocity = [-0.4, 0.0, 0.0] | units.km / units.s
instance.particles.add_particles(stars)
self.assertRaises(exceptions.AmuseException, instance.evolve_model,
0.3 | units.yr)
pos21 = quantities.as_vector_quantity(instance.get_position(1))
pos22 = quantities.as_vector_quantity(instance.get_position(2))
instance.stop()
# positions will not be the same but distances will be comparable
self.assertAlmostRelativeEquals((pos22 - pos21).length(),
(pos12 - pos11).length(), 1)
def test14(self):
convert_nbody=nbody_system.nbody_to_si(1.0|units.MSun, 1.0|units.yr/(2.0*pi))
stars = datamodel.Stars(3)
stars[0].mass = 1.0 | units.MSun
stars[0].radius = 1.0| units.RSun
stars[0].position = [0.0, 0, 0] | units.km
stars[0].velocity = [0.0,0.0,0.0] | units.km/units.s
stars[1].mass = 1.0 | units.MSun
stars[1].radius = 1.0| units.RSun
stars[1].position = [0.1, 0, 0] | units.RSun
stars[1].velocity = [-0.4, 0.0, 0.0] | units.km/units.s
stars[2].mass = 1e-9 | units.MSun
stars[2].radius = 1.0| units.RSun
stars[2].position = [10000, 0, 0] | units.RSun
stars[2].velocity = [0.0, -100, 0.0] | units.km/units.s
instance = Mikkola(convert_nbody)
instance.particles.add_particles(stars)
instance.evolve_model(0.3 | units.yr)
pos11 = quantities.as_vector_quantity(instance.get_position(1))
pos12 = quantities.as_vector_quantity(instance.get_position(2))
instance.stop()
convert_nbody=nbody_system.nbody_to_si(1.0|units.MSun, 1.0|units.yr/(2.0*pi))
instance = Mikkola(convert_nbody)
stars = datamodel.Stars(2)
stars[0].mass = 1.0 | units.MSun
stars[0].radius = 1.0| units.RSun
stars[0].position = [0.0, 0, 0] | units.km
stars[0].velocity = [0.0,0.0,0.0] | units.km/units.s
stars[1].mass = 1.0 | units.MSun
stars[1].radius = 1.0| units.RSun
stars[1].position = [0.1, 0, 0] | units.RSun
stars[1].velocity = [-0.4, 0.0, 0.0] | units.km/units.s
instance.particles.add_particles(stars)
self.assertRaises(exceptions.AmuseException, instance.evolve_model, 0.3 | units.yr)
pos21 = quantities.as_vector_quantity(instance.get_position(1))
pos22 = quantities.as_vector_quantity(instance.get_position(2))
instance.stop()
# positions will not be the same but distances will be comparable
self.assertAlmostRelativeEquals((pos22-pos21).length(), (pos12-pos11).length(), 1)
def test8(self):
convert_nbody = nbody_system.nbody_to_si(1.0 | units.MSun,
1.0 | units.yr / (2.0 * pi))
stars = datamodel.Stars(3)
stars[0].mass = 1.0 | units.MSun
stars[0].radius = 1.0 | units.RSun
stars[0].position = [0.0, 0, 0] | units.km
stars[0].velocity = [0.0, 0.0, 0.0] | units.km / units.s
stars[1].mass = 1.0 | units.MSun
stars[1].radius = 1.0 | units.RSun
stars[1].position = [0.1, 0, 0] | units.RSun
stars[1].velocity = [-0.4, 0.0, 0.0] | units.km / units.s
stars[2].mass = 0.01 | units.MSun
stars[2].radius = 1.0 | units.RSun
stars[2].position = [6000, 0, 0] | units.RSun
stars[2].velocity = [0.0, -10, 0.0] | units.km / units.s
instance1 = Mikkola(convert_nbody)
instance1.particles.add_particles(stars)
instance1.evolve_model(0.45 | units.yr)
pos11 = quantities.as_vector_quantity(instance1.get_position(1))
pos12 = quantities.as_vector_quantity(instance1.get_position(2))
posmerger1 = quantities.as_vector_quantity(instance1.get_position(4))
instance1.stop()
instance2 = Mikkola(convert_nbody)
instance2.particles.add_particles(stars)
instance2.evolve_model(0.15 | units.yr)
self.assertAlmostRelativeEquals(instance2.model_time, 0.15 | units.yr,
3)
instance2.evolve_model(0.3 | units.yr)
print(instance2.model_time.as_quantity_in(units.yr))
self.assertAlmostRelativeEquals(instance2.model_time, 0.45 | units.yr,
3)
pos21 = quantities.as_vector_quantity(instance2.get_position(1))
pos22 = quantities.as_vector_quantity(instance2.get_position(2))
posmerger2 = quantities.as_vector_quantity(instance2.get_position(4))
print(pos11, pos21)
print(pos12, pos22)
print(posmerger1, posmerger2)
instance2.stop()
self.assertAlmostRelativeEquals(pos11, pos21)
self.assertAlmostRelativeEquals(pos12, pos22)
self.assertAlmostRelativeEquals(posmerger1[0], posmerger2[0], 4)
self.assertAlmostRelativeEquals(posmerger1[1], posmerger2[1], 1)
def test8(self):
convert_nbody=nbody_system.nbody_to_si(1.0|units.MSun, 1.0|units.yr/(2.0*pi))
stars = datamodel.Stars(3)
stars[0].mass = 1.0 | units.MSun
stars[0].radius = 1.0| units.RSun
stars[0].position = [0.0, 0, 0] | units.km
stars[0].velocity = [0.0,0.0,0.0] | units.km/units.s
stars[1].mass = 1.0 | units.MSun
stars[1].radius = 1.0| units.RSun
stars[1].position = [0.1, 0, 0] | units.RSun
stars[1].velocity = [-0.4, 0.0, 0.0] | units.km/units.s
stars[2].mass = 0.01 | units.MSun
stars[2].radius = 1.0| units.RSun
stars[2].position = [6000, 0, 0] | units.RSun
stars[2].velocity = [0.0, -10, 0.0] | units.km/units.s
instance1 = Mikkola(convert_nbody)
instance1.particles.add_particles(stars)
instance1.evolve_model(0.45 | units.yr)
pos11 = quantities.as_vector_quantity(instance1.get_position(1))
pos12 = quantities.as_vector_quantity(instance1.get_position(2))
posmerger1 = quantities.as_vector_quantity(instance1.get_position(4))
instance1.stop()
instance2 = Mikkola(convert_nbody)
instance2.particles.add_particles(stars)
instance2.evolve_model(0.15 | units.yr)
self.assertAlmostRelativeEquals(instance2.model_time,0.15 | units.yr, 3)
instance2.evolve_model(0.3 | units.yr)
print instance2.model_time.as_quantity_in(units.yr)
self.assertAlmostRelativeEquals(instance2.model_time, 0.45| units.yr, 3)
pos21 = quantities.as_vector_quantity(instance2.get_position(1))
pos22 = quantities.as_vector_quantity(instance2.get_position(2))
posmerger2 = quantities.as_vector_quantity(instance2.get_position(4))
print pos11, pos21
print pos12, pos22
print posmerger1, posmerger2
instance2.stop()
self.assertAlmostRelativeEquals(pos11, pos21)
self.assertAlmostRelativeEquals(pos12, pos22)
self.assertAlmostRelativeEquals(posmerger1[0], posmerger2[0], 4)
self.assertAlmostRelativeEquals(posmerger1[1], posmerger2[1], 1)
def test11(self):
instance = Mikkola()
self.assertTrue(instance.parameters.evolve_to_exact_time)
instance.parameters.evolve_to_exact_time = False
self.assertFalse(instance.parameters.evolve_to_exact_time)
self.assertEqual(instance.parameters.maximum_number_of_particles, 100)
instance.parameters.maximum_number_of_particles = 200
self.assertEqual(instance.parameters.maximum_number_of_particles, 200)
self.assertAlmostRelativeEquals(instance.parameters.tolerance, 1e-13,
6)
instance.parameters.tolerance = 1e-10
self.assertAlmostRelativeEquals(instance.parameters.tolerance, 1e-10)
instance.stop()
def test11(self):
instance = Mikkola()
self.assertTrue(instance.parameters.evolve_to_exact_time)
instance.parameters.evolve_to_exact_time = False
self.assertFalse(instance.parameters.evolve_to_exact_time)
self.assertEquals(instance.parameters.maximum_number_of_particles, 100)
instance.parameters.maximum_number_of_particles = 200
self.assertEquals(instance.parameters.maximum_number_of_particles, 200)
self.assertAlmostRelativeEquals(instance.parameters.tolerance, 1e-13, 6)
instance.parameters.tolerance = 1e-10
self.assertAlmostRelativeEquals(instance.parameters.tolerance, 1e-10)
instance.stop()
