diff --git a/navipy/processing/test.py b/navipy/processing/test.py index d507586dfa93cfa14382e9901395e23c3ba242cf..1606be7f8cd0de17b758f3d86fc869d752871263 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 0514db71c994a70c40b6ee811161483f75786abc..05847f7b69666d7590467a15d8e3aff2eb26d091 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)