def test_template_file_pass(self): """Read template excel file""" imp_funcs = ImpactFuncSet() imp_funcs.read_excel(ENT_TEMPLATE_XLS) # Check some results self.assertEqual(len(imp_funcs._data), 10) self.assertEqual(len(imp_funcs._data['TC'][3].paa), 9) self.assertEqual(len(imp_funcs._data['EQ'][1].intensity), 14) self.assertEqual(len(imp_funcs._data['HS'][1].mdd), 16)
def test_impact_funcs_pass(self): """Plot diferent impact functions.""" myfuncs = ImpactFuncSet() myfuncs.read_excel(ENT_DEMO_TODAY) myax = myfuncs.plot() self.assertEqual(2, len(myax)) self.assertIn('TC 1: Tropical cyclone default', \ myax[0].title.get_text()) self.assertIn('TC 3: TC Building code', myax[1].title.get_text())
def test_write_read_pass(self): """Write + read excel file""" imp_funcs = ImpactFuncSet() imp_funcs.tag.file_name = 'No file name' imp_funcs.tag.description = 'test writer' imp1 = ImpactFunc() imp1.id = 1 imp1.name = 'code 1' imp1.intensity_unit = 'm/s' imp1.haz_type = 'TC' imp1.intensity = np.arange(100) imp1.mdd = np.arange(100) * 0.5 imp1.paa = np.ones(100) imp_funcs.append(imp1) imp2 = ImpactFunc() imp2.id = 2 imp2.name = 'code 2' imp2.intensity_unit = 'm/s' imp2.haz_type = 'TC' imp2.intensity = np.arange(102) imp2.mdd = np.arange(102) * 0.25 imp2.paa = np.ones(102) imp_funcs.append(imp2) imp3 = ImpactFunc() imp3.id = 1 imp3.name = 'code 1' imp3.intensity_unit = 'm' imp3.haz_type = 'FL' imp3.intensity = np.arange(86) imp3.mdd = np.arange(86) * 0.15 imp3.paa = np.ones(86) imp_funcs.append(imp3) imp4 = ImpactFunc() imp4.id = 15 imp4.name = 'code 15' imp4.intensity_unit = 'K' imp4.haz_type = 'DR' imp4.intensity = np.arange(5) imp4.mdd = np.arange(5) imp4.paa = np.ones(5) imp_funcs.append(imp4) file_name = CONFIG.impact_funcs.test_data.dir().joinpath('test_write.xlsx') imp_funcs.write_excel(file_name) imp_res = ImpactFuncSet() imp_res.read_excel(file_name) self.assertEqual(imp_res.tag.file_name, str(file_name)) self.assertEqual(imp_res.tag.description, '') # first function for fun_haz, fun_dict in imp_res.get_func().items(): for fun_id, fun in fun_dict.items(): if fun_haz == 'TC' and fun_id == 1: ref_fun = imp1 elif fun_haz == 'TC' and fun_id == 2: ref_fun = imp2 elif fun_haz == 'FL' and fun_id == 1: ref_fun = imp3 elif fun_haz == 'DR' and fun_id == 15: ref_fun = imp4 else: self.assertEqual(1, 0) self.assertEqual(ref_fun.haz_type, fun.haz_type) self.assertEqual(ref_fun.id, fun.id) self.assertEqual(ref_fun.name, fun.name) self.assertEqual(ref_fun.intensity_unit, fun.intensity_unit) self.assertTrue(np.allclose(ref_fun.intensity, fun.intensity)) self.assertTrue(np.allclose(ref_fun.mdd, fun.mdd)) self.assertTrue(np.allclose(ref_fun.paa, fun.paa))
def test_demo_file_pass(self): """Read demo excel file""" # Read demo excel file imp_funcs = ImpactFuncSet() description = 'One single file.' imp_funcs.read_excel(ENT_DEMO_TODAY, description) # Check results n_funcs = 2 hazard = 'TC' first_id = 1 second_id = 3 self.assertEqual(len(imp_funcs._data), 1) self.assertEqual(len(imp_funcs._data[hazard]), n_funcs) # first function self.assertEqual(imp_funcs._data[hazard][first_id].id, 1) self.assertEqual(imp_funcs._data[hazard][first_id].name, 'Tropical cyclone default') self.assertEqual(imp_funcs._data[hazard][first_id].intensity_unit, 'm/s') self.assertEqual(imp_funcs._data[hazard][first_id].intensity.shape, (9,)) self.assertEqual(imp_funcs._data[hazard][first_id].intensity[0], 0) self.assertEqual(imp_funcs._data[hazard][first_id].intensity[1], 20) self.assertEqual(imp_funcs._data[hazard][first_id].intensity[2], 30) self.assertEqual(imp_funcs._data[hazard][first_id].intensity[3], 40) self.assertEqual(imp_funcs._data[hazard][first_id].intensity[4], 50) self.assertEqual(imp_funcs._data[hazard][first_id].intensity[5], 60) self.assertEqual(imp_funcs._data[hazard][first_id].intensity[6], 70) self.assertEqual(imp_funcs._data[hazard][first_id].intensity[7], 80) self.assertEqual(imp_funcs._data[hazard][first_id].intensity[8], 100) self.assertEqual(imp_funcs._data[hazard][first_id].mdd.shape, (9,)) self.assertEqual(imp_funcs._data[hazard][first_id].mdd[0], 0) self.assertEqual(imp_funcs._data[hazard][first_id].mdd[8], 0.41079600) self.assertEqual(imp_funcs._data[hazard][first_id].paa.shape, (9,)) self.assertEqual(imp_funcs._data[hazard][first_id].paa[0], 0) self.assertEqual(imp_funcs._data[hazard][first_id].paa[8], 1) # second function self.assertEqual(imp_funcs._data[hazard][second_id].id, 3) self.assertEqual(imp_funcs._data[hazard][second_id].name, 'TC Building code') self.assertEqual(imp_funcs._data[hazard][first_id].intensity_unit, 'm/s') self.assertEqual(imp_funcs._data[hazard][second_id].intensity.shape, (9,)) self.assertEqual(imp_funcs._data[hazard][second_id].intensity[0], 0) self.assertEqual(imp_funcs._data[hazard][second_id].intensity[1], 20) self.assertEqual(imp_funcs._data[hazard][second_id].intensity[2], 30) self.assertEqual(imp_funcs._data[hazard][second_id].intensity[3], 40) self.assertEqual(imp_funcs._data[hazard][second_id].intensity[4], 50) self.assertEqual(imp_funcs._data[hazard][second_id].intensity[5], 60) self.assertEqual(imp_funcs._data[hazard][second_id].intensity[6], 70) self.assertEqual(imp_funcs._data[hazard][second_id].intensity[7], 80) self.assertEqual(imp_funcs._data[hazard][second_id].intensity[8], 100) self.assertEqual(imp_funcs._data[hazard][second_id].mdd.shape, (9,)) self.assertEqual(imp_funcs._data[hazard][second_id].mdd[0], 0) self.assertEqual(imp_funcs._data[hazard][second_id].mdd[8], 0.4) self.assertEqual(imp_funcs._data[hazard][second_id].paa.shape, (9,)) self.assertEqual(imp_funcs._data[hazard][second_id].paa[0], 0) self.assertEqual(imp_funcs._data[hazard][second_id].paa[8], 1) # general information self.assertEqual(imp_funcs.tag.file_name, str(ENT_DEMO_TODAY)) self.assertEqual(imp_funcs.tag.description, description)
class Entity(object): """Collects exposures, impact functions, measures and discount rates. Default values set when empty constructor. Attributes: exposures (Exposures): exposures impact_funcs (ImpactFucs): impact functions measures (MeasureSet): measures disc_rates (DiscRates): discount rates def_file (str): Default file from configuration file """ def __init__(self): """ Empty initializator """ self.exposures = Exposures() self.disc_rates = DiscRates() self.impact_funcs = ImpactFuncSet() self.measures = MeasureSet() def read_mat(self, file_name, description=''): """Read MATLAB file of climada. Parameters: file_name (str, optional): file name(s) or folder name containing the files to read description (str or list(str), optional): one description of the data or a description of each data file Raises: ValueError """ self.exposures = Exposures() self.exposures.read_mat(file_name) self.disc_rates = DiscRates() self.disc_rates.read_mat(file_name, description) self.impact_funcs = ImpactFuncSet() self.impact_funcs.read_mat(file_name, description) self.measures = MeasureSet() self.measures.read_mat(file_name, description) def read_excel(self, file_name, description=''): """Read csv or xls or xlsx file following climada's template. Parameters: file_name (str, optional): file name(s) or folder name containing the files to read description (str or list(str), optional): one description of the data or a description of each data file Raises: ValueError """ self.exposures = Exposures(pd.read_excel(file_name)) self.exposures.tag = Tag() self.exposures.tag.file_name = file_name self.exposures.tag.description = description self.disc_rates = DiscRates() self.disc_rates.read_excel(file_name, description) self.impact_funcs = ImpactFuncSet() self.impact_funcs.read_excel(file_name, description) self.measures = MeasureSet() self.measures.read_excel(file_name, description) def write_excel(self, file_name): """ Write excel file following template. """ self.exposures.to_excel(file_name) self.impact_funcs.write_excel(file_name) self.measures.write_excel(file_name) self.disc_rates.write_excel(file_name) def check(self): """Check instance attributes. Raises: ValueError """ self.disc_rates.check() self.exposures.check() self.impact_funcs.check() self.measures.check() def __setattr__(self, name, value): """Check input type before set""" if name == "exposures": if not isinstance(value, Exposures): LOGGER.error("Input value is not (sub)class of Exposures.") raise ValueError elif name == "impact_funcs": if not isinstance(value, ImpactFuncSet): LOGGER.error("Input value is not (sub)class of ImpactFuncSet.") raise ValueError elif name == "measures": if not isinstance(value, MeasureSet): LOGGER.error("Input value is not (sub)class of MeasureSet.") raise ValueError elif name == "disc_rates": if not isinstance(value, DiscRates): LOGGER.error("Input value is not (sub)class of DiscRates.") raise ValueError super().__setattr__(name, value) def __str__(self): return 'Exposures: \n' + self.exposures.tag.__str__() + \ '\nDiscRates: \n' + self.disc_rates.tag.__str__() + \ '\nImpactFuncSet: \n' + self.impact_funcs.tag.__str__() + \ '\nMeasureSet: \n' + self.measures.tag.__str__() __repr__ = __str__