def test_nav_zone_equality(self): nzs = [get_nav_zones(), get_nav_zones_inefficient()] def get_names(z): [n['name'] for n in z] self.assertEqual(*[get_names(nz) for nz in nzs]) def state_names(z): return [s['name'] for s in z['states']] for nzi in range(0, len(nz)): self.assertEqual(*[state_names(nz[nzi]) for nz in nzs]) def ordered_lga_slugs(z): lga_slugs = [] for s in z['states']: for lga in s['lgas']: lga_slugs.append(lga['unique_slug']) return lga_slugs for nzi in range(0, len(nz)): self.assertEqual(*[ordered_lga_slugs(nz[nzi]) for nz in nzs])
def test_nav_zones(self): nav_zones = get_nav_zones() self.assertEqual(len(nav_zones), 6)