def test_passengers_integration(self): passengers = Passengers(c.SAMPLE_DATA) assert passengers.header == c.HEADERS for feature, value in passengers.feature_sets.items(): assert Counter(c.FEATURE_SETS[feature]) == Counter(value) assert len(passengers.feature_sets) == len(c.FEATURE_SETS)
from passengers import Passengers from traxi import Traxi from check_inputs import check_inputs if __name__ == "__main__": check_inputs() p = Passengers() t = Traxi() while True: new_passenger = p.generate() t.manage(new_passenger)