remove unnecessary data files in tests and robust data-path searching

plotly · Kully · Feb 20, 2018 · Feb 19, 2018 · Feb 19, 2018 · Feb 19, 2018
commit 065e59c385b09634d164b76dadb8cb7f37988386
diff --git a/plotly/figure_factory/_county_choropleth.py b/plotly/figure_factory/_county_choropleth.py
@@ -4,6 +4,7 @@
 
 import io
 import numpy as np
+import os
 import pandas as pd
 import warnings
 
@@ -19,17 +20,23 @@
 
 def _create_us_counties_df(st_to_state_name_dict, state_to_st_dict):
     # URLS
-    data_url = 'plotly/package_data/data/'
+    abs_file_path = os.path.realpath(__file__)
+    abs_dir_path = os.path.dirname(abs_file_path)
+
+    abs_plotly_dir_path = abs_dir_path[:abs_dir_path.find('/figure_factory')]
+    abs_data_dir_path = abs_plotly_dir_path + '/package_data/data/'
+
+    print abs_data_dir_path
 
     shape_pre2010 = 'gz_2010_us_050_00_500k/gz_2010_us_050_00_500k.shp'
-    shape_pre2010 = data_url + shape_pre2010
+    shape_pre2010 = abs_data_dir_path + shape_pre2010
     df_shape_pre2010 = gp.read_file(shape_pre2010)
     df_shape_pre2010['FIPS'] = (df_shape_pre2010['STATE'] +
                                 df_shape_pre2010['COUNTY'])
     df_shape_pre2010['FIPS'] = pd.to_numeric(df_shape_pre2010['FIPS'])
 
     states_path = 'cb_2016_us_state_500k/cb_2016_us_state_500k.shp'
-    states_path = data_url + states_path
+    states_path = abs_data_dir_path + states_path
 
     # state df
     df_state = gp.read_file(states_path)

diff --git a/...figure_factory/plotly/package_data/data/cb_2016_us_county_500k/cb_2016_us_county_500k.cpg b/...figure_factory/plotly/package_data/data/cb_2016_us_county_500k/cb_2016_us_county_500k.cpg
diff --git a/...figure_factory/plotly/package_data/data/cb_2016_us_county_500k/cb_2016_us_county_500k.dbf b/...figure_factory/plotly/package_data/data/cb_2016_us_county_500k/cb_2016_us_county_500k.dbf
diff --git a/...figure_factory/plotly/package_data/data/cb_2016_us_county_500k/cb_2016_us_county_500k.prj b/...figure_factory/plotly/package_data/data/cb_2016_us_county_500k/cb_2016_us_county_500k.prj
diff --git a/...figure_factory/plotly/package_data/data/cb_2016_us_county_500k/cb_2016_us_county_500k.shp b/...figure_factory/plotly/package_data/data/cb_2016_us_county_500k/cb_2016_us_county_500k.shp
diff --git a/...ory/plotly/package_data/data/cb_2016_us_county_500k/cb_2016_us_county_500k.shp.ea.iso.xml b/...ory/plotly/package_data/data/cb_2016_us_county_500k/cb_2016_us_county_500k.shp.ea.iso.xml
diff --git a/...actory/plotly/package_data/data/cb_2016_us_county_500k/cb_2016_us_county_500k.shp.iso.xml b/...actory/plotly/package_data/data/cb_2016_us_county_500k/cb_2016_us_county_500k.shp.iso.xml
diff --git a/...re_factory/plotly/package_data/data/cb_2016_us_county_500k/cb_2016_us_county_500k.shp.xml b/...re_factory/plotly/package_data/data/cb_2016_us_county_500k/cb_2016_us_county_500k.shp.xml
diff --git a/...figure_factory/plotly/package_data/data/cb_2016_us_county_500k/cb_2016_us_county_500k.shx b/...figure_factory/plotly/package_data/data/cb_2016_us_county_500k/cb_2016_us_county_500k.shx
diff --git a/...t_figure_factory/plotly/package_data/data/cb_2016_us_state_500k/cb_2016_us_state_500k.cpg b/...t_figure_factory/plotly/package_data/data/cb_2016_us_state_500k/cb_2016_us_state_500k.cpg
diff --git a/...t_figure_factory/plotly/package_data/data/cb_2016_us_state_500k/cb_2016_us_state_500k.prj b/...t_figure_factory/plotly/package_data/data/cb_2016_us_state_500k/cb_2016_us_state_500k.prj
diff --git a/...ctory/plotly/package_data/data/cb_2016_us_state_500k/cb_2016_us_state_500k.shp.ea.iso.xml b/...ctory/plotly/package_data/data/cb_2016_us_state_500k/cb_2016_us_state_500k.shp.ea.iso.xml
diff --git a/..._factory/plotly/package_data/data/cb_2016_us_state_500k/cb_2016_us_state_500k.shp.iso.xml b/..._factory/plotly/package_data/data/cb_2016_us_state_500k/cb_2016_us_state_500k.shp.iso.xml
diff --git a/...gure_factory/plotly/package_data/data/cb_2016_us_state_500k/cb_2016_us_state_500k.shp.xml b/...gure_factory/plotly/package_data/data/cb_2016_us_state_500k/cb_2016_us_state_500k.shp.xml
diff --git a/...figure_factory/plotly/package_data/data/gz_2010_us_050_00_500k/gz_2010_us_050_00_500k.prj b/...figure_factory/plotly/package_data/data/gz_2010_us_050_00_500k/gz_2010_us_050_00_500k.prj
diff --git a/...figure_factory/plotly/package_data/data/gz_2010_us_050_00_500k/gz_2010_us_050_00_500k.xml b/...figure_factory/plotly/package_data/data/gz_2010_us_050_00_500k/gz_2010_us_050_00_500k.xml