def test_fill_econ_indicators_na_pass(self):
"""Test fill_econ_indicators with '' inputs."""
ref_year = 2019
country_isos = {
'CHE': [1, 'Switzerland', 'che_geom'],
'ZMB': [2, 'Zambia', 'zmb_geom']
}
gdp = {'CHE': 1.2 * 1e20, 'ZMB': ''}
inc_grp = {'CHE': '', 'ZMB': 4}
kwargs = {'gdp': gdp, 'inc_grp': inc_grp}
fill_econ_indicators(ref_year, country_isos, SHP_FILE, **kwargs)
# GDP might change if worldbank input data changes,
# check magnitude and adjust value in country_isos_ref if test fails
country_isos_ref = {
'CHE': [1, 'Switzerland', 'che_geom', 2019, gdp['CHE'], 4],
'ZMB':
[2, 'Zambia', 'zmb_geom', 2019, 23309773922, inc_grp['ZMB']]
}
self.assertEqual(country_isos.keys(), country_isos_ref.keys())
for country in country_isos_ref.keys():
for i in [0, 1, 2, 3, 5]: # test elements one by one:
self.assertEqual(country_isos[country][i],
country_isos_ref[country][i])
self.assertAlmostEqual(country_isos[country][4] * 1e-6,
country_isos_ref[country][4] * 1e-6,
places=0)
def test_fill_econ_indicators_pass(self):
""" Test fill_econ_indicators CHE, ZMB."""
ref_year = 2015
country_isos = {
'CHE': [1, 'Switzerland', 'che_geom'],
'ZMB': [2, 'Zambia', 'zmb_geom']
}
fill_econ_indicators(ref_year, country_isos, SHP_FILE)
country_isos_ref = {
'CHE': [1, 'Switzerland', 'che_geom', 2015, 679832291693.257, 4],
'ZMB': [2, 'Zambia', 'zmb_geom', 2015, 21154394545.895, 2]
}
self.assertEqual(country_isos, country_isos_ref)
def test_fill_econ_indicators_pass(self):
""" Test fill_econ_indicators CHE, ZMB."""
ref_year = 2015
country_isos = {
'CHE': [1, 'Switzerland', 'che_geom'],
'ZMB': [2, 'Zambia', 'zmb_geom']
}
fill_econ_indicators(ref_year, country_isos, SHP_FILE)
country_isos_ref = {
'CHE': [1, 'Switzerland', 'che_geom', 2015, 679832391757.542, 4],
'ZMB': [2, 'Zambia', 'zmb_geom', 2015, 21243350632.5008, 2]
}
self.assertEqual(country_isos, country_isos_ref)
def test_fill_econ_indicators_kwargs_pass(self):
"""Test fill_econ_indicators with kwargs inputs."""
ref_year = 2015
country_isos = {'CHE': [1, 'Switzerland', 'che_geom'],
'ZMB': [2, 'Zambia', 'zmb_geom']
}
gdp = {'CHE': 1.2, 'ZMB': 1.3}
inc_grp = {'CHE': 3, 'ZMB': 4}
kwargs = {'gdp': gdp, 'inc_grp': inc_grp}
fill_econ_indicators(ref_year, country_isos, SHP_FILE, **kwargs)
country_isos_ref = {
'CHE': [1, 'Switzerland', 'che_geom', 2015, gdp['CHE'], inc_grp['CHE']],
'ZMB': [2, 'Zambia', 'zmb_geom', 2015, gdp['ZMB'], inc_grp['ZMB']]
}
self.assertEqual(country_isos, country_isos_ref)
def test_fill_econ_indicators_pass(self):
"""Test fill_econ_indicators CHE, ZMB."""
ref_year = 2015
country_isos = {'CHE': [1, 'Switzerland', 'che_geom'],
'ZMB': [2, 'Zambia', 'zmb_geom']
}
fill_econ_indicators(ref_year, country_isos, SHP_FILE)
country_isos_ref = {'CHE': [1, 'Switzerland', 'che_geom', 2015, 679832291693, 4],
'ZMB': [2, 'Zambia', 'zmb_geom', 2015, 21243347377, 2]
}
self.assertEqual(country_isos.keys(), country_isos_ref.keys())
for country in country_isos_ref.keys():
for i in [0, 1, 2, 3, 5]: # test elements one by one:
self.assertEqual(country_isos[country][i],
country_isos_ref[country][i])
self.assertAlmostEqual(country_isos[country][4] * 1e-6,
country_isos_ref[country][4] * 1e-6, places=0)
def test_fill_econ_indicators_na_pass(self):
"""Test fill_econ_indicators with '' inputs."""
ref_year = 2019
country_isos = {
'CHE': [1, 'Switzerland', 'che_geom'],
'ZMB': [2, 'Zambia', 'zmb_geom']
}
gdp = {'CHE': 1.2 * 1e20, 'ZMB': ''}
inc_grp = {'CHE': '', 'ZMB': 4}
kwargs = {'gdp': gdp, 'inc_grp': inc_grp}
fill_econ_indicators(ref_year, country_isos, SHP_FILE, **kwargs)
country_isos_ref = {
'CHE': [1, 'Switzerland', 'che_geom', 2019, gdp['CHE'], 4],
'ZMB':
[2, 'Zambia', 'zmb_geom', 2019, 23064722446, inc_grp['ZMB']]
}
self.assertEqual(country_isos.keys(), country_isos_ref.keys())
for country in country_isos_ref.keys():
for i in [0, 1, 2, 3, 5]: # test elements one by one:
self.assertEqual(country_isos[country][i],
country_isos_ref[country][i])
self.assertAlmostEqual(country_isos[country][4] * 1e-6,
country_isos_ref[country][4] * 1e-6,
places=0)