def test_add_district(self): d = District("TestDistrict", "HS", "9-12") t = Town("Test Town") t.add_district(d) assert d in t.districts
def test_repr(self): t = Town("Test Town") assert t.__repr__() == "Test Town"