def test6(self): convert_nbody = nbody_system.nbody_to_si(5.0 | units.kg, 10.0 | units.m) instance = BHTree(convert_nbody) instance.commit_parameters() indices = instance.new_particle( [15.0, 30.0] | units.kg, [10.0, 20.0] | units.m, [20.0, 40.0] | units.m, [30.0, 50.0] | units.m, #1.0 | units.m/units.s, 1.0 | units.m/units.s, 3.0 | units.m/units.s [0.0, 0.01] | units.m / units.s, [0.0, 0.01] | units.m / units.s, [0.0, 0.01] | units.m / units.s, [10.0, 20.0] | units.m) instance.commit_particles() self.assertEquals(instance.get_mass(indices[0]), 15.0 | units.kg) self.assertEquals(instance.get_mass(indices)[0], 15.0 | units.kg) self.assertRaises( AmuseException, instance.get_mass, [4, 5], expected_message= "Error when calling 'get_mass' of a 'BHTree', errorcode is -1") instance.cleanup_code() instance.stop()
def test7(self): convert_nbody = nbody_system.nbody_to_si(5.0 | units.kg, 10.0 | units.m) instance = BHTree(convert_nbody) instance.commit_parameters() particles = datamodel.Particles(2) self.assertEquals(len(instance.particles), 0) particles.mass = [15.0, 30.0] | units.kg particles.radius = [10.0, 20.0] | units.m particles.position = [[10.0, 20.0, 30.0], [20.0, 40.0, 60.0]] | units.m particles.velocity = [[0.0, 0.0, 0.0], [0.0, 0.0, 0.0] ] | units.m / units.s instance.particles.add_particles(particles) instance.commit_particles() self.assertEquals(instance.get_mass(1), 15.0 | units.kg) self.assertAlmostRelativeEquals( instance.get_position(1)[2], 30.0 | units.m) self.assertEquals(len(instance.particles), 2) self.assertAlmostRelativeEquals(instance.particles.mass[1], 30.0 | units.kg) self.assertAlmostRelativeEquals(instance.particles.position[1][2], 60.0 | units.m) instance.cleanup_code() instance.stop()
def test7(self): convert_nbody = nbody_system.nbody_to_si(5.0 | units.kg, 10.0 | units.m) instance = BHTree(convert_nbody) instance.commit_parameters() particles = datamodel.Particles(2) self.assertEquals(len(instance.particles), 0) particles.mass = [15.0, 30.0] | units.kg particles.radius = [10.0, 20.0] | units.m particles.position = [[10.0, 20.0, 30.0], [20.0, 40.0, 60.0]] | units.m particles.velocity = [[0.0, 0.0, 0.0], [0.0, 0.0, 0.0]] | units.m / units.s instance.particles.add_particles(particles) instance.commit_particles() self.assertEquals(instance.get_mass(1), 15.0| units.kg) self.assertAlmostRelativeEquals(instance.get_position(1)[2], 30.0| units.m) self.assertEquals(len(instance.particles), 2) self.assertAlmostRelativeEquals(instance.particles.mass[1], 30.0 | units.kg) self.assertAlmostRelativeEquals(instance.particles.position[1][2], 60.0 | units.m) instance.cleanup_code() instance.stop()
def test20(self): particles = datamodel.Particles(2) particles.x = [0.0, 10.0] | nbody_system.length particles.y = 0.0 | nbody_system.length particles.z = 0.0 | nbody_system.length particles.radius = 0.005 | nbody_system.length particles.vx = 0.0 | nbody_system.speed particles.vy = 0.0 | nbody_system.speed particles.vz = 0.0 | nbody_system.speed particles.mass = 1.0 | nbody_system.mass very_short_time_to_evolve = 1 | units.s very_long_time_to_evolve = 1e9 | nbody_system.time instance = BHTree() instance.initialize_code() instance.parameters.stopping_conditions_timeout = very_short_time_to_evolve self.assertEquals(instance.parameters.stopping_conditions_timeout, very_short_time_to_evolve) instance.parameters.epsilon_squared = (0.01 | nbody_system.length)**2 instance.particles.add_particles(particles) codeparticles1 = instance.particles instance.particles.add_particle( datamodel.Particle(position=[0, 1, 2] | nbody_system.length, velocity=[0, 0, 0] | nbody_system.speed, radius=0.005 | nbody_system.length, mass=1 | nbody_system.mass)) codeparticles2 = instance.particles self.assertTrue(codeparticles1 is codeparticles2) instance.cleanup_code() codeparticles3 = instance.particles self.assertFalse(codeparticles1 is codeparticles3) instance.stop()
def test20(self): particles = datamodel.Particles(2) particles.x = [0.0,10.0] | nbody_system.length particles.y = 0.0 | nbody_system.length particles.z = 0.0 | nbody_system.length particles.radius = 0.005 | nbody_system.length particles.vx = 0.0 | nbody_system.speed particles.vy = 0.0 | nbody_system.speed particles.vz = 0.0 | nbody_system.speed particles.mass = 1.0 | nbody_system.mass very_short_time_to_evolve = 1 | units.s very_long_time_to_evolve = 1e9 | nbody_system.time instance = BHTree() instance.initialize_code() instance.parameters.stopping_conditions_timeout = very_short_time_to_evolve self.assertEquals(instance.parameters.stopping_conditions_timeout, very_short_time_to_evolve) instance.parameters.epsilon_squared = (0.01 | nbody_system.length)**2 instance.particles.add_particles(particles) codeparticles1 = instance.particles instance.particles.add_particle(datamodel.Particle( position = [0,1,2] | nbody_system.length, velocity = [0,0,0] | nbody_system.speed, radius = 0.005 | nbody_system.length, mass = 1 | nbody_system.mass )) codeparticles2 = instance.particles self.assertTrue(codeparticles1 is codeparticles2) instance.cleanup_code() codeparticles3 = instance.particles self.assertFalse(codeparticles1 is codeparticles3) instance.stop()
def test2(self): #not completed convert_nbody = nbody_system.nbody_to_si(1.0 | units.MSun, 149.5e6 | units.km) instance = BHTree(convert_nbody) #instance.dt_dia = 1 instance.parameters.epsilon_squared = 0.001 | units.AU**2 #instance.timestep = 0.0001 #instance.use_self_gravity = 0 instance.commit_parameters() stars = datamodel.Stars(2) sun = stars[0] sun.mass = units.MSun(1.0) sun.position = units.m(numpy.array((0.0, 0.0, 0.0))) sun.velocity = units.ms(numpy.array((0.0, 0.0, 0.0))) sun.radius = units.RSun(1.0) earth = stars[1] earth.mass = units.kg(5.9736e24) earth.radius = units.km(6371) earth.position = units.km(numpy.array((149.5e6, 0.0, 0.0))) earth.velocity = units.ms(numpy.array((0.0, 29800, 0.0))) instance.particles.add_particles(stars) instance.commit_particles() self.assertAlmostRelativeEquals(sun.radius, instance.particles[0].radius) for x in range(1, 2000, 10): instance.evolve_model(x | units.day) instance.particles.copy_values_of_all_attributes_to(stars) stars.savepoint() if HAS_MATPLOTLIB: figure = pyplot.figure() plot = figure.add_subplot(1, 1, 1) x_points = earth.get_timeline_of_attribute("x") y_points = earth.get_timeline_of_attribute("y") x_points_in_AU = map(lambda (t, x): x.value_in(units.AU), x_points) y_points_in_AU = map(lambda (t, x): x.value_in(units.AU), y_points) plot.scatter(x_points_in_AU, y_points_in_AU, color="b", marker='o') plot.set_xlim(-1.5, 1.5) plot.set_ylim(-1.5, 1.5) test_results_path = self.get_path_to_results() output_file = os.path.join(test_results_path, "bhtree-earth-sun.svg") figure.savefig(output_file) instance.cleanup_code() instance.stop()
def test2(self): #not completed convert_nbody = nbody_system.nbody_to_si(1.0 | units.MSun, 149.5e6 | units.km) instance = BHTree(convert_nbody) #instance.dt_dia = 1 instance.parameters.epsilon_squared = 0.001 | units.AU**2 #instance.timestep = 0.0001 #instance.use_self_gravity = 0 instance.commit_parameters() stars = datamodel.Stars(2) sun = stars[0] sun.mass = units.MSun(1.0) sun.position = units.m(numpy.array((0.0,0.0,0.0))) sun.velocity = units.ms(numpy.array((0.0,0.0,0.0))) sun.radius = units.RSun(1.0) earth = stars[1] earth.mass = units.kg(5.9736e24) earth.radius = units.km(6371) earth.position = units.km(numpy.array((149.5e6,0.0,0.0))) earth.velocity = units.ms(numpy.array((0.0,29800,0.0))) instance.particles.add_particles(stars) instance.commit_particles() self.assertAlmostRelativeEquals(sun.radius, instance.particles[0].radius) for x in range(1,2000,10): instance.evolve_model(x | units.day) instance.particles.copy_values_of_all_attributes_to(stars) stars.savepoint() if HAS_MATPLOTLIB: figure = pyplot.figure() plot = figure.add_subplot(1,1,1) x_points = earth.get_timeline_of_attribute("x") y_points = earth.get_timeline_of_attribute("y") x_points_in_AU = map(lambda (t,x) : x.value_in(units.AU), x_points) y_points_in_AU = map(lambda (t,x) : x.value_in(units.AU), y_points) plot.scatter(x_points_in_AU,y_points_in_AU, color = "b", marker = 'o') plot.set_xlim(-1.5, 1.5) plot.set_ylim(-1.5, 1.5) test_results_path = self.get_path_to_results() output_file = os.path.join(test_results_path, "bhtree-earth-sun.svg") figure.savefig(output_file) instance.cleanup_code() instance.stop()
def test1(self): convert_nbody = nbody_system.nbody_to_si(1.0 | units.MSun, 149.5e6 | units.km) instance = BHTree(convert_nbody) instance.parameters.epsilon_squared = 0.001 | units.AU**2 stars = datamodel.Stars(2) sun = stars[0] sun.mass = units.MSun(1.0) sun.position = [0.0, 0.0, 0.0] | units.m sun.velocity = [0.0, 0.0, 0.0] | units.ms sun.radius = units.RSun(1.0) earth = stars[1] earth.mass = units.kg(5.9736e24) earth.radius = units.km(6371) earth.position = [149.5e6, 0.0, 0.0] | units.km earth.velocity = [0.0, 29800, 0.0] | units.ms #instance.particles.add_particles(stars) instance.particles.add_particles(stars) postion_at_start = earth.position.value_in(units.AU)[0] instance.evolve_model(365.0 | units.day) instance.particles.copy_values_of_all_attributes_to(stars) postion_after_full_rotation = earth.position.value_in(units.AU)[0] self.assertAlmostEqual(postion_at_start, postion_after_full_rotation, 3) instance.evolve_model(365.0 + (365.0 / 2) | units.day) instance.particles.copy_values_of_all_attributes_to(stars) postion_after_half_a_rotation = earth.position.value_in(units.AU)[0] self.assertAlmostEqual(-postion_at_start, postion_after_half_a_rotation, 2) instance.evolve_model(365.0 + (365.0 / 2) + (365.0 / 4) | units.day) instance.particles.copy_values_of_all_attributes_to(stars) postion_after_half_a_rotation = earth.position.value_in(units.AU)[1] self.assertAlmostEqual(-postion_at_start, postion_after_half_a_rotation, 1) instance.cleanup_code() instance.stop()
def test1(self): convert_nbody = nbody_system.nbody_to_si(1.0 | units.MSun, 149.5e6 | units.km) instance = BHTree(convert_nbody) instance.parameters.epsilon_squared = 0.001 | units.AU**2 stars = datamodel.Stars(2) sun = stars[0] sun.mass = units.MSun(1.0) sun.position = [0.0,0.0,0.0] | units.m sun.velocity = [0.0,0.0,0.0] | units.ms sun.radius = units.RSun(1.0) earth = stars[1] earth.mass = units.kg(5.9736e24) earth.radius = units.km(6371) earth.position = [149.5e6, 0.0, 0.0] | units.km earth.velocity = [0.0, 29800, 0.0] | units.ms #instance.particles.add_particles(stars) instance.particles.add_particles(stars) postion_at_start = earth.position.value_in(units.AU)[0] instance.evolve_model(365.0 | units.day) instance.particles.copy_values_of_all_attributes_to(stars) postion_after_full_rotation = earth.position.value_in(units.AU)[0] self.assertAlmostEqual(postion_at_start, postion_after_full_rotation, 3) instance.evolve_model(365.0 + (365.0 / 2) | units.day) instance.particles.copy_values_of_all_attributes_to(stars) postion_after_half_a_rotation = earth.position.value_in(units.AU)[0] self.assertAlmostEqual(-postion_at_start, postion_after_half_a_rotation, 2) instance.evolve_model(365.0 + (365.0 / 2) + (365.0 / 4) | units.day) instance.particles.copy_values_of_all_attributes_to(stars) postion_after_half_a_rotation = earth.position.value_in(units.AU)[1] self.assertAlmostEqual(-postion_at_start, postion_after_half_a_rotation, 1) instance.cleanup_code() instance.stop()
def test5(self): instance = BHTree() instance.commit_parameters() index = instance.new_particle( 15.0 | nbody_system.mass, 10.0 | nbody_system.length, 20.0 | nbody_system.length, 30.0 | nbody_system.length, 1.0 | nbody_system.speed, 1.0 | nbody_system.speed, 3.0 | nbody_system.speed, 10.0 | nbody_system.length ) instance.commit_particles() self.assertEquals(instance.get_mass(index), 15.0| nbody_system.mass) self.assertEquals(instance.get_radius(index), 10.0| nbody_system.length) instance.cleanup_code() instance.stop()
def test5(self): instance = BHTree() instance.commit_parameters() index = instance.new_particle( 15.0 | nbody_system.mass, 10.0 | nbody_system.length, 20.0 | nbody_system.length, 30.0 | nbody_system.length, 1.0 | nbody_system.speed, 1.0 | nbody_system.speed, 3.0 | nbody_system.speed, 10.0 | nbody_system.length) instance.commit_particles() self.assertEquals(instance.get_mass(index), 15.0 | nbody_system.mass) self.assertEquals(instance.get_radius(index), 10.0 | nbody_system.length) instance.cleanup_code() instance.stop()
def test4(self): convert_nbody = nbody_system.nbody_to_si(5.0 | units.kg, 10.0 | units.m) instance = BHTree(convert_nbody) instance.commit_parameters() index = instance.new_particle( 15.0 | units.kg, 10.0 | units.m, 20.0 | units.m, 30.0 | units.m, #1.0 | units.m/units.s, 1.0 | units.m/units.s, 3.0 | units.m/units.s 0.0 | units.m/units.s, 0.0 | units.m/units.s, 0.0 | units.m/units.s, 10.0 | units.m ) instance.commit_particles() self.assertEquals(instance.get_mass(index), 15.0| units.kg) self.assertEquals(instance.get_radius(index), 10.0| units.m) instance.cleanup_code() instance.stop()
def test9(self): instance = BHTree() instance.initialize_code() instance.parameters.epsilon_squared = 0.00001 | nbody_system.length**2 particles = datamodel.Particles(2) particles.mass = [1.0, 1.0] | nbody_system.mass particles.radius = [0.0001, 0.0001] | nbody_system.length particles.position = [[0.0, 0.0, 0.0], [2.0, 0.0, 0.0] ] | nbody_system.length particles.velocity = [[0.0, 0.0, 0.0], [0.0, 0.0, 0.0] ] | nbody_system.speed instance.particles.add_particles(particles) zero = 0.0 | nbody_system.length fx, fy, fz = instance.get_gravity_at_point(zero, 1.0 | nbody_system.length, zero, zero) self.assertAlmostEqual(fx, 0.0 | nbody_system.acceleration, 3) self.assertAlmostEqual(fy, 0.0 | nbody_system.acceleration, 3) self.assertAlmostEqual(fz, 0.0 | nbody_system.acceleration, 3) for x in (0.25, 0.5, 0.75): x0 = x | nbody_system.length x1 = (2.0 - x) | nbody_system.length potential0 = instance.get_potential_at_point(zero, x0, zero, zero) potential1 = instance.get_potential_at_point(zero, x1, zero, zero) fx0, fy0, fz0 = instance.get_gravity_at_point(zero, x0, zero, zero) fx1, fy1, fz1 = instance.get_gravity_at_point(zero, x1, zero, zero) self.assertAlmostEqual(fy0, 0.0 | nbody_system.acceleration, 3) self.assertAlmostEqual(fz0, 0.0 | nbody_system.acceleration, 3) self.assertAlmostEqual(fy1, 0.0 | nbody_system.acceleration, 3) self.assertAlmostEqual(fz1, 0.0 | nbody_system.acceleration, 3) self.assertAlmostEqual(fx0, -1.0 * fx1, 5) fx = (-1.0 / (x0**2) + 1.0 / (x1**2)) * (1.0 | nbody_system.length**3 / nbody_system.time**2) self.assertAlmostEqual(fx, fx0, 2) self.assertAlmostEqual(potential0, potential1, 5) instance.cleanup_code() instance.stop()
def test6(self): convert_nbody = nbody_system.nbody_to_si(5.0 | units.kg, 10.0 | units.m) instance = BHTree(convert_nbody) instance.commit_parameters() indices = instance.new_particle( [15.0, 30.0] | units.kg, [10.0, 20.0] | units.m, [20.0, 40.0] | units.m, [30.0, 50.0] | units.m, #1.0 | units.m/units.s, 1.0 | units.m/units.s, 3.0 | units.m/units.s [0.0, 0.01] | units.m/units.s, [0.0, 0.01] | units.m/units.s, [0.0, 0.01] | units.m/units.s, [10.0, 20.0] | units.m ) instance.commit_particles() self.assertEquals(instance.get_mass(indices[0]), 15.0| units.kg) self.assertEquals(instance.get_mass(indices)[0], 15.0| units.kg) self.assertRaises(AmuseException, instance.get_mass, [4,5], expected_message = "Error when calling 'get_mass' of a 'BHTree', errorcode is -1") instance.cleanup_code() instance.stop()
def test4(self): convert_nbody = nbody_system.nbody_to_si(5.0 | units.kg, 10.0 | units.m) instance = BHTree(convert_nbody) instance.commit_parameters() index = instance.new_particle( 15.0 | units.kg, 10.0 | units.m, 20.0 | units.m, 30.0 | units.m, #1.0 | units.m/units.s, 1.0 | units.m/units.s, 3.0 | units.m/units.s 0.0 | units.m / units.s, 0.0 | units.m / units.s, 0.0 | units.m / units.s, 10.0 | units.m) instance.commit_particles() self.assertEquals(instance.get_mass(index), 15.0 | units.kg) self.assertEquals(instance.get_radius(index), 10.0 | units.m) instance.cleanup_code() instance.stop()
def test9(self): instance = BHTree() instance.initialize_code() instance.parameters.epsilon_squared = 0.00001 | nbody_system.length**2 particles = datamodel.Particles(2) particles.mass = [1.0, 1.0] | nbody_system.mass particles.radius = [0.0001, 0.0001] | nbody_system.length particles.position = [[0.0,0.0,0.0], [2.0,0.0,0.0]] | nbody_system.length particles.velocity = [[0.0, 0.0, 0.0], [0.0, 0.0, 0.0]] | nbody_system.speed instance.particles.add_particles(particles) zero = 0.0 | nbody_system.length fx, fy, fz = instance.get_gravity_at_point(zero, 1.0 | nbody_system.length, zero, zero) self.assertAlmostEqual(fx, 0.0 | nbody_system.acceleration, 3) self.assertAlmostEqual(fy, 0.0 | nbody_system.acceleration, 3) self.assertAlmostEqual(fz, 0.0 | nbody_system.acceleration, 3) for x in (0.25, 0.5, 0.75): x0 = x | nbody_system.length x1 = (2.0 - x) | nbody_system.length potential0 = instance.get_potential_at_point(zero, x0, zero, zero) potential1 = instance.get_potential_at_point(zero, x1, zero, zero) fx0, fy0, fz0 = instance.get_gravity_at_point(zero, x0, zero, zero) fx1, fy1, fz1 = instance.get_gravity_at_point(zero, x1, zero, zero) self.assertAlmostEqual(fy0, 0.0 | nbody_system.acceleration, 3) self.assertAlmostEqual(fz0, 0.0 | nbody_system.acceleration, 3) self.assertAlmostEqual(fy1, 0.0 | nbody_system.acceleration, 3) self.assertAlmostEqual(fz1, 0.0 | nbody_system.acceleration, 3) self.assertAlmostEqual(fx0, -1.0 * fx1, 5) fx = (-1.0 / (x0**2) + 1.0 / (x1**2)) * (1.0 | nbody_system.length ** 3 / nbody_system.time ** 2) self.assertAlmostEqual(fx, fx0, 2) self.assertAlmostEqual(potential0, potential1, 5) instance.cleanup_code() instance.stop()
def test3(self): convert_nbody = nbody_system.nbody_to_si(1.0 | units.MSun, 149.5e6 | units.km) instance = BHTree(convert_nbody) #instance.dt_dia = 1 instance.parameters.epsilon_squared = 0.001 | units.AU**2 #instance.timestep = 0.0001 #instance.use_self_gravity = 0 instance.commit_parameters() stars = datamodel.Stars(2) star1 = stars[0] star2 = stars[1] star1.mass = units.MSun(1.0) star1.position = units.AU(numpy.array((-.10, 0.0, 0.0))) star1.velocity = units.AUd(numpy.array((0.0, 0.0, 0.0))) star1.radius = units.RSun(1.0) star2.mass = units.MSun(1.0) star2.position = units.AU(numpy.array((.10, 0.0, 0.0))) star2.velocity = units.AUd(numpy.array((0.0, 0.0, 0.0))) star2.radius = units.RSun(100.0) instance.particles.add_particles(stars) instance.commit_particles() for x in range(1, 200, 1): instance.evolve_model(x | units.day) instance.particles.copy_values_of_all_attributes_to(stars) #instance.get_indices_of_colliding_particles() #print stars[0].position-stars[1].position stars.savepoint() instance.cleanup_code() instance.stop()
def test3(self): convert_nbody = nbody_system.nbody_to_si(1.0 | units.MSun, 149.5e6 | units.km) instance = BHTree(convert_nbody) #instance.dt_dia = 1 instance.parameters.epsilon_squared = 0.001 | units.AU**2 #instance.timestep = 0.0001 #instance.use_self_gravity = 0 instance.commit_parameters() stars = datamodel.Stars(2) star1 = stars[0] star2 = stars[1] star1.mass = units.MSun(1.0) star1.position = units.AU(numpy.array((-.10,0.0,0.0))) star1.velocity = units.AUd(numpy.array((0.0,0.0,0.0))) star1.radius = units.RSun(1.0) star2.mass = units.MSun(1.0) star2.position = units.AU(numpy.array((.10,0.0,0.0))) star2.velocity = units.AUd(numpy.array((0.0,0.0,0.0))) star2.radius = units.RSun(100.0) instance.particles.add_particles(stars) instance.commit_particles() for x in range(1,200,1): instance.evolve_model(x | units.day) instance.particles.copy_values_of_all_attributes_to(stars) #instance.get_indices_of_colliding_particles() #print stars[0].position-stars[1].position stars.savepoint() instance.cleanup_code() instance.stop()