37 lines
1.3 KiB
Python
37 lines
1.3 KiB
Python
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
|