diff --git a/tests/test_pipelines.py b/tests/test_pipelines.py index 91393d2b..9dc67b0d 100644 --- a/tests/test_pipelines.py +++ b/tests/test_pipelines.py @@ -66,6 +66,15 @@ "yBackground", } +# LATISS common inputs, in addition to COMMON_INPUTS +LATISS_INPUTS = { + "atlas_refcat2_20220201", + "defects", + "gaia_dr3_20230707", + "sky", + "transmission_atmosphere_fgcm", +} + # LSSTCam-imSim common inputs, in addition to COMMON_INPUTS LSSTCAM_IMSIM_INPUTS = { "bfk", @@ -126,6 +135,48 @@ "srcMatchFull", } +# LATISS common outputs, in addition to COMMON_OUTPUTS +LATISS_OUTPUTS = { + "diaObjectTable_tract", + "diaSourceTable", + "diaSourceTable_tract", + "fgcm_Cycle0_FitParameters", + "fgcm_Cycle1_FitParameters", + "fgcm_Cycle2_FitParameters", + "fgcm_Cycle3_FitParameters", + "fgcm_Cycle4_FitParameters", + "fgcm_Cycle0_FlaggedStars", + "fgcm_Cycle1_FlaggedStars", + "fgcm_Cycle2_FlaggedStars", + "fgcm_Cycle3_FlaggedStars", + "fgcm_Cycle4_FlaggedStars", + "fgcm_reference_stars", + "fgcm_star_ids", + "fgcm_star_observations", + "forcedSourceOnDiaObjectTable", + "forcedSourceOnDiaObjectTable_tract", + "forcedSourceTable", + "forcedSourceTable_tract", + "forced_diff", + "forced_diff_diaObject", + "forced_src_diaObject", + "goodSeeingCoadd", + "goodSeeingCoadd_nImage", + "goodSeeingDiff_assocDiaSrcTable", + "goodSeeingDiff_diaObjTable", + "goodSeeingDiff_diaSrc", + "goodSeeingDiff_diaSrcTable", + "goodSeeingDiff_differenceExp", + "goodSeeingDiff_differenceTempExp", + "goodSeeingDiff_fullDiaObjTable", + "goodSeeingDiff_matchedExp", + "goodSeeingDiff_templateExp", + "goodSeeingVisits", + "mergedForcedSource", + "mergedForcedSourceOnDiaObject", + "transmission_atmosphere_fgcm", +} + # LSSTCam-imSim common outputs, in addition to COMMON_OUTPUTS LSSTCAM_IMSIM_OUTPUTS = { "deepCoadd_measMatch", @@ -442,6 +493,32 @@ def test_hsc_drp_rc2_subset(self): ) tester.run(butler, self) + def test_latiss_drp(self): + butler = self.makeButler(writeable=True) + tester = PipelineStepTester( + os.path.join(PIPELINES_DIR, "LATISS", "DRP.yaml"), + [ + "#step1", + "#step2a", + "#step2bcde", + "#step3a", + "#step3b", + "#step3c", + "#step4", + "#step5", + "#step6", + ], + [ + ("ps1_pv3_3pi_20170110", {"htm7"}, "SimpleCatalog", False), + ("gaia_dr2_20200414", {"htm7"}, "SimpleCatalog", False), + ("gaia_dr3_20230707", {"htm7"}, "SimpleCatalog", False), + ("atlas_refcat2_20220201", {"htm7"}, "SimpleCatalog", False), + ], + expected_inputs=COMMON_INPUTS | LATISS_INPUTS | {"fgcmLookUpTable"}, + expected_outputs=COMMON_OUTPUTS | LATISS_OUTPUTS, + ) + tester.run(butler, self) + def test_lsstcam_imsim_drp_ci_imsim(self): butler = self.makeButler(writeable=True) tester = PipelineStepTester(