def test_gets_stops_from_route_bart(self):
stop_list = u"['Daly City', 'Balboa Park (SF)', 'Glen Park (SF)', '24th St. Mission (SF)', '16th St. Mission (SF)', 'Civic Center (SF)', 'Powell St. (SF)', 'Montgomery St. (SF)', 'Embarcadero (SF)', 'West Oakland', 'Lake Merritt (Oakland)', 'Fruitvale (Oakland)', 'Coliseum/Oakland Airport', 'San Leandro', 'Bayfair (San Leandro)', 'Castro Valley', 'West Dublin', 'Dublin/Pleasanton']"
agency_id = 3
self.assertEqual(gets_stops_from_route(stop_list, agency_id), [
u'Daly City', u'Balboa Park (SF)', u'Glen Park (SF)',
u'24th St. Mission (SF)', u'16th St. Mission (SF)',
u'Civic Center (SF)', u'Powell St. (SF)', u'Montgomery St. (SF)',
u'Embarcadero (SF)', u'West Oakland', u'Lake Merritt (Oakland)',
u'Fruitvale (Oakland)', u'Coliseum/Oakland Airport',
u'San Leandro', u'Bayfair (San Leandro)', u'Castro Valley',
u'West Dublin', u'Dublin/Pleasanton'
])
def test_gets_stops_from_route_muni(self):
stop_list = u"[('Mission St and 13th St', '15546'), ('Mission St and 14th St', '15547'), ('Mission St and 16th St', '15551'), ('Mission St and 18th St', '15553'), ('Mission St and 20th St', '15557'), ('Mission St and 22nd St', '15561'), ('Mission St and 24th St', '15565'), ('Mission St and 26th St', '15567'), ('Mission St and 30th St', '15571'), ('Mission St and Appleton Ave', '15577'), ('Mission St and Brazil Ave', '15582'), ('Mission St and Cortland Ave', '15583'), ('Mission St and Excelsior Ave', '15586'), ('Mission St and Highland Ave', '15596'), ('Mission St and Murray St', '15605'), ('Mission St and Richland Ave', '15613'), ('Mission St and Silver Ave', '15620'), ('Mission St and Trumbull St', '15624'), ('Ocean Ave and Balboa Park Bart', '15781'), ('Ocean Ave and Cayuga Ave', '15783'), ('Ocean Ave and Howth St', '15791'), ('Ocean Ave and Otsego Ave', '15800'), ('Ocean Ave and San Jose St', '15805'), ('City College Terminal Phelan Loop', '15926'), ('S Van Ness Ave and Mission St', '16473'), ('Van Ness Ave and Broadway', '16798'), ('Van Ness Ave and Chestnut St', '16800'), ('Van Ness Ave and Clay St', '16803'), ('Van Ness Ave and Eddy St', '16804'), ('Van Ness Ave and Bay St', '16806'), ('Van Ness Ave and Jackson St', '16813'), ('Van Ness Ave and McAllister St', '16815'), ('Van Ness Ave and Market St', '16817'), ('Van Ness Ave and North Point St', '16818'), ('Van Ness Ave and No Point St', '16820'), ('Van Ness Ave and OFarrell St', '16823'), ('Van Ness Ave and Sutter St', '16829'), ('Van Ness Ave and Union St', '16833'), ('Mission St and Power St', '17841')]"
agency_id = 1
self.assertEqual(gets_stops_from_route(stop_list, agency_id),
[(u'Mission St and 13th St', u'15546'),
(u'Mission St and 14th St', u'15547'),
(u'Mission St and 16th St', u'15551'),
(u'Mission St and 18th St', u'15553'),
(u'Mission St and 20th St', u'15557'),
(u'Mission St and 22nd St', u'15561'),
(u'Mission St and 24th St', u'15565'),
(u'Mission St and 26th St', u'15567'),
(u'Mission St and 30th St', u'15571'),
(u'Mission St and Appleton Ave', u'15577'),
(u'Mission St and Brazil Ave', u'15582'),
(u'Mission St and Cortland Ave', u'15583'),
(u'Mission St and Excelsior Ave', u'15586'),
(u'Mission St and Highland Ave', u'15596'),
(u'Mission St and Murray St', u'15605'),
(u'Mission St and Richland Ave', u'15613'),
(u'Mission St and Silver Ave', u'15620'),
(u'Mission St and Trumbull St', u'15624'),
(u'Ocean Ave and Balboa Park Bart', u'15781'),
(u'Ocean Ave and Cayuga Ave', u'15783'),
(u'Ocean Ave and Howth St', u'15791'),
(u'Ocean Ave and Otsego Ave', u'15800'),
(u'Ocean Ave and San Jose St', u'15805'),
(u'City College Terminal Phelan Loop', u'15926'),
(u'S Van Ness Ave and Mission St', u'16473'),
(u'Van Ness Ave and Broadway', u'16798'),
(u'Van Ness Ave and Chestnut St', u'16800'),
(u'Van Ness Ave and Clay St', u'16803'),
(u'Van Ness Ave and Eddy St', u'16804'),
(u'Van Ness Ave and Bay St', u'16806'),
(u'Van Ness Ave and Jackson St', u'16813'),
(u'Van Ness Ave and McAllister St', u'16815'),
(u'Van Ness Ave and Market St', u'16817'),
(u'Van Ness Ave and North Point St', u'16818'),
(u'Van Ness Ave and No Point St', u'16820'),
(u'Van Ness Ave and OFarrell St', u'16823'),
(u'Van Ness Ave and Sutter St', u'16829'),
(u'Van Ness Ave and Union St', u'16833'),
(u'Mission St and Power St', u'17841')])
def test_gets_stops_from_route_muni(self):
stop_list = u"[('Mission St and 13th St', '15546'), ('Mission St and 14th St', '15547'), ('Mission St and 16th St', '15551'), ('Mission St and 18th St', '15553'), ('Mission St and 20th St', '15557'), ('Mission St and 22nd St', '15561'), ('Mission St and 24th St', '15565'), ('Mission St and 26th St', '15567'), ('Mission St and 30th St', '15571'), ('Mission St and Appleton Ave', '15577'), ('Mission St and Brazil Ave', '15582'), ('Mission St and Cortland Ave', '15583'), ('Mission St and Excelsior Ave', '15586'), ('Mission St and Highland Ave', '15596'), ('Mission St and Murray St', '15605'), ('Mission St and Richland Ave', '15613'), ('Mission St and Silver Ave', '15620'), ('Mission St and Trumbull St', '15624'), ('Ocean Ave and Balboa Park Bart', '15781'), ('Ocean Ave and Cayuga Ave', '15783'), ('Ocean Ave and Howth St', '15791'), ('Ocean Ave and Otsego Ave', '15800'), ('Ocean Ave and San Jose St', '15805'), ('City College Terminal Phelan Loop', '15926'), ('S Van Ness Ave and Mission St', '16473'), ('Van Ness Ave and Broadway', '16798'), ('Van Ness Ave and Chestnut St', '16800'), ('Van Ness Ave and Clay St', '16803'), ('Van Ness Ave and Eddy St', '16804'), ('Van Ness Ave and Bay St', '16806'), ('Van Ness Ave and Jackson St', '16813'), ('Van Ness Ave and McAllister St', '16815'), ('Van Ness Ave and Market St', '16817'), ('Van Ness Ave and North Point St', '16818'), ('Van Ness Ave and No Point St', '16820'), ('Van Ness Ave and OFarrell St', '16823'), ('Van Ness Ave and Sutter St', '16829'), ('Van Ness Ave and Union St', '16833'), ('Mission St and Power St', '17841')]"
agency_id = 1
self.assertEqual(gets_stops_from_route(stop_list, agency_id), [(u'Mission St and 13th St', u'15546'), (u'Mission St and 14th St', u'15547'), (u'Mission St and 16th St', u'15551'), (u'Mission St and 18th St', u'15553'), (u'Mission St and 20th St', u'15557'), (u'Mission St and 22nd St', u'15561'), (u'Mission St and 24th St', u'15565'), (u'Mission St and 26th St', u'15567'), (u'Mission St and 30th St', u'15571'), (u'Mission St and Appleton Ave', u'15577'), (u'Mission St and Brazil Ave', u'15582'), (u'Mission St and Cortland Ave', u'15583'), (u'Mission St and Excelsior Ave', u'15586'), (u'Mission St and Highland Ave', u'15596'), (u'Mission St and Murray St', u'15605'), (u'Mission St and Richland Ave', u'15613'), (u'Mission St and Silver Ave', u'15620'), (u'Mission St and Trumbull St', u'15624'), (u'Ocean Ave and Balboa Park Bart', u'15781'), (u'Ocean Ave and Cayuga Ave', u'15783'), (u'Ocean Ave and Howth St', u'15791'), (u'Ocean Ave and Otsego Ave', u'15800'), (u'Ocean Ave and San Jose St', u'15805'), (u'City College Terminal Phelan Loop', u'15926'), (u'S Van Ness Ave and Mission St', u'16473'), (u'Van Ness Ave and Broadway', u'16798'), (u'Van Ness Ave and Chestnut St', u'16800'), (u'Van Ness Ave and Clay St', u'16803'), (u'Van Ness Ave and Eddy St', u'16804'), (u'Van Ness Ave and Bay St', u'16806'), (u'Van Ness Ave and Jackson St', u'16813'), (u'Van Ness Ave and McAllister St', u'16815'), (u'Van Ness Ave and Market St', u'16817'), (u'Van Ness Ave and North Point St', u'16818'), (u'Van Ness Ave and No Point St', u'16820'), (u'Van Ness Ave and OFarrell St', u'16823'), (u'Van Ness Ave and Sutter St', u'16829'), (u'Van Ness Ave and Union St', u'16833'), (u'Mission St and Power St', u'17841')])
def test_gets_stops_from_route_bart(self):
stop_list = u"['Daly City', 'Balboa Park (SF)', 'Glen Park (SF)', '24th St. Mission (SF)', '16th St. Mission (SF)', 'Civic Center (SF)', 'Powell St. (SF)', 'Montgomery St. (SF)', 'Embarcadero (SF)', 'West Oakland', 'Lake Merritt (Oakland)', 'Fruitvale (Oakland)', 'Coliseum/Oakland Airport', 'San Leandro', 'Bayfair (San Leandro)', 'Castro Valley', 'West Dublin', 'Dublin/Pleasanton']"
agency_id = 3
self.assertEqual(gets_stops_from_route(stop_list, agency_id), [u'Daly City', u'Balboa Park (SF)', u'Glen Park (SF)', u'24th St. Mission (SF)', u'16th St. Mission (SF)', u'Civic Center (SF)', u'Powell St. (SF)', u'Montgomery St. (SF)', u'Embarcadero (SF)', u'West Oakland', u'Lake Merritt (Oakland)', u'Fruitvale (Oakland)', u'Coliseum/Oakland Airport', u'San Leandro', u'Bayfair (San Leandro)', u'Castro Valley', u'West Dublin', u'Dublin/Pleasanton'])
