2 # vi: set softtabstop=4 shiftwidth=4 tabstop=8 expandtab:
10 class TestItinGrab(unittest
.TestCase
):
13 fp
= open(self
.filename
, "r")
17 itinp
= Itinerary
.ItineraryParser()
21 self
.entries
= itinp
.entries
23 def checkAllRecognized(self
):
24 for ie
in self
.entries
:
25 self
.assertNotEquals(ie
.type, Itinerary
.TYPE_UNKNOWN
)
28 class TestChampagneItin(TestItinGrab
):
30 filename
= "grabs/champagne-itin.html"
32 def testNumEntries(self
):
33 self
.assertEquals(len(self
.entries
), 4)
36 self
.checkAllRecognized()
38 class TestLongItin(TestItinGrab
):
40 filename
= "grabs/long-itin.html"
42 def testNumEntries(self
):
43 self
.assertEquals(len(self
.entries
), 9)
46 self
.checkAllRecognized()
50 self
.assertEquals(ie
.type, Itinerary
.TYPE_TAKE_BUS
)
51 self
.assertEquals(ie
.busStop
, "1635")
52 self
.assertEquals(ie
.duration
, None)
53 self
.assertEquals(ie
.startTime
, '11:10 AM')
54 self
.assertEquals(ie
.endTime
, '11:20 AM')
55 self
.assertEquals(ie
.route
, '116')
56 self
.assertEquals(ie
.direction
, 'Baseline')
57 self
.assertEquals(ie
.destination
, 'MERIVALE / COLONNADE')
59 def testShortFormatter(self
):
60 sf
= ShortFormatter
.ShortFormatter(self
.entries
)
63 if __name__
== '__main__':