tests/test_latlongtools.py

   1 # vi: set softtabstop=4 shiftwidth=4 tabstop=8 expandtab:
   2
   3 import unittest
   4 import LatLongTools
   5 from BusStopMashup import StopLocation
   6
   7 class TestFindCorners(unittest.TestCase):
   8     def testSimple(self):
   9         center = StopLocation(45, -75)
  10         points = { 'sw1': (44.1, -75.1), 'sw2': (44.1, -75.05),
  11                    'sw3': (44.1, -75.2),
  12                    'nw1': (45.1, -76),
  13                    'ne1': (45.1, -74),
  14                    'se1': (44, -74) }
  15         corners = LatLongTools.findCorners(center,
  16                               [(StopLocation(points[k][0], points[k][1]), k)
  17                                   for k in points.iterkeys()])
  18
  19         self.assertEquals(4, len(corners))
  20         self.assertEquals(1, len(corners[LatLongTools.NORTHWEST]))
  21         self.assertEquals(1, len(corners[LatLongTools.NORTHEAST]))
  22         self.assertEquals(3, len(corners[LatLongTools.SOUTHWEST]))
  23         self.assertEquals(1, len(corners[LatLongTools.SOUTHEAST]))
  24
  25         self.assertEquals("nw1", corners[LatLongTools.NORTHWEST][0])
  26         self.assertEquals("ne1", corners[LatLongTools.NORTHEAST][0])
  27         self.assertEquals("se1", corners[LatLongTools.SOUTHEAST][0])
  28
  29         sw = corners[LatLongTools.SOUTHWEST]
  30         self.assertEquals("sw3", sw[0])
  31         self.assertEquals("sw1", sw[1])
  32         self.assertEquals("sw2", sw[2])
  33
  34 if __name__ == '__main__':
  35     unittest.main()