1 from nose.tools import assert_equal, assert_is_instance
4 from datetime import time, datetime
6 # bananas for the monkey
7 class datetime_static(datetime):
10 return datetime(2000, 1, 1, 11, 50)
12 from gotovienna import realtime
13 realtime.datetime = datetime_static
16 sys.path.insert(0, os.path.dirname(os.path.dirname(__file__)))
17 from gotovienna.realtime import Departure
18 from utils import sort_departures
20 departures = [Departure('U2', 'Karlsplatz', 'Aspernstrasse', time(13, 37), True),
21 Departure('U1', 'Karlsplatz', 'Reumannplatz', time(12, 24), False),
22 Departure('U4', 'Karlsplatz', 'Huetteldorf', time(11, 42), False),
23 Departure('U4', 'Karlsplatz', 'Heiligenstadt', 5, True),
24 Departure('U2', 'Karlsplatz', 'Aspernstrasse', time(13, 38), True)]
26 departures_sorted = [Departure('U4', 'Karlsplatz', 'Huetteldorf', time(11, 42), False),
27 Departure('U4', 'Karlsplatz', 'Heiligenstadt', 5, True),
28 Departure('U1', 'Karlsplatz', 'Reumannplatz', time(12, 24), False),
29 Departure('U2', 'Karlsplatz', 'Aspernstrasse', time(13, 37), True),
30 Departure('U2', 'Karlsplatz', 'Aspernstrasse', time(13, 38), True)]
33 assert_equal(departures_sorted, sort_departures(departures))
36 for dep in departures:
37 assert_is_instance(dep['departure'], datetime)
40 for dep in departures:
41 assert_is_instance(dep['ftime'], str)
44 for dep in departures:
45 assert_is_instance(dep['time'], int)