def test_pre_execute(self):
# Simply tests that `pre_execute` calls `write_uh_spectra`.
# That's all for now.
calc = UHSCalculator(self.job_ctxt)
with helpers.patch(
'%s.write_uh_spectra' % self.UHS_CORE_MODULE) as write_mock:
calc.pre_execute()
self.assertEqual(1, write_mock.call_count)
def test_pre_execute(self):
# Simply tests that `pre_execute` calls `write_uh_spectra`.
# That's all for now.
calc = UHSCalculator(self.job_ctxt)
with helpers.patch(
'%s.write_uh_spectra' % self.UHS_CORE_MODULE) as write_mock:
calc.pre_execute()
self.assertEqual(1, write_mock.call_count)
