예제 #1
0
class TestValidTrip(unittest.TestCase):
    def setUp(self):
        """Create train instance with valid train"""
        self.train = 79 # A valid Via Train (Toronto-Windsor)
        self.date = "2015-02-02" # Runs everyday
        expected_stations = ["TORONTO", "OAKVILLE", "ALDERSHOT", "BRANTFORD",
                            "WOODSTOCK", "INGERSOLL", "LONDON", "GLENCOE",
                            "CHATHAM", "WINDSOR"]
        self.trip = Trip(self.train, self.date)

    def test_number_of_stations(self):
        """Number of stations in trip"""
        assertEqual(self.trip.num_stations, len(self.expected_stations))

    def test_scheduled_stations(self):
        """List of stations for trip"""
        stations = []
        schedule = self.trip.get_schedule()
        for s in schedule: stations.append(upper(s.station_name))
        assert sorted(stations) == sorted(self.expected_stations)

    def test_departure_station(self):
        """The first station of the trip"""
        assertEqual(upper(self.trip.get_schedule()[0]["station_name"]), expected_stations[0]) 

    def test_arrival_station(self):
        """The last station of the trip"""
        assertEqual(upper(self.trip.get_schedule()[-1]["station_name"]), expected_stations[-1])
예제 #2
0
 def setUp(self):
     """Create train instance with valid train"""
     self.train = 79 # A valid Via Train (Toronto-Windsor)
     self.date = "2015-02-02" # Runs everyday
     expected_stations = ["TORONTO", "OAKVILLE", "ALDERSHOT", "BRANTFORD",
                         "WOODSTOCK", "INGERSOLL", "LONDON", "GLENCOE",
                         "CHATHAM", "WINDSOR"]
     self.trip = Trip(self.train, self.date)