def test_propagation_includes_pwv_transmission(self) -> None:
"""Test that the ``propagate`` method applies PWV absorption"""
# Get the expected transmission
wave = np.arange(4000, 5000)
transmission_model = pwv.PWVTransmissionModel(
resolution=self.propagation_effect._transmission_res)
transmission = transmission_model.calc_transmission(pwv=self.pwv,
wave=wave)
# Get the returned flux
propagated_flux = self.propagation_effect.propagate(wave,
np.ones_like(wave),
time=0)
np.testing.assert_equal(transmission, propagated_flux[0])
def setUp(self) -> None:
"""Create a dummy ``Transmission`` object"""
self.transmission = pwv.PWVTransmissionModel(4)