Add testGetNumberQuantaForTask unit test

lsst · Oct 11, 2022 · e483969 · e483969
1 parent f0489bb
commit e483969
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/doc/changes/DM-36145.feature.md b/doc/changes/DM-36145.feature.md
@@ -1 +1,3 @@
+* Add unit test to cover the new getNumberOfQuantaForTask method.
 * Add graph interface, getNumberOfQuantaForTask, to determine number of quanta associated with a given taskDef.
+* modifications to getQuantaForTask to support showing added additional quanta information in the logger.
diff --git a/tests/test_quantumGraph.py b/tests/test_quantumGraph.py
@@ -258,6 +258,10 @@ def testGetQuantaForTask(self):
         for task in self.tasks:
             self.assertEqual(self.qGraph.getQuantaForTask(task), self.quantumMap[task])
 
+    def testGetNumberOfQuantaForTask(self):
+        for task in self.tasks:
+            self.assertEqual(self.qGraph.getNumberOfQuantaForTask(task), len(self.quantumMap[task]))
+
     def testGetNodesForTask(self):
         for task in self.tasks:
             nodes: Iterable[QuantumNode] = self.qGraph.getNodesForTask(task)