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)