Script to grab stop codes from allstops.xml and sort
[ottawa-travel-planner.git] / tests / test_landmark_matcher.py
blobf9e239967d310862a0c6d4f6cd36c4b96226cd3c
2 # vi: set softtabstop=4 shiftwidth=4 tabstop=8 expandtab:
4 import unittest
5 import LandmarkMatcher
7 class TestMatch(unittest.TestCase):
9 def setUp(self):
10 self.lm = LandmarkMatcher.LandmarkMatcher()
12 def testFullMatch(self):
13 self.assertEquals(self.lm.match("rideau centre"), "RIDEAU CENTRE")
15 def testPartialMatchRideau(self):
16 self.assertEquals(self.lm.match("rideau"), "RIDEAU TRANSIT MALL")
18 def testPartialMatchBayshore(self):
19 self.assertEquals(self.lm.match("bayshore"), "BAYSHORE STATION")
21 if __name__ == '__main__':
22 unittest.main()