diff --git a/tests/extractor/test_scrape.py b/tests/extractor/test_scrape.py index 8fe1ecc..a8d54bb 100644 --- a/tests/extractor/test_scrape.py +++ b/tests/extractor/test_scrape.py @@ -48,8 +48,10 @@ 'FILTER': 'F150W', 'PUPIL': 'CLEAR', 'GRATING': 'N/A', + 'FXD_SLIT': 'NaN', 'EXP_TYPE': 'NRC_IMAGE', 'CHANNEL': 'SHORT', + 'BAND': 'NaN', 'SUBARRAY': 'FULL', 'NUMDTHPT': 5, 'GS_RA': 339.0885699921705, @@ -78,7 +80,9 @@ 'PUPIL': 'N/A', 'GRATING': 'N/A', 'EXP_TYPE': 'MIR_IMAGE', + 'FXD_SLIT': 'NaN', 'CHANNEL': 'N/A', + 'BAND': 'NaN', 'SUBARRAY': 'FULL', 'NUMDTHPT': 2, 'GS_RA': 339.061734363325, @@ -106,8 +110,10 @@ 'FILTER': 'CLEAR', 'PUPIL': 'F150W', 'GRATING': 'N/A', + 'FXD_SLIT': 'NaN', 'EXP_TYPE': 'NIS_IMAGE', 'CHANNEL': 'NONE', + 'BAND': 'NaN', 'SUBARRAY': 'FULL', 'NUMDTHPT': 4, 'GS_RA': 80.3457270042323, diff --git a/tests/preprocessor/test_scrub.py b/tests/preprocessor/test_scrub.py index 4cb1028..b3fe99c 100644 --- a/tests/preprocessor/test_scrub.py +++ b/tests/preprocessor/test_scrub.py @@ -131,7 +131,7 @@ def test_jwst_cal_scrubber(jwstcal_input_path): assert len(scrubber.imgpix) == 3 imgpix_products = list(scrubber.imgpix.keys()) for product in imgpix_products: - assert len(scrubber.imgpix[product].keys()) == 46 + assert len(scrubber.imgpix[product].keys()) == 48 image_inputs = scrubber.scrub_inputs(exp_type="IMAGE") assert len(image_inputs) == 3 assert list(image_inputs.columns) == JWST_SCRUBBED_COLS