Skip to content

Commit

Permalink
Merge pull request #130 from lsst/tickets/DM-44749
Browse files Browse the repository at this point in the history
DM-44749: Test LATISS pipeline in unit tests
  • Loading branch information
arunkannawadi committed Jun 11, 2024
2 parents efba1c4 + 9a1b435 commit 00ab748
Showing 1 changed file with 83 additions and 2 deletions.
85 changes: 83 additions & 2 deletions tests/test_pipelines.py
Original file line number Diff line number Diff line change
Expand Up @@ -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",
Expand All @@ -87,8 +96,6 @@
"deepCoadd_forced_src",
"deepCoadd_inputMap",
"deepCoadd_meas",
"deepCoadd_measMatch",
"deepCoadd_measMatchFull",
"deepCoadd_mergeDet",
"deepCoadd_nImage",
"deepCoadd_obj",
Expand Down Expand Up @@ -118,6 +125,8 @@
HSC_OUTPUTS = {
"calexp_skyCorr_visit_mosaic",
"calexpBackground_skyCorr_visit_mosaic",
"deepCoadd_measMatch",
"deepCoadd_measMatchFull",
"forced_src",
"preSource",
"preSourceTable",
Expand All @@ -126,8 +135,52 @@
"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",
"deepCoadd_measMatchFull",
"diaObjectTable_tract",
"diaSourceTable",
"diaSourceTable_tract",
Expand Down Expand Up @@ -219,6 +272,8 @@ def test_decam_drp_merian(self):
},
expected_outputs=COMMON_OUTPUTS
| {
"deepCoadd_measMatch",
"deepCoadd_measMatchFull",
"goodSeeingCoadd",
"goodSeeingCoadd_nImage",
"goodSeeingVisits",
Expand Down Expand Up @@ -438,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(
Expand Down

0 comments on commit 00ab748

Please sign in to comment.