def test_classical_psha_based_job_mixes_in_properly(self):
with Mixin(self.job, RiskJobMixin, key="risk"):
self.assertTrue(RiskJobMixin in self.job.__class__.__bases__)
with Mixin(self.job, ClassicalPSHABasedMixin):
self.assertTrue(
ClassicalPSHABasedMixin in self.job.__class__.__bases__)
def test_job_mixes_in_properly(self):
with Mixin(self.job, RiskJobMixin, key="risk"):
self.assertTrue(RiskJobMixin in self.job.__class__.__bases__)
self.assertTrue(
ProbabilisticEventMixin in self.job.__class__.__bases__)
with Mixin(self.job, ProbabilisticEventMixin):
self.assertTrue(
ProbabilisticEventMixin in self.job.__class__.__bases__)
def launch(self):
""" Based on the behaviour specified in the configuration, mix in the
correct behaviour for the tasks and then execute them.
"""
output_dir = os.path.join(self.base_path, self['OUTPUT_DIR'])
if not os.path.exists(output_dir):
os.makedirs(output_dir)
results = []
self._partition()
for (key, mixin) in Mixin.ordered_mixins():
if key.upper() not in self.sections:
continue
with Mixin(self, mixin, key=key):
# The mixin defines a preload decorator to handle the needed
# data for the tasks and decorates _execute(). the mixin's
# _execute() method calls the expected tasks.
LOG.debug("Job %s Launching %s for %s" % (self.id, mixin, key))
results.extend(self.execute())
return results
