Update tests to support doWriteAssociatedSources=True

lsst · Mar 13, 2024 · 7ee01af · 7ee01af
1 parent d40764c
commit 7ee01af
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests/test_testPipeline.py b/tests/test_testPipeline.py
@@ -128,6 +128,8 @@ def setUpClass(cls):
         butlerTests.addDatasetType(cls.repo, "deepDiff_assocDiaSrc", cls.visitId.dimensions, "DataFrame")
         butlerTests.addDatasetType(cls.repo, "deepDiff_longTrailedSrc", cls.visitId.dimensions, "DataFrame")
         butlerTests.addDatasetType(cls.repo, "deepRealBogusSources", cls.visitId.dimensions, "Catalog")
+        butlerTests.addDatasetType(cls.repo, "deepDiff_diaForcedSrc", cls.visitId.dimensions, "DataFrame")
+        butlerTests.addDatasetType(cls.repo, "deepDiff_diaObject", cls.visitId.dimensions, "DataFrame")
 
     def setUp(self):
         super().setUp()
@@ -324,6 +326,8 @@ def testMockDiaPipelineTask(self):
              "apdbMarker": self.visitId,
              "associatedDiaSources": self.visitId,
              "longTrailedSources": self.visitId,
+             "diaForcedSources": self.visitId,
+             "diaObjects": self.visitId,
              })
         pipelineTests.runTestQuantum(task, self.butler, quantum, mockRun=False)