Fixed relative pathing issues for tests

peekxc · Dec 4, 2023 · f19232e · f19232e
1 parent 6aee229
commit f19232e
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 8 deletions.
diff --git a/tests/test_lanczos.py b/tests/test_lanczos.py
@@ -8,8 +8,8 @@
 ## Add the test directory to the sys path 
 import sys
 import primate
-ind = [i for i, c in enumerate(primate.__file__) if c == '/'][-3]
-sys.path.insert(0, primate.__file__[:ind] + '/tests')
+rel_dir = primate.__file__[:(primate.__file__.find('primate') + 7)]
+sys.path.insert(0, rel_dir + '/tests')
 
 def test_basic():
   from primate.diagonalize import lanczos

diff --git a/tests/test_random.py b/tests/test_random.py
@@ -6,8 +6,8 @@
 ## Add the test directory to the sys path 
 import sys
 import primate
-ind = [i for i, c in enumerate(primate.__file__) if c == '/'][-3]
-sys.path.insert(0, primate.__file__[:ind] + '/tests')
+rel_dir = primate.__file__[:(primate.__file__.find('primate') + 7)]
+sys.path.insert(0, rel_dir + '/tests')
 
 def test_seeding():
   s1 = random.rademacher(250, seed = -1)

diff --git a/tests/test_slq.py b/tests/test_slq.py
@@ -8,8 +8,8 @@
 ## Add the test directory to the sys path 
 import sys
 import primate
-ind = [i for i, c in enumerate(primate.__file__) if c == '/'][-3]
-sys.path.insert(0, primate.__file__[:ind] + '/tests')
+rel_dir = primate.__file__[:(primate.__file__.find('primate') + 7)]
+sys.path.insert(0, rel_dir + '/tests')
 
 def test_stochastic_quadrature():
   from primate.diagonalize import _lanczos

diff --git a/tests/test_stats.py b/tests/test_stats.py
@@ -7,8 +7,8 @@
 ## Add the test directory to the sys path 
 import sys
 import primate
-ind = [i for i, c in enumerate(primate.__file__) if c == '/'][-3]
-sys.path.insert(0, primate.__file__[:ind] + '/tests')
+rel_dir = primate.__file__[:(primate.__file__.find('primate') + 7)]
+sys.path.insert(0, rel_dir + '/tests')
 
 def test_incremental_est():
   np.random.seed(1234)