From c544eb1fca4bcaefc1eb699f88ceae62741d14ae Mon Sep 17 00:00:00 2001 From: Josh Shields Date: Fri, 21 Jun 2024 11:40:21 -0400 Subject: [PATCH] fix test and units to output --- .../opacities/opacities_solvers/broadening.py | 10 +++++++++- .../opacities_solvers/tests/test_broadening.py | 9 +++++---- 2 files changed, 14 insertions(+), 5 deletions(-) diff --git a/stardis/radiation_field/opacities/opacities_solvers/broadening.py b/stardis/radiation_field/opacities/opacities_solvers/broadening.py index 05fe6c44..6dd2dc77 100644 --- a/stardis/radiation_field/opacities/opacities_solvers/broadening.py +++ b/stardis/radiation_field/opacities/opacities_solvers/broadening.py @@ -749,4 +749,12 @@ def rotation_broadening( if np.abs(v_rot) < 1e-5: return (wavelength, flux) - return wavelength, convolve1d(flux, rotational_profile / rotational_profile.sum()) + broadened_fluxes = ( + convolve1d(flux, rotational_profile / rotational_profile.sum()) + * u.erg + / u.s + / u.cm**2 + / u.Angstrom + ) + + return wavelength, broadened_fluxes diff --git a/stardis/radiation_field/opacities/opacities_solvers/tests/test_broadening.py b/stardis/radiation_field/opacities/opacities_solvers/tests/test_broadening.py index 28e0d462..2470d81f 100644 --- a/stardis/radiation_field/opacities/opacities_solvers/tests/test_broadening.py +++ b/stardis/radiation_field/opacities/opacities_solvers/tests/test_broadening.py @@ -616,10 +616,11 @@ def test_calc_gamma_van_der_waals_cuda_wrapped_sample_cuda_values( ) -def test_rotational_broadening(example_stellar_output): +def test_rotational_broadening(example_stardis_output): actual_wavelengths, actual_fluxes = rotation_broadening( 20 * u.km / u.s, - example_stellar_output.lambdas, - example_stellar_output.spectrum_lambda, + example_stardis_output.lambdas, + example_stardis_output.spectrum_lambda, + v_rot=0 * u.km / u.s, ) - assert np.allclose(actual_wavelengths, example_stellar_output.lambdas) + assert np.allclose(actual_fluxes, example_stardis_output.spectrum_lambda)