def test_extinction_shapes():
pytest.importorskip("scipy")
for model in extinction_models:
# single value should work
extinction(1.e4 * u.angstrom, a_v=1., model=model)
# multiple values should return appropriate shape
assert extinction([1.e4] * u.angstrom, a_v=1., model=model).shape == (1,)
assert extinction([1.e4, 2.e4] * u.angstrom, a_v=1., model=model).shape == (2,)
def test_extinction_shapes():
pytest.importorskip("scipy")
for model in extinction_models:
# single value should work
extinction(1.e4 * u.angstrom, a_v=1., model=model)
# multiple values should return appropriate shape
assert extinction([1.e4] * u.angstrom, a_v=1.,
model=model).shape == (1, )
assert extinction([1.e4, 2.e4] * u.angstrom, a_v=1.,
model=model).shape == (2, )
def test_general_extinction_function(extinction_model_name):
pytest.importorskip('scipy')
specific_extinction_function = extinction_module.__getattribute__('extinction_{0}'.format(extinction_model_name))
wave = 5000 * u.angstrom
a_v = 1.
assert specific_extinction_function(wave, a_v) == extinction(wave, a_v, model=extinction_model_name)
def test_general_extinction_function(extinction_model_name):
pytest.importorskip('scipy')
specific_extinction_function = extinction_module.__getattribute__(
'extinction_{0}'.format(extinction_model_name))
wave = 5000 * u.angstrom
a_v = 1.
assert specific_extinction_function(wave, a_v) == extinction(
wave, a_v, model=extinction_model_name)
