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__
