update setup.py

mauryaland · Aug 28, 2017 · d42f85d · d42f85d
1 parent 687e91d
commit d42f85d
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/setup.py b/setup.py
@@ -35,9 +35,12 @@
 data_files = []
 
 for item in os.listdir("geopandas/datasets"):
-    if os.path.isdir(os.path.join("geopandas/datasets/", item)) \
-            and not item.startswith('__'):
-        data_files.append(os.path.join("datasets", item, '*'))
+    if not item.startswith('__'):
+        if os.path.isdir(os.path.join("geopandas/datasets/", item)):
+            data_files.append(os.path.join("datasets", item, '*'))
+        elif item.endswith('.zip'):
+            data_files.append(os.path.join("datasets", item))
+
 
 setup(name='geopandas',
       version=versioneer.get_version(),
@@ -48,7 +51,8 @@
       url='http://geopandas.org',
       long_description=LONG_DESCRIPTION,
       packages=['geopandas', 'geopandas.io', 'geopandas.tools',
-                'geopandas.datasets'],
+                'geopandas.datasets',
+                'geopandas.tests', 'geopandas.tools.tests'],
       package_data={'geopandas': data_files},
       install_requires=INSTALL_REQUIRES,
       cmdclass=versioneer.get_cmdclass())