def test_get_department_from_postal_code(self): self.assertRaises(ValueError, get_department_from_postal_code, '') self.assertRaises(ValueError, get_department_from_postal_code, '1234') pas_de_calais = Department.objects.get(pk='62') paris = Department.objects.get(pk='75') corse_du_sud = Department.objects.get(pk='2A') reunion = Department.objects.get(pk='974') self.assertEquals(get_department_from_postal_code('62138'), pas_de_calais) self.assertEquals(get_department_from_postal_code('75012'), paris) self.assertEquals(get_department_from_postal_code('20600'), corse_du_sud) self.assertEquals(get_department_from_postal_code('97412'), reunion)
def process(self, postal_code): return get_department_from_postal_code(postal_code)