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}'