From: Florian Schweikert Date: Fri, 16 Dec 2011 14:32:04 +0000 (+0100) Subject: fixed unittest X-Git-Url: https://vcs.maemo.org/git/?p=pywienerlinien;a=commitdiff_plain;h=936d29b9f2a032e8c4efeac32f9bc97cbf8fa975 fixed unittest add unittest --- diff --git a/gotovienna/realtime.py b/gotovienna/realtime.py index 6454667..7bb2773 100644 --- a/gotovienna/realtime.py +++ b/gotovienna/realtime.py @@ -240,7 +240,6 @@ class ITipParser: t = th[-1] else: # all other lines - print th[-1].find('img') and th[-1].img.has_key('alt') d['lowfloor'] = th[-1].find('img') and th[-1].img.has_key('alt') d['line'] = th[0].text.replace(' ', '') d['direction'] = th[1].text.replace(' ', '') diff --git a/gotovienna/tests/realtime.py b/gotovienna/tests/realtime.py index b61aecb..7d998ae 100644 --- a/gotovienna/tests/realtime.py +++ b/gotovienna/tests/realtime.py @@ -55,6 +55,8 @@ def test_departures_by_station(): l = list(set(map(lambda x: x['line'], dep))) # there are 8 different lines assert_equal(8, len(l)) + assert_equal(u'Leopoldau', dep[0]['direction']) + assert_equal(u'Karlsplatz', dep[0]['station']) def test_departures_by_station_lowfloor(): dep = parser.parse_departures_by_station(stationbased) @@ -65,9 +67,21 @@ def test_departures_by_station_datetime(): dep = parser.parse_departures_by_station(stationbased) assert_equal(int, type(dep[13]['time'])) assert_equal(time, type(dep[14]['time'])) + assert_equal(18, dep[13]['time']) + assert_equal(time(13, 5), dep[14]['time']) def test_departures(): dep = parser.parse_departures(line_station) + assert_equal(0, dep[0]['time']) + assert_equal(10, dep[2]['time']) + assert_equal(6, len(dep)) + assert_equal(u'Stefan-Fadinger-Platz', dep[0]['direction']) + assert_equal(u'Kärntner Ring, Oper', dep[0]['station']) + +def test_departures_lowfloor(): + dep = parser.parse_departures(line_station) + assert_false(dep[1]['lowfloor']) + assert_true(dep[2]['lowfloor']) def test_error_page(): dep = parser.parse_departures(errorpage)