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(open( '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(open( '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 simulation_simple(config_verysimple, atomic_dataset): """ Instantiate SDEC plotter using a simple simulation model. Parameters ---------- config_verysimple : tardis.io.config_reader.Configuration Configuration object for a very simple simulation. atomic_dataset : str or tardis.atomic.AtomData Atomic data. Returns ------- sim: tardis.simulation.base.Simulation Simulation object. """ # Setup simulation configuration using config_verysimple and # override properties in such a way to make the simulation run faster config_verysimple.montecarlo.iterations = 3 config_verysimple.montecarlo.no_of_packets = 4000 config_verysimple.montecarlo.last_no_of_packets = -1 config_verysimple.spectrum.virtual.virtual_packet_logging = True config_verysimple.montecarlo.no_of_virtual_packets = 1 config_verysimple.spectrum.num = 2000 atomic_data = deepcopy(atomic_dataset) sim = run_tardis( config_verysimple, atom_data=atomic_data, show_convergence_plots=False, ) return sim
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(open("tardis/io/tests/data/tardis_configv1_verysimple.yml")) self.config_yaml["atom_data"] = self.atom_data_filename self.model = run_tardis(self.config_yaml)