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'))