def test7(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.assertEqual(instance.get_id_of_added_particle(0), 4) child1, child2 = instance.get_children_of_particle([4]) self.assertEqual(child1[0], instance.particles[0]) self.assertEqual(child2[0], instance.particles[1]) for i in range(3): child1, child2 = instance.get_children_of_particle([i + 1]) self.assertEqual(child1[0], None) self.assertEqual(child2[0], None)
def test13(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(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) self.assertEqual(instance.get_number_of_particles_added(), 1) self.assertEqual(instance.get_id_of_added_particle(0), 3) instance.update_particle_set() child1, child2 = instance.get_children_of_particle([3]) self.assertEqual(child1[0], instance.particles[0]) self.assertEqual(child2[0], instance.particles[1]) self.assertAlmostRelativeEquals(instance.get_mass(3), 2 | units.MSun)
def test13(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(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) self.assertEquals(instance.get_number_of_particles_added(), 1) self.assertEquals(instance.get_id_of_added_particle(0), 3) instance.update_particle_set() child1, child2 = instance.get_children_of_particle([3]) self.assertEquals(child1[0], instance.particles[0]) self.assertEquals(child2[0], instance.particles[1]) self.assertAlmostRelativeEquals(instance.get_mass(3), 2 | units.MSun)
def test12(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(4) 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 = 1.0 | units.MSun stars[2].radius = 1.0 | units.RSun stars[2].position = [2, 0, 0] | units.RSun stars[2].velocity = [-0.1, 0.0, 0.0] | units.km / units.s stars[3].mass = 0.01 | units.MSun stars[3].radius = 1.0 | units.RSun stars[3].position = [6000, 0, 0] | units.RSun stars[3].velocity = [0.0, -10, 0.0] | units.km / units.s instance.particles.add_particles(stars) instance.evolve_model(0.000001 | units.yr) instance.update_particle_set() self.assertEqual(instance.get_number_of_particles_added(), 1) self.assertEqual(instance.get_id_of_added_particle(0), 5) instance.evolve_model(0.30 | units.yr) self.assertEqual(instance.get_number_of_particles_added(), 1) self.assertEqual(instance.get_id_of_added_particle(0), 6) instance.update_particle_set() child1, child2 = instance.get_children_of_particle([5]) self.assertEqual(child1[0], instance.particles[0]) self.assertEqual(child2[0], instance.particles[1]) child1, child2 = instance.get_children_of_particle([6]) self.assertEqual(child1[0], instance.particles[2]) self.assertEqual(child2[0], instance.particles[4])
def test12(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(4) 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 = 1.0 | units.MSun stars[2].radius = 1.0| units.RSun stars[2].position = [2, 0, 0] | units.RSun stars[2].velocity = [-0.1, 0.0, 0.0] | units.km/units.s stars[3].mass = 0.01 | units.MSun stars[3].radius = 1.0| units.RSun stars[3].position = [6000, 0, 0] | units.RSun stars[3].velocity = [0.0, -10, 0.0] | units.km/units.s instance.particles.add_particles(stars) instance.evolve_model(0.000001 | units.yr) instance.update_particle_set() self.assertEquals(instance.get_number_of_particles_added(), 1) self.assertEquals(instance.get_id_of_added_particle(0), 5) instance.evolve_model(0.30 | units.yr) self.assertEquals(instance.get_number_of_particles_added(), 1) self.assertEquals(instance.get_id_of_added_particle(0), 6) instance.update_particle_set() child1, child2 = instance.get_children_of_particle([5]) self.assertEquals(child1[0], instance.particles[0]) self.assertEquals(child2[0], instance.particles[1]) child1, child2 = instance.get_children_of_particle([6]) self.assertEquals(child1[0], instance.particles[2]) self.assertEquals(child2[0], instance.particles[4])
def test7(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.assertEquals(instance.get_id_of_added_particle(0), 4) child1, child2 = instance.get_children_of_particle([4]) self.assertEquals(child1[0], instance.particles[0]) self.assertEquals(child2[0], instance.particles[1]) for i in range(3): child1, child2 = instance.get_children_of_particle([i+1]) self.assertEquals(child1[0], None) self.assertEquals(child2[0], None)