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 test5(self):
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(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
instance.particles.add_particles(stars)
instance.evolve_model(0.25 | units.yr)
self.assertEqual(instance.get_number_of_particles_added(), 1)
self.assertRaises(Exception, instance.get_id_of_added_particle, [2])
self.assertEqual(instance.get_id_of_added_particle(0), 4)
self.assertAlmostRelativeEquals(instance.get_mass(4), 2 | units.MSun)
self.assertAlmostRelativeEquals(instance.get_mass(1), 1 | units.MSun)
self.assertAlmostRelativeEquals(instance.get_mass(2), 1 | units.MSun)
pos1 = quantities.as_vector_quantity(instance.get_position(1))
pos2 = quantities.as_vector_quantity(instance.get_position(2))
posmerger = quantities.as_vector_quantity(instance.get_position(4))
print(pos1)
print(pos2)
print(posmerger)
distance = (pos1 - pos2).length()
self.assertTrue(distance < 1e-4 | units.RSun)
self.assertTrue(distance > 1e-6 | units.RSun)
distance = (pos1 - posmerger).length()
print(distance.as_quantity_in(units.RSun))
self.assertAlmostRelativeEquals(distance, 0.718548184852 | units.RSun,
5)
distance = (pos2 - posmerger).length()
print(distance.as_quantity_in(units.RSun))
self.assertAlmostRelativeEquals(distance, 0.718548184852 | units.RSun,
5)
instance.update_particle_set()
print(instance.particles[0].mass.as_quantity_in(units.MSun))
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 test5(self):
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(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
instance.particles.add_particles(stars)
instance.evolve_model(0.25 | units.yr)
self.assertEquals(instance.get_number_of_particles_added(), 1)
self.assertRaises(Exception, instance.get_id_of_added_particle, [2])
self.assertEquals(instance.get_id_of_added_particle(0), 4)
self.assertAlmostRelativeEquals(instance.get_mass(4), 2 | units.MSun)
self.assertAlmostRelativeEquals(instance.get_mass(1), 1 | units.MSun)
self.assertAlmostRelativeEquals(instance.get_mass(2), 1 | units.MSun)
pos1 = quantities.as_vector_quantity(instance.get_position(1))
pos2 = quantities.as_vector_quantity(instance.get_position(2))
posmerger = quantities.as_vector_quantity(instance.get_position(4))
print pos1
print pos2
print posmerger
distance = (pos1-pos2).length()
self.assertTrue(distance < 1e-4 | units.RSun)
self.assertTrue(distance > 1e-6 | units.RSun)
distance = (pos1-posmerger).length()
print distance.as_quantity_in(units.RSun)
self.assertAlmostRelativeEquals(distance,0.718548184852 | units.RSun, 5)
distance = (pos2-posmerger).length()
print distance.as_quantity_in(units.RSun)
self.assertAlmostRelativeEquals(distance,0.718548184852 | units.RSun, 5)
instance.update_particle_set()
print instance.particles[0].mass.as_quantity_in(units.MSun)
