def testFilter(self): """Test if filtering works properly.""" expected_values = { "CITY1": 0, "CITY2": 0, "CITY3": 0, "CITY4": 0, "CITY5": 0, "CITY6": 0, "CITY7": 0, "CITY8": 0, "CITY9": 0, "CITY10": 0, "CITY11": 1, "CITY12": 1, } filter = unusual_trip_filter.UnusualTripFilter(0.1, quiet=True) input = self.GetPath("tests", "data", "filter_unusual_trips") loader = transitfeed.Loader(input, extra_validation=True) schedule = loader.Load() filter.filter(schedule) for trip_id, expected_trip_type in expected_values.items(): actual_trip_type = schedule.trips[trip_id]["trip_type"] try: self.assertEquals(int(actual_trip_type), expected_trip_type) except ValueError: self.assertEquals(actual_trip_type, "")
def testFilter(self): """Test if filtering works properly.""" expected_values = { 'CITY1': 0, 'CITY2': 0, 'CITY3': 0, 'CITY4': 0, 'CITY5': 0, 'CITY6': 0, 'CITY7': 0, 'CITY8': 0, 'CITY9': 0, 'CITY10': 0, 'CITY11': 1, 'CITY12': 1, } filter = unusual_trip_filter.UnusualTripFilter(0.1, quiet=True) input = self.GetPath('test', 'data', 'filter_unusual_trips') loader = transitfeed.Loader(input, extra_validation=True) schedule = loader.Load() filter.filter(schedule) for trip_id, expected_trip_type in expected_values.items(): actual_trip_type = schedule.trips[trip_id]['trip_type'] try: self.assertEquals(int(actual_trip_type), expected_trip_type) except ValueError: self.assertEquals(actual_trip_type, '')
def testFilterAppliedForRouteTypeSpecifiedByName(self): """Setting integer route_type filters trips of this route type.""" filter = unusual_trip_filter.UnusualTripFilter(0.1, quiet=True, route_type='Bus') input = self.GetPath('tests', 'data', 'filter_unusual_trips') loader = transitfeed.Loader(input, extra_validation=True) schedule = loader.Load() filter.filter(schedule) actual_trip_type = schedule.trips['CITY11']['trip_type'] self.assertEquals(actual_trip_type, '1')
def testFilterNotAppliedForUnspecifiedRouteType(self): """Setting integer route_type filters trips of this route type.""" filter = unusual_trip_filter.UnusualTripFilter(0.1, quiet=True, route_type=2) input = self.GetPath("tests", "data", "filter_unusual_trips") loader = transitfeed.Loader(input, extra_validation=True) schedule = loader.Load() filter.filter(schedule) actual_trip_type = schedule.trips["CITY11"]["trip_type"] self.assertEquals(actual_trip_type, "")
def testFilterForceFilter(self): """Test that force==True does set default values""" filter = unusual_trip_filter.UnusualTripFilter(0.1, force=True, quiet=False) input = self.GetPath('tests', 'data', 'filter_unusual_trips') loader = transitfeed.Loader(input, extra_validation=True) schedule = loader.Load() schedule.trips['CITY2'].trip_type = 'odd-trip' filter.filter(schedule) trip1 = schedule.trips['CITY1'] self.assertEquals(trip1['trip_type'], '0') trip2 = schedule.trips['CITY2'] self.assertEquals(trip2['trip_type'], '0')
def testFilterForceFilter(self): """Test that force==True does set default values""" filter = unusual_trip_filter.UnusualTripFilter(0.1, force=True, quiet=False) input = self.GetPath("tests", "data", "filter_unusual_trips") loader = transitfeed.Loader(input, extra_validation=True) schedule = loader.Load() schedule.trips["CITY2"].trip_type = "odd-trip" filter.filter(schedule) trip1 = schedule.trips["CITY1"] self.assertEquals(trip1["trip_type"], "0") trip2 = schedule.trips["CITY2"] self.assertEquals(trip2["trip_type"], "0")