def test_country_iso_geom_pass(self): """Test country_iso_geom pass.""" countries = ['Switzerland'] _, cntry_admin1 = country_iso_geom(countries, SHP_FILE) self.assertEqual(cntry_admin1, {'CHE': []}) self.assertIsInstance(countries, list) countries = {'Switzerland': ['Zürich']} _, cntry_admin1 = country_iso_geom(countries, SHP_FILE) self.assertEqual(len(cntry_admin1['CHE']), 1) self.assertIsInstance(cntry_admin1['CHE'][0], shapely.geometry.polygon.Polygon)
def test_bolivia_pass(self): """BOL""" country_name = ['Bolivia'] iso_name, _ = country_iso_geom(country_name, SHP_FILE) self.assertEqual(len(iso_name), len(country_name)) self.assertEqual(iso_name['BOL'][0], 68) self.assertEqual(iso_name['BOL'][1], 'Bolivia') self.assertIsInstance(iso_name['BOL'][2], shapely.geometry.polygon.Polygon)
def test_haiti_pass(self): """HTI""" country_name = ['HaITi'] iso_name, _ = country_iso_geom(country_name, SHP_FILE) self.assertEqual(len(iso_name), len(country_name)) self.assertEqual(iso_name['HTI'][0], 332) self.assertEqual(iso_name['HTI'][1], 'Haiti') self.assertIsInstance(iso_name['HTI'][2], shapely.geometry.multipolygon.MultiPolygon)
def test_che_kos_pass(self): """CHE, KOS""" country_name = ['Switzerland', 'Kosovo'] iso_name, _ = country_iso_geom(country_name, SHP_FILE) self.assertEqual(len(iso_name), len(country_name)) self.assertTrue('CHE' in iso_name) self.assertTrue('KOS' in iso_name) self.assertEqual(iso_name['CHE'][0], 756) self.assertEqual(iso_name['CHE'][1], 'Switzerland') self.assertIsInstance(iso_name['CHE'][2], shapely.geometry.polygon.Polygon) self.assertEqual(iso_name['KOS'][0], 0) self.assertEqual(iso_name['KOS'][1], 'Kosovo') self.assertIsInstance(iso_name['KOS'][2], shapely.geometry.polygon.Polygon)
def test_korea_pass(self): """PRK""" country_name = ['Korea'] with self.assertRaises(ValueError): country_iso_geom(country_name, SHP_FILE)
def test_wrong_fail(self): """Wrong name""" country_name = ['Kasovo'] with self.assertRaises(ValueError): country_iso_geom(country_name, SHP_FILE)