import unittest
 2 import datetime
 3 from seamonapi.waypoint import parse_drivepoints
 4 from .printer import printer # pretty printer
 5 from routes import set_osrm_server
 6
 7 set_osrm_server("http://api.seamon.io:5000")
 8
 9 class TestWaypoint(unittest.TestCase):
10
11     def test_drive_1wp(self):
12         pts = ['(Reykjavik;initial_time)']
13         initial_time = datetime.datetime(2008, 11, 10, 17, 53, 59)
14         end_time = initial_time + datetime.timedelta(hours=2)
15         res = parse_drivepoints(pts, initial_time, end_time)
16
17         self.assertEqual( len(res[0]), 2 )
18         self.assertEqual( len(res[1]), 2 )
19         self.assertEqual( res[1][1]['label'], 'Reykjavik')
20         self.assertEqual( res[1][1]['time'], (end_time-datetime.datetime(1970, 1, 1)).total_seconds())
21
22     def test_drive_latlon(self):
23         pts = ['(64.14;-21.94;initial_time)','(Selfoss;25kph)']
24         initial_time = datetime.datetime(2008, 11, 10, 17, 53, 59)
25         end_time = initial_time + datetime.timedelta(hours=2)
26         res = parse_drivepoints(pts, initial_time, end_time)
27
28         # should return reasonably large latlon and waypts results
29         self.assertGreater( len(res[0]), 10 )
30         self.assertGreater( len(res[1]), 10 )
31
32     def test_drive_3wp(self):
33         pts = ['(Reykjavik;initial_time)','(64.161075;-21.583501;80kph)','(Selfoss;80kph)']
34         initial_time = datetime.datetime(2008, 11, 10, 17, 53, 59)
35         end_time = initial_time + datetime.timedelta(hours=2)
36         res = parse_drivepoints(pts, initial_time, end_time)
37
38         self.assertGreater( len(res[0]), 10 )
39         self.assertGreater( len(res[1]), 10 )
40         self.assertEqual( res[1][0]['label'], 'Reykjavik')
41         self.assertEqual( res[1][-1]['label'], 'Selfoss')
42
43     def assertApproxEqual(self, actual, expected, frac_error=0.001):
44         """ asserts that actual and expected argument are equal to within a fraction of expected value """
45         self.assertTrue( (1.0-frac_error)*expected <= actual <= expected*(1.0+frac_error) )
46
47     def test_drive_unknown_address(self):
48         pts = ['(64.14;-21.94;initial_time)','(SelfossKK;25kph)']
49         initial_time = datetime.datetime(2008, 11, 10, 17, 53, 59)
50         end_time = initial_time + datetime.timedelta(hours=2)
51
52         # catch expected exception
53         with self.assertRaises(ValueError) as e:
54             parse_drivepoints(pts, initial_time, end_time)
55
56         # exception message should report 'selfosskk' not found
57         msg = e.exception.message
58         self.assertIn("'selfosskk' not found", msg)