Create better test for not creating a PackageAlerts task.

lsst · May 29, 2020 · 0f0b29b · 0f0b29b
1 parent 274e88f
commit 0f0b29b
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/tests/test_diaPipe.py b/tests/test_diaPipe.py
@@ -109,8 +109,7 @@ def testRunWithAlerts(self):
     def testRunWithoutAlerts(self):
         """Test running without creating and packaging alerts.
         """
-        with self.assertRaises(AttributeError):
-            self._testRun(False)
+        self._testRun(False)
 
     def _testRun(self, doPackageAlerts):
         """Test the normal workflow of each ap_pipe step.
@@ -132,7 +131,7 @@ def _testRun(self, doPackageAlerts):
             if doPackageAlerts:
                 subtasks.alertPackager.run.assert_called_once()
             else:
-                subtasks.alertPackager.run.assert_not_called()
+                self.assertFalse(hasattr(subtasks, 'alertPackager'))
             self.assertEqual(result.apdb_marker.db_url, "sqlite://")
             self.assertEqual(result.apdb_marker.isolation_level,
                              "READ_UNCOMMITTED")