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)
