def test_get(self): path = pf.get("shp/transporte") exp_path = os.path.join(pf.get_project_dir(), "shp/transporte") self.assertEqual(path, exp_path) path = pf.get("/Users/abenassi/github/tod/shp/transporte") exp_path = "/Users/abenassi/github/tod/shp/transporte" self.assertEqual(path, exp_path)
def test_get_shp(self): path = pf.get_shp("subte-estaciones") exp_path = pf.get( "shp/transporte/subte-estaciones/estaciones_de_subte") self.assertEqual(path, exp_path) path = pf.get_shp("estaciones_de_subte") exp_path = pf.get( "shp/transporte/subte-estaciones/estaciones_de_subte") self.assertEqual(path, exp_path)
from __future__ import unicode_literals from __future__ import print_function from __future__ import with_statement import os import sys import zipfile import shapefile import glob from shapely.ops import cascaded_union import path_finders as pf from geo_utils import iter_shp_as_shapely import geo_utils import utils BASE_DIR = pf.get("shp/transporte") BUFFER_DIR = "buffers" BUFFERS = [300, 500, 750, 1000, 1500, 2000] CONTEXT_SHP = pf.get_shp("comunas_caba_censo_2010") def _create_shp_name(shp_dir, distance): return os.path.basename(shp_dir) + "-buffer" + unicode(distance) def _create_shp_path(directory, shp_name, buffer_dir): return os.path.join(os.path.dirname(directory), buffer_dir, shp_name, shp_name) def write_shapefile(sf_est, shapely_shapes, shp_path):