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)