def update_country_data():
from geodata.importer.country import CountryImport
ci = CountryImport()
ci.update_country_center()
ci.update_polygon()
ci.update_regions()
ci.update_alt_name()
class CountryImportTestCase(TestCase):
def setUp(self):
self.country_import = CountryImport()
self.country = CountryFactory.create(code="AF")
def test_update_polygon(self):
data = {
"type": "FeatureCollection", "features": [{
"type": "Feature",
"id": "AFG",
"properties": {
"name": "Afghanistan",
"iso2": "AF"
},
"geometry": {
"type": "Polygon",
"coordinates": [
[[61.210817, 35.650072], [62.230651, 35.270664]]
]}},
]}
self.country_import.get_json_data = MagicMock(return_value=data)
self.country_import.update_polygon()
self.assertEqual(1, self.country_import.get_json_data.call_count)
self.assertEqual(1, Country.objects.all().count())
country = Country.objects.all()[0]
self.assertIsNotNone(country.polygon)
self.assertEqual(country.code, self.country.code)
def test_update_country_center(self):
data = {"AF": {"latitude": "42.30", "longitude": "1.30"}, }
self.country_import.get_json_data = MagicMock(return_value=data)
self.country_import.update_country_center()
country = Country.objects.all()[0]
self.assertEqual(country.center_longlat.y, 42.3)
self.assertEqual(country.center_longlat.x, 1.3)
def test_update_regions(self):
rv = vocabulary_factory.RegionVocabularyFactory()
region = RegionFactory.create(code='689', region_vocabulary=rv)
data = [{
"country_name": "Afghanistan",
"iso2": "AF",
"dac_region_code": "689",
}, ]
self.country_import.get_json_data = MagicMock(return_value=data)
self.country_import.update_regions()
country = Country.objects.all()[0]
self.assertEqual(country.region, region)
class CountryImportTestCase(TestCase):
def setUp(self):
self.country_import = CountryImport()
self.country = CountryFactory.create(code="AF")
def test_update_polygon(self):
data = {
"type": "FeatureCollection", "features": [{
"type": "Feature",
"id": "AFG",
"properties": {
"name": "Afghanistan",
"iso2": "AF"
},
"geometry": {
"type": "Polygon",
"coordinates": [
[[61.210817, 35.650072], [62.230651, 35.270664]]
]}},
]}
self.country_import.get_json_data = MagicMock(return_value=data)
self.country_import.update_polygon()
self.assertEqual(1, self.country_import.get_json_data.call_count)
self.assertEqual(1, Country.objects.all().count())
country = Country.objects.all()[0]
self.assertIsNotNone(country.polygon)
self.assertEqual(country.code, self.country.code)
def test_update_country_center(self):
data = {"AF": {"latitude": "42.30", "longitude": "1.30"}, }
self.country_import.get_json_data = MagicMock(return_value=data)
self.country_import.update_country_center()
country = Country.objects.all()[0]
self.assertEqual(country.center_longlat.y, 42.3)
self.assertEqual(country.center_longlat.x, 1.3)
def test_update_regions(self):
region = RegionFactory.create(code='689')
data = [{
"country_name": "Afghanistan",
"iso2": "AF",
"dac_region_code": "689",
}, ]
self.country_import.get_json_data = MagicMock(return_value=data)
self.country_import.update_regions()
country = Country.objects.all()[0]
self.assertEqual(country.region, region)
from geodata.importer.country import CountryImport
# from geodata.importer.region import RegionImport
from geodata.importer.subnational import SubnationalImport
# print('Region data')
# ri = RegionImport()
# ri.update_region_center()
print('Country data')
ci = CountryImport()
ci.update_polygon()
ci.update_alt_name()
ci.update_country_center()
ci.update_regions()
ci.update_hd_polygons()
ci.update_region_polygons_centers()
print('Subnational data')
si = SubnationalImport()
si.update_polygon()
si.update_kenya()
si.update_kenya_county_centers()
def update_country_data():
from geodata.importer.country import CountryImport
ci = CountryImport()
ci.update_country_center()
ci.update_polygon()
ci.update_regions()