Fix data dir for pytest

CovertLab · Dec 5, 2024 · b793f56 · b793f56
1 parent b2f25fd
commit b793f56
Show file tree

Hide file tree

Showing 8 changed files with 6 additions and 2 deletions.
diff --git a/data/complexation/beginning-final.json → test/data/complexation/beginning-final.json b/data/complexation/beginning-final.json → test/data/complexation/beginning-final.json
diff --git a/data/complexation/beginning-initial.json → .../data/complexation/beginning-initial.json b/data/complexation/beginning-initial.json → .../data/complexation/beginning-initial.json
diff --git a/data/complexation/large-initial.json → test/data/complexation/large-initial.json b/data/complexation/large-initial.json → test/data/complexation/large-initial.json
diff --git a/data/complexation/simple-final.json → test/data/complexation/simple-final.json b/data/complexation/simple-final.json → test/data/complexation/simple-final.json
diff --git a/data/complexation/simple-initial.json → test/data/complexation/simple-initial.json b/data/complexation/simple-initial.json → test/data/complexation/simple-initial.json
diff --git a/data/complexation/stoichiometry.json → test/data/complexation/stoichiometry.json b/data/complexation/stoichiometry.json → test/data/complexation/stoichiometry.json
diff --git a/test/negative_counts.py b/test/negative_counts.py
@@ -1,12 +1,16 @@
 import json
+import os
 
 from stochastic_arrow import StochasticSystem
 import numpy as np
 
 
 duration = 2**31
 
-with open('data/complexation/large-initial.json') as f:
+# Build the path to the data file relative to the script's directory
+data_path = os.path.join(os.path.dirname(__file__), 'data', 'complexation', 'large-initial.json')
+
+with open(data_path) as f:
     data = json.load(f)
 
 stoich = np.array(data['stoich'])

diff --git a/test/test_arrow.py b/test/test_arrow.py
@@ -84,7 +84,7 @@ def test_dimerization():
 
 
 def load_complexation(prefix='simple'):
-    fixtures_root = os.path.join('data', 'complexation')
+    fixtures_root = os.path.join(os.path.dirname(__file__), 'data', 'complexation')
 
     def load_state(filename):
         with open(os.path.join(fixtures_root, filename)) as f: