Add longTrailedSources to testPipeline unit tests

lsst · Nov 16, 2023 · 8773aeb · 8773aeb
1 parent 60239a5
commit 8773aeb
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/python/lsst/ap/verify/testPipeline.py b/python/lsst/ap/verify/testPipeline.py
@@ -568,4 +568,5 @@ def run(self,
                       associatedDiaSources=pandas.DataFrame(),
                       diaForcedSources=pandas.DataFrame(),
                       diaObjects=pandas.DataFrame(),
+                      longTrailedSources=pandas.DataFrame(),
                       )
diff --git a/tests/test_testPipeline.py b/tests/test_testPipeline.py
@@ -120,7 +120,8 @@ def setUpClass(cls):
         butlerTests.addDatasetType(cls.repo, "deepDiff_diaSrcTable", cls.visitId.keys(), "DataFrame")
         butlerTests.addDatasetType(cls.repo, "visitSsObjects", cls.visitOnlyId.keys(), "DataFrame")
         butlerTests.addDatasetType(cls.repo, "apdb_marker", cls.visitId.keys(), "Config")
-        butlerTests.addDatasetType(cls.repo, "deepDiff_associDiaSrc", cls.visitId.keys(), "DataFrame")
+        butlerTests.addDatasetType(cls.repo, "deepDiff_assocDiaSrc", cls.visitId.keys(), "DataFrame")
+        butlerTests.addDatasetType(cls.repo, "deepDiff_longTrailedSrc", cls.visitId.keys(), "DataFrame")
 
     def setUp(self):
         super().setUp()
@@ -285,6 +286,7 @@ def testMockDiaPipelineTask(self):
              "template": self.visitId,
              "apdbMarker": self.visitId,
              "associatedDiaSources": self.visitId,
+             "longTrailedSources": self.visitId,
              })
         pipelineTests.runTestQuantum(task, self.butler, quantum, mockRun=False)