def test_pathloss_corrpars(rtdata):
"""Test PathLossStep using correction_pars"""
with dm.open(rtdata.get_data('nirspec/ifu/nrs1_flat_field.fits')) as data:
pls = PathLossStep()
corrected = pls.run(data)
pls.use_correction_pars = True
corrected_corrpars = pls.run(data)
assert np.allclose(corrected.data, corrected_corrpars.data, equal_nan=True)
def test_pathloss_corrpars(rtdata):
"""Test PathLossStep using correction_pars"""
with dm.open(rtdata.get_data('nirspec/fs/nrs1_flat_field.fits')) as data:
pls = PathLossStep()
corrected = pls.run(data)
pls.use_correction_pars = True
corrected_corrpars = pls.run(data)
bad_slits = []
for idx, slits in enumerate(zip(corrected.slits, corrected_corrpars.slits)):
corrected_slit, corrected_corrpars_slit = slits
if not np.allclose(corrected_slit.data, corrected_corrpars_slit.data, equal_nan=True):
bad_slits.append(idx)
assert not bad_slits, f'correction_pars failed for slits {bad_slits}'
def test_pathloss_corrpars(jail, rtdata_module):
"""Test PathLossStep using correction_pars"""
rtdata = rtdata_module
data = dm.open(rtdata.get_data('nirspec/mos/usf_wavecorr.fits'))
pls = PathLossStep()
corrected = pls.run(data)
pls.use_correction_pars = True
corrected_corrpars = pls.run(data)
bad_slits = []
for idx, slits in enumerate(zip(corrected.slits,
corrected_corrpars.slits)):
corrected_slit, corrected_corrpars_slit = slits
if not np.allclose(corrected_slit.data,
corrected_corrpars_slit.data,
equal_nan=True):
bad_slits.append(idx)
assert not bad_slits, f'correction_pars failed for slits {bad_slits}'
