Python extract_stations Examples

Example #1

File: test_extract.py Project: zhangzijie/gzbus

def test_extract_current_routine(normal_bus_station_page):
    stations = extract_stations(normal_bus_station_page)
    current = extract_current_routine(normal_bus_station_page, stations)

    assert current['destinate_station'] == '华工大总站①'
    assert current['bus_station'] == '星海学院站'
    assert current['waiting_station'] == '科学院地化所站'

Example #2

File: test_extract.py Project: zhangzijie/gzbus

def test_extract_stations(normal_bus_station_page):
    stations = extract_stations(normal_bus_station_page)
    terminals = sorted(list(stations['terminal'].keys()))

    assert terminals == ['华工大总站①', '大学城(广中医)总站']
    assert len(stations['stations']) == 22