def test1(): "Get table with departures for a specific stop." table = vbbvg.get_next_departures(u'bundesplatz') assert len(table) > 5 assert u'Ringbahn S 42' in table.Destination.values assert u'Ringbahn S 41' in table.Destination.values
def test2(): "Compare tables received for stop name and ID given for the same stop." table1 = vbbvg.get_next_departures(u'möckernbrücke') table2 = vbbvg.get_next_departures(u'9017104') assert set(table1.Destination.values) == set(table2.Destination.values)