def test_get_points(self):
read_model = species.ReadModel("ames-cond",
filter_name="Paranal/NACO.H")
points = read_model.get_points()
assert np.sum(points["teff"]) == 13500.0
assert np.sum(points["logg"]) == 28.0
def test_get_model(self):
read_model = species.ReadModel("ames-cond",
filter_name="Paranal/NACO.H")
model_box = read_model.get_model(self.model_param,
spec_res=100.0,
magnitude=False,
smooth=True)
assert np.sum(model_box.wavelength) == pytest.approx(92.26773310928259,
rel=self.limit,
abs=0.0)
assert np.sum(model_box.flux) == pytest.approx(1.6347074150483604e-12,
rel=self.limit,
abs=0.0)
model_box = read_model.get_model(self.model_param,
spec_res=100.0,
magnitude=True,
smooth=True)
assert np.sum(model_box.wavelength) == pytest.approx(92.26773310928259,
rel=self.limit,
abs=0.0)
assert np.sum(model_box.flux) == pytest.approx(650.0527540140317,
rel=self.limit,
abs=0.0)
def test_get_bounds(self):
read_model = species.ReadModel("ames-cond",
filter_name="Paranal/NACO.H")
bounds = read_model.get_bounds()
assert bounds["teff"] == (2000.0, 2500.0)
assert bounds["logg"] == (2.5, 5.5)
def test_get_points(self):
read_model = species.ReadModel('ames-cond',
filter_name='Paranal/NACO.H')
points = read_model.get_points()
assert np.sum(points['teff']) == 13500.
assert np.sum(points['logg']) == 28.
def test_get_bounds(self):
read_model = species.ReadModel('ames-cond',
filter_name='Paranal/NACO.H')
bounds = read_model.get_bounds()
assert bounds['teff'] == (2000., 2500.)
assert bounds['logg'] == (2.5, 5.5)
def test_get_flux(self):
read_model = species.ReadModel("ames-cond",
filter_name="Paranal/NACO.H")
flux = read_model.get_flux(self.model_param)
assert flux[0] == pytest.approx(3.33686187140077e-14,
rel=self.limit,
abs=0.0)
def test_get_wavelengths(self):
read_model = species.ReadModel("ames-cond",
filter_name="Paranal/NACO.H")
wavelengths = read_model.get_wavelengths()
assert np.sum(wavelengths) == pytest.approx(813.2224003071026,
rel=1e-7,
abs=0.0)
def test_get_flux(self):
read_model = species.ReadModel('ames-cond',
filter_name='Paranal/NACO.H')
flux = read_model.get_flux(self.model_param)
assert flux[0] == pytest.approx(3.336861738800749e-14,
rel=self.limit,
abs=0.)
def test_read_model(self):
database = species.Database()
database.add_model('ames-cond',
wavel_range=(1., 5.),
spec_res=100.,
teff_range=(2000., 2500))
read_model = species.ReadModel('ames-cond')
assert read_model.model == 'ames-cond'
def test_get_magnitude(self):
read_model = species.ReadModel("ames-cond",
filter_name="Paranal/NACO.H")
magnitude = read_model.get_magnitude(self.model_param)
assert magnitude[0] == pytest.approx(11.357135619661243,
rel=self.limit,
abs=0.0)
assert magnitude[1] == pytest.approx(11.357135619661243,
rel=self.limit,
abs=0.0)
def test_get_data(self):
read_model = species.ReadModel("ames-cond",
filter_name="Paranal/NACO.H")
model_box = read_model.get_data(self.model_param)
assert np.sum(model_box.wavelength) == pytest.approx(92.26773310928259,
rel=self.limit,
abs=0.0)
assert np.sum(model_box.flux) == pytest.approx(1.6346900092886714e-12,
rel=self.limit,
abs=0.0)
def test_read_model(self):
database = species.Database()
database.add_model(
"ames-cond",
wavel_range=(1.0, 5.0),
spec_res=100.0,
teff_range=(2000.0, 2500.0),
)
read_model = species.ReadModel("ames-cond")
assert read_model.model == "ames-cond"
def test_get_parameters(self):
read_model = species.ReadModel("ames-cond",
filter_name="Paranal/NACO.H")
parameters = read_model.get_parameters()
assert parameters == ["teff", "logg"]
def test_get_parameters(self):
read_model = species.ReadModel('ames-cond',
filter_name='Paranal/NACO.H')
parameters = read_model.get_parameters()
assert parameters == ['teff', 'logg']