def test2(self):
convert_nbody = nbody_system.nbody_to_si(1.0 | units.MSun, 149.5e6 | units.km)
bhtree = BHTree(convert_nbody)
bhtree.initialize_code()
bhtree.eps2_for_gravity = 0.001
bhtree_particles = self.new_system_sun_and_earth()
bhtree.particles.add_particles(bhtree_particles)
if bhtree.legacy_interface.channel_type == 'mpi':
from mpi4py import MPI
if not MPI.Query_thread() == MPI.THREAD_MULTIPLE:
bhtree.stop()
self.skip("can only test parallel with multiple thread support in mpi implementation")
hermite = Hermite(convert_nbody)
hermite.dt_dia = 5000
hermite.commit_parameters()
hermite_particles = self.new_system_sun_and_earth()
hermite.particles.add_particles(hermite_particles)
thread1 = threading.Thread(target = self.evolve_model_unit_day, args = (bhtree, bhtree_particles, 10))
thread2 = threading.Thread(target = self.evolve_model_unit_day, args = (hermite, hermite_particles, 10))
thread1.start()
thread2.start()
thread1.join()
thread2.join()
if HAS_MATPLOTLIB:
figure = pyplot.figure()
plot = figure.add_subplot(1,1,1)
earth = bhtree_particles[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')
earth = hermite_particles[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 = "g", 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, "parallel-earth-sun.svg")
figure.savefig(output_file)
bhtree.stop()
hermite.stop()
bhtree.stop()
def test2(self):
convert_nbody = nbody_system.nbody_to_si(1.0 | units.MSun,
149.5e6 | units.km)
bhtree = BHTree(convert_nbody)
bhtree.initialize_code()
bhtree.eps2_for_gravity = 0.001
bhtree_particles = self.new_system_sun_and_earth()
bhtree.particles.add_particles(bhtree_particles)
if bhtree.legacy_interface.channel_type == 'mpi':
from mpi4py import MPI
if not MPI.Query_thread() == MPI.THREAD_MULTIPLE:
bhtree.stop()
self.skip(
"can only test parallel with multiple thread support in mpi implementation"
)
hermite = Hermite(convert_nbody)
hermite.dt_dia = 5000
hermite.commit_parameters()
hermite_particles = self.new_system_sun_and_earth()
hermite.particles.add_particles(hermite_particles)
thread1 = threading.Thread(target=self.evolve_model_unit_day,
args=(bhtree, bhtree_particles, 10))
thread2 = threading.Thread(target=self.evolve_model_unit_day,
args=(hermite, hermite_particles, 10))
thread1.start()
thread2.start()
thread1.join()
thread2.join()
if HAS_MATPLOTLIB:
figure = pyplot.figure()
plot = figure.add_subplot(1, 1, 1)
earth = bhtree_particles[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')
earth = hermite_particles[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="g", 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,
"parallel-earth-sun.svg")
figure.savefig(output_file)
bhtree.stop()
hermite.stop()
bhtree.stop()
