tests: allow to run out-of-tree

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

3 files changed, 19 insertions, 16 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]
 
diff --git a/test/test_regression/test_regression_15.py b/test/test_regression/test_regression_15.py
index 42a39ea..dcacae2 100644
--- a/test/test_regression/test_regression_15.py
+++ b/test/test_regression/test_regression_15.py
@@ -16,11 +16,13 @@ from numpy.testing import assert_equal, assert_almost_equal, \
 
 import pyspike as spk
 
+import os
+TEST_PATH = os.path.dirname(os.path.realpath(__file__))
+TEST_DATA = os.path.join(TEST_PATH, "..", "SPIKE_Sync_Test.txt")
 
 def test_regression_15_isi():
     # load spike trains
-    spike_trains = spk.load_spike_trains_from_txt("test/SPIKE_Sync_Test.txt",
-                                                  edges=[0, 4000])
+    spike_trains = spk.load_spike_trains_from_txt(TEST_DATA, edges=[0, 4000])
 
     N = len(spike_trains)
 
@@ -38,8 +40,7 @@ def test_regression_15_isi():
 
 def test_regression_15_spike():
     # load spike trains
-    spike_trains = spk.load_spike_trains_from_txt("test/SPIKE_Sync_Test.txt",
-                                                  edges=[0, 4000])
+    spike_trains = spk.load_spike_trains_from_txt(TEST_DATA, edges=[0, 4000])
 
     N = len(spike_trains)
 
@@ -57,8 +58,7 @@ def test_regression_15_spike():
 
 def test_regression_15_sync():
     # load spike trains
-    spike_trains = spk.load_spike_trains_from_txt("test/SPIKE_Sync_Test.txt",
-                                                  edges=[0, 4000])
+    spike_trains = spk.load_spike_trains_from_txt(TEST_DATA, edges=[0, 4000])
 
     N = len(spike_trains)
 
diff --git a/test/test_spikes.py b/test/test_spikes.py
index d4eb131..609a819 100644
--- a/test/test_spikes.py
+++ b/test/test_spikes.py
@@ -13,10 +13,12 @@ from numpy.testing import assert_equal
 
 import pyspike as spk
 
+import os
+TEST_PATH = os.path.dirname(os.path.realpath(__file__))
+TEST_DATA = os.path.join(TEST_PATH, "PySpike_testdata.txt")
 
 def test_load_from_txt():
-    spike_trains = spk.load_spike_trains_from_txt("test/PySpike_testdata.txt",
-                                                  edges=(0, 4000))
+    spike_trains = spk.load_spike_trains_from_txt(TEST_DATA, edges=(0, 4000))
     assert len(spike_trains) == 40
 
     # check the first spike train
@@ -48,8 +50,7 @@ def check_merged_spikes(merged_spikes, spike_trains):
 
 def test_merge_spike_trains():
     # first load the data
-    spike_trains = spk.load_spike_trains_from_txt("test/PySpike_testdata.txt",
-                                                  edges=(0, 4000))
+    spike_trains = spk.load_spike_trains_from_txt(TEST_DATA, edges=(0, 4000))
 
     merged_spikes = spk.merge_spike_trains([spike_trains[0], spike_trains[1]])
     # test if result is sorted