class ImportDataTest(TestCase):
def setUp(self):
self.data_loader = DataLoader()
def test_create_sectors(self):
self.data_loader.create_sectors()
sectors = dict([(s.slug, s.name) for s in Sector.objects.all()])
expected_dict = {
'education': 'Education',
'health': 'Health',
'water': 'Water'
}
self.assertEquals(sectors, expected_dict)
def test_create_facility_types(self):
self.data_loader.create_facility_types()
self.assertEquals(FacilityType.objects.count(), 23)
def test_limited_import(self):
# this needs to be rewritten with small csvs
# data_loader.load()
# expected_info = {
# "number of facilities": 41,
# "unused variables": [],
# "facilities without lgas": 0,
# "number of lga records": 183,
# "number of facility records": 4419
# }
# self.assertEquals(data_loader.get_info(), expected_info)
pass
def count_all_objects(self):
result = {}
for ct in ContentType.objects.all():
result[ct.natural_key()] = ct.model_class().objects.count()
return result
