def test_hazard_curves_to_hazard_map_mean(self):
mean_haz_curves = models.HazardCurve.objects.filter(
output__oq_job=self.job,
statistics='mean')
with mock.patch('%s.compute_hazard_maps' % MOCK_PREFIX) as compute:
compute.return_value = self.MOCK_HAZARD_MAP
for curve in mean_haz_curves:
hazard_curves_to_hazard_map(
self.job.id, curve.id, self.TEST_POES)
hm_0_1, hm_0_02 = models.HazardMap.objects.filter(
output__oq_job=self.job,
statistics='mean').order_by('-poe')
self._test_maps(curve, hm_0_1, hm_0_02)
def test_hazard_curves_to_hazard_map_logic_tree(self):
lt_haz_curves = models.HazardCurve.objects.filter(
output__oq_job=self.job,
lt_realization__isnull=False)
with mock.patch('%s.compute_hazard_maps' % MOCK_PREFIX) as compute:
compute.return_value = self.MOCK_HAZARD_MAP
for curve in lt_haz_curves:
hazard_curves_to_hazard_map(
self.job.id, curve.id, self.TEST_POES)
lt_rlz = curve.lt_realization
# There should be two maps: 1 for each PoE
hm_0_1, hm_0_02 = models.HazardMap.objects.filter(
output__oq_job=self.job,
lt_realization=lt_rlz).order_by('-poe')
self._test_maps(curve, hm_0_1, hm_0_02, lt_rlz=lt_rlz)
def test_hazard_curves_to_hazard_map_quantile(self):
with mock.patch('%s.compute_hazard_maps' % MOCK_PREFIX) as compute:
compute.return_value = self.MOCK_HAZARD_MAP
for quantile in (0.1, 0.9):
quantile_haz_curves = models.HazardCurve.objects.filter(
output__oq_job=self.job,
imt__isnull=False,
statistics='quantile',
quantile=quantile)
for curve in quantile_haz_curves:
hazard_curves_to_hazard_map(
self.job.id, curve.id, self.TEST_POES)
hm_0_1, hm_0_02 = models.HazardMap.objects.filter(
output__oq_job=self.job,
statistics='quantile',
quantile=quantile).order_by('-poe')
self._test_maps(curve, hm_0_1, hm_0_02)
