def setup(self):
self.atom_data_filename = os.path.expanduser(os.path.expandvars(
pytest.config.getvalue('atomic-dataset')))
assert os.path.exists(self.atom_data_filename), ("{0} atomic datafiles"
" does not seem to "
"exist".format(
self.atom_data_filename))
self.config_yaml = yaml_load_config_file(
'tardis/plasma/tests/data/plasma_test_config_lte.yml')
self.config_yaml['atom_data'] = self.atom_data_filename
self.lte_model = run_tardis(self.config_yaml)
self.config_yaml = yaml_load_config_file(
'tardis/plasma/tests/data/plasma_test_config_nlte.yml')
self.config_yaml['atom_data'] = self.atom_data_filename
self.nlte_model = run_tardis(self.config_yaml)
def setup(self):
self.atom_data_filename = os.path.expanduser(
os.path.expandvars(pytest.config.getvalue('atomic-dataset')))
assert os.path.exists(
self.atom_data_filename), ("{0} atomic datafiles"
" does not seem to "
"exist".format(self.atom_data_filename))
self.config_yaml = yaml_load_config_file(
'tardis/plasma/tests/data/plasma_test_config_lte.yml')
self.config_yaml['atom_data'] = self.atom_data_filename
self.lte_model = run_tardis(self.config_yaml)
self.config_yaml = yaml_load_config_file(
'tardis/plasma/tests/data/plasma_test_config_nlte.yml')
self.config_yaml['atom_data'] = self.atom_data_filename
self.nlte_model = run_tardis(self.config_yaml)
def test_ascii_reader_exponential_law():
yaml_data = yaml_load_config_file(
data_path('tardis_configv1_density_exponential_test.yml'))
#for later use
density_data = yaml_data['model']['structure']['density']
t_explosion = density_data['time_0']
rho_0 = density_data['rho_0']
v0 = density_data['v_0']
v_inner = yaml_data['model']['structure']['velocity']['start']
v_outer = yaml_data['model']['structure']['velocity']['stop']
my_conf = config_reader.Configuration.from_yaml(
data_path('tardis_configv1_density_exponential_test.yml'),
test_parser=True)
structure = my_conf['structure']
expected_densites = [
5.18114795e-14, 4.45945537e-14, 3.83828881e-14, 3.30364579e-14,
2.84347428e-14, 2.44740100e-14, 2.10649756e-14, 1.81307925e-14,
1.56053177e-14, 1.34316215e-14, 1.15607037e-14, 9.95038990e-15,
8.56437996e-15, 7.37143014e-15, 6.34464872e-15, 5.46088976e-15,
4.70023138e-15, 4.04552664e-15, 3.48201705e-15, 2.99699985e-15
]
expected_unit = 'g / (cm3)'
assert structure['no_of_shells'] == 20
for i, mdens in enumerate(expected_densites):
assert_almost_equal(structure['mean_densities'][i].value, mdens)
assert structure['mean_densities'][i].unit == u.Unit(expected_unit)
def test_last_no_of_packets():
yaml_data = yaml_load_config_file(data_path('paper1_tardis_configv1.yml'))
del yaml_data['montecarlo']['last_no_of_packets']
config = config_reader.Configuration.from_config_dict(
yaml_data, test_parser=True, config_dirname=data_path(''))
assert (config.montecarlo.last_no_of_packets ==
config.montecarlo.no_of_packets)
def test_ascii_reader_power_law():
yaml_data = yaml_load_config_file(
data_path('tardis_configv1_density_power_law_test.yml'))
#for later use
density_data = yaml_data['model']['structure']['density']
t_explosion = density_data['time_0']
rho_0 = density_data['rho_0']
exponent = density_data['exponent']
v_inner = yaml_data['model']['structure']['velocity']['start']
v_outer = yaml_data['model']['structure']['velocity']['stop']
my_conf = config_reader.Configuration.from_yaml(data_path('tardis_configv1_density_power_law_test.yml'),
test_parser=True)
structure = my_conf['structure']
expected_densites = [3.29072513e-14, 2.70357804e-14, 2.23776573e-14,
1.86501954e-14, 1.56435277e-14, 1.32001689e-14, 1.12007560e-14,
9.55397475e-15, 8.18935779e-15, 7.05208050e-15, 6.09916083e-15,
5.29665772e-15, 4.61758699e-15, 4.04035750e-15, 3.54758837e-15,
3.12520752e-15, 2.76175961e-15, 2.44787115e-15, 2.17583442e-15,
1.93928168e-15]
assert structure['no_of_shells'] == 20
for i, mdens in enumerate(expected_densites):
assert_almost_equal(structure['mean_densities'][i].to(
u.Unit('g / (cm3)')).value, mdens)
def test_ascii_reader_power_law():
yaml_data = yaml_load_config_file(
data_path('tardis_configv1_density_power_law_test.yml'))
#for later use
density_data = yaml_data['model']['structure']['density']
t_explosion = density_data['time_0']
rho_0 = density_data['rho_0']
exponent = density_data['exponent']
v_inner = yaml_data['model']['structure']['velocity']['start']
v_outer = yaml_data['model']['structure']['velocity']['stop']
my_conf = config_reader.Configuration.from_yaml(
data_path('tardis_configv1_density_power_law_test.yml'),
test_parser=True)
structure = my_conf['structure']
expected_densites = [
3.29072513e-14, 2.70357804e-14, 2.23776573e-14, 1.86501954e-14,
1.56435277e-14, 1.32001689e-14, 1.12007560e-14, 9.55397475e-15,
8.18935779e-15, 7.05208050e-15, 6.09916083e-15, 5.29665772e-15,
4.61758699e-15, 4.04035750e-15, 3.54758837e-15, 3.12520752e-15,
2.76175961e-15, 2.44787115e-15, 2.17583442e-15, 1.93928168e-15
]
assert structure['no_of_shells'] == 20
for i, mdens in enumerate(expected_densites):
assert_almost_equal(
structure['mean_densities'][i].to(u.Unit('g / (cm3)')).value,
mdens)
def test_last_no_of_packets():
yaml_data = yaml_load_config_file(data_path('paper1_tardis_configv1.yml'))
del yaml_data['montecarlo']['last_no_of_packets']
config = config_reader.Configuration.from_config_dict(yaml_data,
test_parser=True,
config_dirname=data_path(''))
assert (config.montecarlo.last_no_of_packets ==
config.montecarlo.no_of_packets)
def setup(self):
#general parsing of the paper config
filename = 'tardis_configv1_uniform_density.yml'
self.yaml_data = yaml_load_config_file(data_path(filename))
self.config = config_reader.Configuration.from_config_dict(
self.yaml_data, test_parser=True, config_dirname=data_path(''))
def setup(self):
#general parsing of the paper config
filename = 'tardis_configv1_uniform_density.yml'
self.yaml_data = yaml_load_config_file(data_path(filename))
self.config = config_reader.Configuration.from_config_dict(self.yaml_data,
test_parser=True,
config_dirname=data_path(''))
def setup(self):
#general parsing of the paper config
filename = 'tardis_configv1_uniform_density.yml'
self.yaml_data = yaml_load_config_file(data_path(filename))
self.yaml_data['plasma']['initial_t_inner'] = quantity_from_str("2508 K")
self.config = config_reader.Configuration.from_config_dict(self.yaml_data,
test_parser=True,
config_dirname=data_path(''))
def setup(self):
self.atom_data_filename = os.path.expanduser(os.path.expandvars(
pytest.config.getvalue('atomic-dataset')))
assert os.path.exists(self.atom_data_filename), ("{0} atomic datafiles"
" does not seem to "
"exist".format(
self.atom_data_filename))
self.config_yaml = yaml_load_config_file(
'tardis/plasma/tests/data/plasma_test_config_lte.yml')
self.config_yaml['atom_data'] = self.atom_data_filename
conf = Configuration.from_config_dict(self.config_yaml)
self.lte_simulation = Simulation.from_config(conf)
self.lte_simulation.run()
self.config_yaml = yaml_load_config_file(
'tardis/plasma/tests/data/plasma_test_config_nlte.yml')
self.config_yaml['atom_data'] = self.atom_data_filename
conf = Configuration.from_config_dict(self.config_yaml)
self.nlte_simulation = Simulation.from_config(conf)
self.nlte_simulation.run()
def setup(self):
#general parsing of the paper config
filename = 'tardis_configv1_uniform_density.yml'
self.yaml_data = yaml_load_config_file(data_path(filename))
self.yaml_data['plasma']['initial_t_inner'] = quantity_from_str(
"2508 K")
self.config = config_reader.Configuration.from_config_dict(
self.yaml_data, test_parser=True, config_dirname=data_path(''))
def setup(self):
#general parsing of the paper config
filename = 'tardis_configv1_ascii_density_abund.yml'
self.yaml_data = yaml_load_config_file(data_path(filename))
self.yaml_data['model']['structure']['filename'] = 'density.dat'
self.yaml_data['model']['abundances']['filename'] = 'abund.dat'
self.config = config_reader.Configuration.from_config_dict(
self.yaml_data, test_parser=True, config_dirname=data_path(''))
def setup(self):
#general parsing of the paper config
filename = 'tardis_configv1_ascii_density_abund.yml'
self.yaml_data = yaml_load_config_file(data_path(filename))
self.yaml_data['model']['structure']['filename'] = 'density.dat'
self.yaml_data['model']['abundances']['filename'] = 'abund.dat'
self.config = config_reader.Configuration.from_config_dict(self.yaml_data,
test_parser=True,
config_dirname=data_path(''))
def setup(self):
#general parsing of the paper config
filename = 'tardis_configv1_artis_density_v_slice.yml'
self.yaml_data = yaml_load_config_file(data_path(filename))
self.yaml_data['model']['abundances'] = {'type': 'file',
'filename': 'artis_abundances.dat',
'filetype': 'artis'}
self.config = config_reader.Configuration.from_config_dict(self.yaml_data,
test_parser=True,
config_dirname=data_path(''))
def setup(self):
#general parsing of the paper config
filename = 'tardis_configv1_artis_density_v_slice.yml'
self.yaml_data = yaml_load_config_file(data_path(filename))
self.yaml_data['model']['abundances'] = {
'type': 'file',
'filename': 'artis_abundances.dat',
'filetype': 'artis'
}
self.config = config_reader.Configuration.from_config_dict(
self.yaml_data, test_parser=True, config_dirname=data_path(''))
def setup(self):
self.atom_data_filename = os.path.expanduser(
os.path.expandvars(pytest.config.getvalue('atomic-dataset')))
assert os.path.exists(
self.atom_data_filename), ("{0} atomic datafiles"
" does not seem to "
"exist".format(self.atom_data_filename))
self.config_yaml = yaml_load_config_file(
'tardis/io/tests/data/tardis_configv1_verysimple.yml')
self.config_yaml['atom_data'] = self.atom_data_filename
tardis_config = Configuration.from_config_dict(self.config_yaml)
self.simulation = Simulation.from_config(tardis_config)
self.simulation.run()
def setup(self):
self.atom_data_filename = os.path.expanduser(os.path.expandvars(
pytest.config.getvalue('atomic-dataset')))
assert os.path.exists(self.atom_data_filename), ("{0} atomic datafiles"
" does not seem to "
"exist".format(
self.atom_data_filename))
self.config_yaml = yaml_load_config_file(
'tardis/io/tests/data/tardis_configv1_verysimple.yml')
self.config_yaml['atom_data'] = self.atom_data_filename
tardis_config = Configuration.from_config_dict(self.config_yaml)
self.simulation = Simulation.from_config(tardis_config)
self.simulation.run()
def test_ascii_reader_exponential_law():
yaml_data = yaml_load_config_file(
data_path('tardis_configv1_density_exponential_test.yml'))
#for later use
density_data = yaml_data['model']['structure']['density']
t_explosion = density_data['time_0']
rho_0 = density_data['rho_0']
v0 = density_data['v_0']
v_inner = yaml_data['model']['structure']['velocity']['start']
v_outer = yaml_data['model']['structure']['velocity']['stop']
my_conf = config_reader.Configuration.from_yaml(data_path('tardis_configv1_density_exponential_test.yml'),
test_parser=True)
structure = my_conf['structure']
expected_densites = [5.18114795e-14, 4.45945537e-14, 3.83828881e-14, 3.30364579e-14, 2.84347428e-14, 2.44740100e-14, 2.10649756e-14, 1.81307925e-14, 1.56053177e-14, 1.34316215e-14, 1.15607037e-14, 9.95038990e-15, 8.56437996e-15, 7.37143014e-15, 6.34464872e-15, 5.46088976e-15, 4.70023138e-15, 4.04552664e-15, 3.48201705e-15, 2.99699985e-15]
expected_unit = 'g / (cm3)'
assert structure['no_of_shells'] == 20
for i, mdens in enumerate(expected_densites):
assert_almost_equal(structure['mean_densities'][i].value,mdens)
assert structure['mean_densities'][i].unit == u.Unit(expected_unit)
def setup(self):
#general parsing of the paper config
filename = 'tardis_configv1_artis_density.yml'
self.config = config_reader.Configuration.from_yaml(
data_path(filename), test_parser=True)
self.yaml_data = yaml_load_config_file(data_path(filename))
def setup(self):
#general parsing of the paper config
filename = 'tardis_configv1_artis_density.yml'
self.config = config_reader.Configuration.from_yaml(data_path(filename),
test_parser=True)
self.yaml_data = yaml_load_config_file(data_path(filename))
def tardis_config_verysimple():
return yaml_load_config_file(
"tardis/io/tests/data/tardis_configv1_verysimple.yml"
)
def tardis_config_verysimple():
return yaml_load_config_file(
'tardis/io/tests/data/tardis_configv1_verysimple.yml')
def setup(self):
#general parsing of the paper config
self.config = config_reader.Configuration.from_yaml(
data_path('paper1_tardis_configv1.yml'), test_parser=True)
self.yaml_data = yaml_load_config_file(
data_path('paper1_tardis_configv1.yml'))
def setup(self):
#general parsing of the paper config
self.config = config_reader.Configuration.from_yaml(data_path('paper1_tardis_configv1.yml'),
test_parser=True)
self.yaml_data = yaml_load_config_file(
data_path('paper1_tardis_configv1.yml'))
