tests: allow to run out-of-tree

Signed-off-by: Igor Gnatenko <i.gnatenko.brain@gmail.com>

1 files changed, 8 insertions, 6 deletions

diff --git a/test/test_distance.py b/test/test_distance.py
index d5bce30..d405839 100644
--- a/test/test_distance.py
+++ b/test/test_distance.py
@@ -17,6 +17,8 @@ from numpy.testing import assert_equal, assert_almost_equal, \
 import pyspike as spk
 from pyspike import SpikeTrain
 
+import os
+TEST_PATH = os.path.dirname(os.path.realpath(__file__))
 
 def test_isi():
     # generate two spike trains:
@@ -275,8 +277,8 @@ def test_multi_spike_sync():
                         expected, decimal=15)
 
     # multivariate regression test
-    spike_trains = spk.load_spike_trains_from_txt("test/SPIKE_Sync_Test.txt",
-                                                  edges=[0, 4000])
+    spike_trains = spk.load_spike_trains_from_txt(
+        os.path.join(TEST_PATH, "SPIKE_Sync_Test.txt"), edges=[0, 4000])
     # extract all spike times
     spike_times = np.array([])
     for st in spike_trains:
@@ -352,8 +354,8 @@ def test_regression_spiky():
 
     # multivariate check
 
-    spike_trains = spk.load_spike_trains_from_txt("test/PySpike_testdata.txt",
-                                                  (0.0, 4000.0))
+    spike_trains = spk.load_spike_trains_from_txt(
+        os.path.join(TEST_PATH, "PySpike_testdata.txt"), (0.0, 4000.0))
     isi_dist = spk.isi_distance_multi(spike_trains)
     # get the full precision from SPIKY
     assert_almost_equal(isi_dist, 0.17051816816999129656, decimal=15)
@@ -371,8 +373,8 @@ def test_regression_spiky():
 
 
 def test_multi_variate_subsets():
-    spike_trains = spk.load_spike_trains_from_txt("test/PySpike_testdata.txt",
-                                                  (0.0, 4000.0))
+    spike_trains = spk.load_spike_trains_from_txt(
+        os.path.join(TEST_PATH, "PySpike_testdata.txt"), (0.0, 4000.0))
     sub_set = [1, 3, 5, 7]
     spike_trains_sub_set = [spike_trains[i] for i in sub_set]