def add_content(self, more_content, no_docstring=False):
if not is_mpd_running():
return
try:
cls = self.object
instance = cls(must_start_worker = False, must_handle_state = False)
try:
#instance.initialize_code()
parameter_documentation = self.get_sphinx_doc_for_parameters(instance.parameters)
finally:
instance.stop()
except Exception as ex:
print(ex)
return
if self.analyzer:
# prevent encoding errors when the file name is non-ASCII
filename = str(self.analyzer.srcname)
sourcename = '%s:docstring of %s' % (filename, self.fullname)
else:
sourcename = 'docstring of %s' % self.fullname
encoding = self.analyzer # and self.analyzer.encoding
lines = prepare_docstring(force_decode(parameter_documentation, encoding))
for i, line in enumerate(self.process_doc([lines,])):
self.add_line(line, sourcename, i)
def slowtest_evrard_gadget():
assert is_mpd_running()
run_evrard([new_code("gadget")],
10000,
random_seed=12345,
name_of_the_figure=os.path.join(get_path_to_results(),
"evrard_test_gadget_10000.png"))
def setUp(self):
TestCase.setUp(self)
from amuse.rfi.core import is_mpd_running
self.assertTrue(
is_mpd_running(),
"MPICH2 mpd deamon process not running, cannot run this test as it requires MPI"
)
def add_content(self, more_content, no_docstring=False):
if not is_mpd_running():
return
try:
cls = self.object
instance = cls(must_start_worker = False, must_handle_state = False)
try:
#instance.initialize_code()
parameter_documentation = self.get_sphinx_doc_for_parameters(instance.parameters)
finally:
instance.stop()
except Exception as ex:
print ex
return
if self.analyzer:
# prevent encoding errors when the file name is non-ASCII
filename = unicode(self.analyzer.srcname,
sys.getfilesystemencoding(), 'replace')
sourcename = u'%s:docstring of %s' % (filename, self.fullname)
else:
sourcename = u'docstring of %s' % self.fullname
encoding = self.analyzer and self.analyzer.encoding
lines = prepare_docstring(force_decode(parameter_documentation, encoding))
for i, line in enumerate(self.process_doc([lines,])):
self.add_line(line, sourcename, i)
def slowtest_evrard_gadget():
assert is_mpd_running()
run_evrard(
[new_code("gadget")],
10000,
random_seed = 12345,
name_of_the_figure = os.path.join(get_path_to_results(), "evrard_test_gadget_10000.png")
)
def test_simulate_small_cluster():
"""test_simulate_small_cluster
This method is found by the testing framework and automatically
run with all other tests. This method simulates
a too small cluster, this is done to limit the testing time.
"""
assert is_mpd_running()
simulate_small_cluster(4, 4 | units.Myr)
def slowtest_evrard_compare():
assert is_mpd_running()
global labels
labels = fi_gadget_labels
run_evrard([new_code("fi"), new_code("gadget")],
10000,
random_seed=12345,
name_of_the_figure=os.path.join(
get_path_to_results(), "evrard_test_compare_10000.png"))
def test_simulate_one_star():
assert is_mpd_running()
code = new_code("sse", 1)
test_results_path = get_path_to_results()
output_file = os.path.join(test_results_path, "HR_evolution_tracks.png")
simulate_evolution_tracks(
code,
[20.0] | units.MSun,
name_of_the_figure=output_file,
)
def test_simulate_one_star():
assert is_mpd_running()
code = new_code("sse", 1)
test_results_path = get_path_to_results()
output_file = os.path.join(test_results_path, "HR_evolution_tracks.png")
simulate_evolution_tracks(
code,
[20.0] | units.MSun,
name_of_the_figure=output_file,
)
def slowtest_evrard_compare():
assert is_mpd_running()
global labels
labels = fi_gadget_labels
run_evrard(
[new_code("fi"), new_code("gadget")],
10000,
random_seed = 12345,
name_of_the_figure = os.path.join(get_path_to_results(), "evrard_test_compare_10000.png")
)
def test_simulate_small_cluster():
"""test_simulate_small_cluster
This method is found by the testing framework and automatically
run with all other tests. This method simulates
a too small cluster, this is done to limit the testing time.
"""
assert is_mpd_running()
test_results_path = get_path_to_results()
output_file = os.path.join(test_results_path, "test-2.svg")
simulate_small_cluster(4, 4 | units.Myr, name_of_the_figure=output_file)
def test_simulate_small_cluster():
"""test_simulate_small_cluster
This method is found by the testing framework and automatically
run with all other tests. This method simulates
a too small cluster, this is done to limit the testing time.
"""
assert is_mpd_running()
test_results_path = get_path_to_results()
output_file = os.path.join(test_results_path, "test-2.svg")
simulate_small_cluster(4, 4 | units.Myr,
name_of_the_figure=output_file)
def test_simulate_short():
assert is_mpd_running()
code = new_code("sse", 100)
test_results_path = get_path_to_results()
output_file = os.path.join(test_results_path, "cluster_HR_diagram.png")
simulate_stellar_evolution(code,
number_of_stars=100,
end_time=2.0 | units.Myr,
name_of_the_figure=output_file)
def test_simulate_short():
assert is_mpd_running()
code = new_code("sse", 100)
test_results_path = get_path_to_results()
output_file = os.path.join(test_results_path, "cluster_HR_diagram.png")
simulate_stellar_evolution(
code,
number_of_stars=100,
end_time=2.0 | units.Myr,
name_of_the_figure=output_file
)
def main():
assert is_mpd_running()
seed = None
nstars = 128
if len(sys.argv) > 1:
stars = int(sys.argv[1])
with_units = len(sys.argv) > 2
if not with_units:
mass_unit = nbody_system.mass
length_unit = nbody_system.length
else:
mass_unit = units.MSun
length_unit = units.parsec
m_min = 0.1 | mass_unit
m_max = 100 | mass_unit
alpha = -2.35
r_vir = 1 | length_unit
masses = new_salpeter_mass_distribution(nstars, m_min, m_max, alpha)
m_tot = masses.sum()
if not with_units:
convert_nbody = None
masses /= m_tot.value_in(nbody_system.mass) # scale to unit mass
m_tot = 1 | nbody_system.mass
else:
convert_nbody = nbody_system.nbody_to_si(m_tot, r_vir)
convert_nbody.set_as_default()
print m_tot
stars = new_plummer_model(nstars, convert_nbody, random_state=seed)
stars.mass = masses
LagrangianRadii(stars, verbose=True)
def test_cloud():
if not is_mpd_running():
return
run_cloud(1000)
def test_evrard():
if not is_mpd_running():
return
run_evrard(64)
default=None,
dest="cacheDir",
help="Use/write cache from directory"
)
result.add_option(
"-p",
"--plot_file",
type="string",
default="HR_evolution_tracks.png",
dest="plot_filename",
help="Name of the file to plot to"
)
return result
if __name__ == '__main__':
if not is_mpd_running():
print "There is no mpd server running. Please do 'mpd &' first."
sys.exit()
parser = new_commandline_option_parser()
(options, arguments) = parser.parse_args()
if arguments:
parser.error("unknown arguments '{0}'".format(arguments))
mass_list = [0.5, 1.0, 2.0, 5.0, 10.0, 20.0, 30.0] | units.MSun
code = new_code(options.code, len(mass_list))
if not (options.cacheDir is None):
print "Using cache directory: %s" % (options.cacheDir)
# As a special case, we use caching of the underlying models instead of the model output for EVtwin2SSE
if (options.code == "evtwin2sse"):
code.cache_underlying_models(options.cacheDir)
else:
infile = a
elif o == "-g":
use_gpu = 0
elif o == "-G":
use_gpu = 0
gpu_worker = 0
elif o == "-i":
gpu_id = int(a)
elif o == "-n":
N = int(a)
elif o == "-s":
random_seed = int(a)
elif o == "-t":
t_end = float(a) | nbody_system.time
elif o == "-w":
n_workers = int(a)
else:
print "unexpected argument", o
if random_seed <= 0:
numpy.random.seed()
random_seed = numpy.random.randint(1, pow(2, 31) - 1)
numpy.random.seed(random_seed)
print "random seed =", random_seed
#os.system('env')
assert is_mpd_running()
run_ph4(infile, N, t_end, delta_t, n_workers, use_gpu, gpu_worker, gpu_id,
accuracy_parameter, softening_length, manage_encounters)
def test_cloud():
if not is_mpd_running():
return
run_cloud(1000)
def setUp(self):
TestCase.setUp(self)
from amuse.rfi.core import is_mpd_running
self.assertTrue(is_mpd_running(), "MPICH2 mpd deamon process not running, cannot run this test as it requires MPI")
def test_evrard():
if not is_mpd_running():
return
run_evrard(64)
dest="cacheDir",
help="Use/write cache from directory"
)
result.add_option(
"-p",
"--plot_file",
type="string",
default="HR_evolution_tracks.png",
dest="plot_filename",
help="Name of the file to plot to"
)
return result
if __name__ == '__main__':
if not is_mpd_running():
print("There is no mpd server running. Please do 'mpd &' first.")
sys.exit()
parser = new_commandline_option_parser()
(options, arguments) = parser.parse_args()
if arguments:
parser.error("unknown arguments '{0}'".format(arguments))
mass_list = [0.5, 1.0, 2.0, 5.0, 10.0, 20.0, 30.0] | units.MSun
code = new_code(options.code, len(mass_list))
if not (options.cacheDir is None):
print("Using cache directory: %s" % (options.cacheDir))
# As a special case, we use caching of the underlying models instead of
# the model output for EVtwin2SSE
if (options.code == "evtwin2sse"):
code.cache_underlying_models(options.cacheDir)
0)
kep.stop()
if __name__ == '__main__':
mass = 1 | nbody_system.mass
semi = 1 | nbody_system.length
ecc = 0.5
time = 5.0 | nbody_system.time
try:
opts, args = getopt.getopt(sys.argv[1:], "a:e:m:t:")
except getopt.GetoptError, err:
print str(err)
sys.exit(1)
for o, a in opts:
if o == "-a":
semi = float(a) | nbody_system.length
elif o == "-e":
ecc = float(a)
elif o == "-m":
time = float(a) | nbody_system.mass
elif o == "-t":
time = float(a) | nbody_system.time
else:
print "unexpected argument", o
assert is_mpd_running()
run_kepler(mass, semi, ecc, time)
