Pure Python implementation of clients for Mapbox services

2017-12-27 13:43:14 +01:00
parent b16f7e6ed5
commit 8c0af7d51d
13 changed files with 620 additions and 1 deletions
@@ -0,0 +1,36 @@
+import unittest
+from cartodb_services.mapbox.isolines import MapboxIsolines
+from cartodb_services.mapbox.matrix_client import DEFAULT_PROFILE
+from cartodb_services.mapbox.matrix_client import MapboxMatrixClient
+from cartodb_services.mapbox.routing import MapboxRouting
+from cartodb_services.tools import Coordinate
+from cartodb_services.tools.coordinates import (validate_coordinates,
+                                                marshall_coordinates)
+
+VALID_ORIGIN = Coordinate(-73.989, 40.733)
+
+
+class MapboxIsolinesTestCase(unittest.TestCase):
+
+    def setUp(self):
+        matrix_client = MapboxMatrixClient()
+        routing_client = MapboxRouting()
+        self.mapbox_isolines = MapboxIsolines(matrix_client, routing_client)
+
+    def test_calculate_isochrone(self):
+        time_range = 10 * 60  # 10 minutes
+        solution = self.mapbox_isolines.calculate_isochrone(
+            origin=VALID_ORIGIN,
+            profile=DEFAULT_PROFILE,
+            time_range=time_range)
+
+        assert solution
+
+    def test_calculate_isodistance(self):
+        distance_range = 10000
+        solution = self.mapbox_isolines.calculate_isodistance(
+            origin=VALID_ORIGIN,
+            profile=DEFAULT_PROFILE,
+            distance_range=distance_range)
+
+        assert solution