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")
