def test_change_all_hazard_pass(self): """Test _change_all_hazard method""" meas = Measure() meas.hazard_set = HAZ_DEMO_H5 ref_haz = Hazard.from_hdf5(HAZ_DEMO_H5) hazard = Hazard('TC') new_haz = meas._change_all_hazard(hazard) self.assertEqual(new_haz.tag.file_name, ref_haz.tag.file_name) self.assertEqual(new_haz.tag.haz_type, ref_haz.tag.haz_type) self.assertTrue(np.array_equal(new_haz.frequency, ref_haz.frequency)) self.assertTrue(np.array_equal(new_haz.date, ref_haz.date)) self.assertTrue(np.array_equal(new_haz.orig, ref_haz.orig)) self.assertTrue( np.array_equal(new_haz.centroids.coord, ref_haz.centroids.coord)) self.assertTrue( np.array_equal(new_haz.intensity.data, ref_haz.intensity.data)) self.assertTrue( np.array_equal(new_haz.fraction.data, ref_haz.fraction.data))
def _change_all_hazard(self, hazard): """ Change hazard to provided hazard_set. Parameters ---------- hazard : climada.hazard.Hazard hazard instance Returns ------- new_haz : climada.hazard.Hazard Hazard """ if self.hazard_set == NULL_STR: return hazard LOGGER.debug('Setting new hazard %s', self.hazard_set) from climada.hazard.base import Hazard new_haz = Hazard.from_hdf5(self.hazard_set) new_haz.check() return new_haz