comparison scripts/utils_spatial.py @ 8:0f3eba42b425 branch-tests

added notebooks and utils

 """
 import numpy as np
 import json
 import pysal # before shapely in util_plots
 import fiona
+import os
 import sys
 sys.path.append('../misc')
 import matplotlib.pyplot as plt
 
 
-def neighbors_from_json_file(data_countries, json_DB='../MergeBL-Smith/data/countries.json'):
+DATA_DIR = os.path.join(os.path.dirname(__file__), 'util_data')
+JSON_DB = os.path.join(DATA_DIR, 'countries.json')
+SHAPEFILE = os.path.join(DATA_DIR, 'shapefiles', 'ne_10m_admin_0_countries.shp')
+
+
+def neighbors_from_json_file(data_countries, json_DB=JSON_DB):
     neighbors = {}
     with open(json_DB) as json_file:
         countries_dict = json.load(json_file)
     country_names = []
     country_iso = []

     shp.close()
     return countries
 
 
 def replace_empty_neighbours_with_KNN(data_countries, w):
-    shapefile = "../MergeBL-Smith/shapefiles/ne_10m_admin_0_countries.shp"
+    shapefile = SHAPEFILE
     no_neighbors_idx = w.islands
     knn = 10
     wknn = pysal.knnW_from_shapefile(shapefile, knn)
     knn_countries = get_countries_from_shapefile(shapefile)
     neighbors = w.neighbors