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'] == '科学院地化所站'
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