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)