def test_getAdjacent(): # Put two areas in a zone a = Area(name="A", desc="area_desc", domain=[8], action=lambda: 5, resource_id="a_id") b = Area(name="B", desc="area_desc", domain=[9], action=lambda: 5, resource_id="b_id") z = Zone([a, b]) for x in z.areas: x.zone = z assert a.zone == z assert b.zone == z # Test adjacency assert a.getAdjacent() == b assert b.getAdjacent() == a