def testDistrictExclusion(self): old_sp = self.contact.supply_point self.contact.supply_point = SupplyPoint.objects.exclude\ (type__code=SupplyPointCodes.DISTRICT)[0] self.contact.save() self.assertEqual(0, len(list(reports.get_district_people()))) self.contact.supply_point = old_sp self.contact.save() self.assertEqual(1, len(list(reports.get_district_people())))
def testBasicList(self): people = list(reports.get_district_people()) self.assertEqual(1, len(people)) for person in people: self.assertEqual(self.contact, person)