def test_post_execute(self):
calc = UHSCalculator(self.job_ctxt)
expected_call_args = ((self.job_id,), {})
with helpers.patch(
'openquake.utils.stats.delete_job_counters') as del_mock:
calc.post_execute()
self.assertEqual(1, del_mock.call_count)
self.assertEqual(expected_call_args, del_mock.call_args)
def test_post_execute(self):
calc = UHSCalculator(self.job_ctxt)
expected_call_args = ((self.job_id,), {})
with helpers.patch(
'openquake.utils.stats.delete_job_counters') as del_mock:
calc.post_execute()
self.assertEqual(1, del_mock.call_count)
self.assertEqual(expected_call_args, del_mock.call_args)
