def setUp(self): self.power = Variable.objects.create( slug='power', data_type='string' ) self.has_water = Variable.objects.create( slug='has_water', data_type='boolean' ) self.num_doctors = Variable.objects.create( slug='num_doctors', data_type='float' ) self.zone = Zone.objects.create(name='Zone', slug='zone') self.state = State.objects.create(name='Zone', slug='zone', zone=self.zone) lga_names = ['LGA_1', 'LGA_2'] self.lgas = [LGA.objects.create(name=n, slug=n, state=self.state) for n in lga_names] self.sector = Sector.objects.create(name='Test', slug='test') self.facility_type = FacilityType.add_root(name='Test', slug='test') self.facilities = [] for lga in self.lgas: for facility_id in ['a', 'b', 'c', 'd']: self.facilities.append( Facility.objects.create(facility_id=facility_id, lga=lga, sector=self.sector, facility_type=self.facility_type) ) for facility, value in zip(self.facilities, ['none', 'good', 'bad', 'none']): facility.set(self.power, value) for facility, value in zip(self.facilities, [True, False, True, False]): facility.set(self.has_water, value) for facility, value in zip(self.facilities, [10.0, 20.0, 10.0, 30.0]): facility.set(self.num_doctors, value)
def setUp(self): # I took this set up code from LGAIndicatorTest self.has_water = Variable.objects.create( slug='has_water', data_type='boolean' ) self.zone = Zone.objects.create(name='Zone', slug='zone') self.state = State.objects.create(name='Zone', slug='zone', zone=self.zone) lga_names = ['LGA_1', 'LGA_2'] self.lgas = [LGA.objects.create(name=n, slug=n, state=self.state, unique_slug='state_%s' % n, data_available=True, data_load_in_progress=False, data_loaded=True) for n in lga_names] self.sector = Sector.objects.create(name='Test', slug='test') self.facilities = [] ftype_root = FacilityType.add_root(slug='facility_type', name='Facility Type') ftype = ftype_root.add_child(name="Toilet", slug="toilet") for lga in self.lgas: for facility_id in ['a', 'b']: self.facilities.append( Facility.objects.create(facility_id=facility_id, lga=lga, sector=self.sector, facility_type=ftype) ) self.assertEquals(len(self.facilities), 4) for facility, value in zip(self.facilities, [True, True, True, False]): facility.set(self.has_water, value) admin, created = User.objects.get_or_create( username="******", email="*****@*****.**", is_staff=True, is_superuser=True ) admin.set_password("pass") admin.save() self.client = Client() self.client.login(username='******', password='******')
def setUp(self): self.zone = Zone.objects.create(name='Zone', slug='zone') self.state = State.objects.create(name='State', slug='state', zone=self.zone) self.lga = LGA.objects.create(name='LGA', slug='LGA', state=self.state) self.sector = Sector.objects.create(name='Education', slug='education') self.facility_type = FacilityType.add_root(name='Education', slug='education') fields = ['slug', 'data_type', 'formula'] data_dictionary_types = [ ['student_teacher_ratio', 'float', "d['num_students_total'] / d['num_tchrs_total']"], ['sufficient_number_of_teachers', 'boolean', "d['student_teacher_ratio'] <= 35"], ['ideal_number_of_classrooms', 'float', "d['num_students_total'] / 35"], ] for ddt in data_dictionary_types: d = dict(zip(fields, ddt)) CalculatedVariable.objects.get_or_create(**d)
def setUp(self): self.power = Variable.objects.create( slug='power', data_type='string' ) self.has_water = Variable.objects.create( slug='has_water', data_type='boolean' ) self.num_doctors = Variable.objects.create( slug='num_doctors', data_type='float' ) self.zone = Zone.objects.create(name='Zone', slug='zone') self.state = State.objects.create(name='Zone', slug='zone', zone=self.zone) self.lga = LGA.objects.create(name='Local Govt Area', slug='lga', state=self.state) self.sector = Sector.objects.create(name='Test', slug='test') self.facility_type = FacilityType.add_root(name='Test', slug='test') self.facility = Facility.objects.create(facility_id='x', lga=self.lga, sector=self.sector, facility_type=self.facility_type) self.variable_values = [ (self.power, 'none'), (self.has_water, True), (self.num_doctors, 10) ] for variable, value in self.variable_values: self.facility.set(variable, value)
def create_node(d, parent): children = d.pop('children') result = FacilityType.add_root(**d) if parent is None else parent.add_child(**d) for child in children: create_node(child, result) return result