def test_add_planet_to_sector(): universe = Universe() sector = Sector(sector_name='Test Sector') planet = Station(station_name='Test Planet') sector.add_planet(planet) universe.add_sector(sector) assert(planet in sector.planets)
def test_add_station_to_sector(): universe = Universe() sector = Sector(sector_name='Test Sector') station = Station(station_name='Test Station') sector.add_station(station) universe.add_sector(sector) assert(station in sector.stations)