From bfa783bf82e1647ec99fd5b4726d892947a4e0e8 Mon Sep 17 00:00:00 2001
From: "Olivier J.N. Bertrand" <bolirev@hotmail.com>
Date: Sun, 26 Nov 2017 20:57:50 +0100
Subject: [PATCH] Add resource in setup py

---
 navipy/processing/test.py | 4 +++-
 setup.py                  | 4 +++-
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/navipy/processing/test.py b/navipy/processing/test.py
index d507586..1606be7 100644
--- a/navipy/processing/test.py
+++ b/navipy/processing/test.py
@@ -5,6 +5,7 @@ import pandas as pd
 import os
 from navipy.database import database
 import navipy.processing as processing
+import pkg_resources
 
 
 def is_numeric_array(array):
@@ -64,7 +65,8 @@ def build_test_db():
 
 class TestCase(unittest.TestCase):
     def setUp(self):
-        self.mydb_filename = os.path.abspath('navipy/resources/database.db')
+        self.mydb_filename = pkg_resources.resource_filename(
+            'navipy', 'resources/database.db')
         self.mydb = database.DataBaseLoad(self.mydb_filename)
 
 
diff --git a/setup.py b/setup.py
index 0514db7..05847f7 100644
--- a/setup.py
+++ b/setup.py
@@ -29,6 +29,8 @@ setup_dict = {'name': 'navipy',
               'description': 'Insect Navigation Toolbox',
               'packages': create_package_list("navipy"),
               'requires': ['numpy', 'pandas', 'matplotlib', 'scipy'],
-              'install_requires': ["numpy", 'pandas', 'matplotlib', 'scipy']}
+              'install_requires': ["numpy", 'pandas', 'matplotlib', 'scipy'],
+              'package_data': {'navipy': ['resources/database.db']},
+              'include_package_data': True, }
 
 setup(**setup_dict)
-- 
GitLab