Make pickle test self contained

SSAGESLabs · Jan 4, 2024 · 6ef2bde · 6ef2bde
1 parent 934b42f
commit 6ef2bde
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 3 deletions.
diff --git a/tests/test_abf_result.pickle b/tests/test_abf_result.pickle
diff --git a/tests/test_pickle.py b/tests/test_pickle.py
@@ -4,6 +4,7 @@
 
 import dill as pickle
 import numpy as np
+import test_simulations.abf as abf_example
 
 import pysages
 import pysages.colvars
@@ -167,10 +168,9 @@ def test_pickle_colvars():
 
 
 def test_pickle_results():
-    with open("tests/test_abf_result.pickle", "rb") as f:
-        test_result = pickle.load(f)
-
     with tempfile.NamedTemporaryFile() as tmp_pickle:
+        test_result = abf_example.run_simulation(10, write_output=False)
+
         pickle.dump(test_result, tmp_pickle)
         tmp_pickle.flush()
 

diff --git a/tests/test_simulations/__init__.py b/tests/test_simulations/__init__.py
diff --git a/tests/test_simulations/abf.py b/tests/test_simulations/abf.py
@@ -0,0 +1 @@
+../../examples/ase/abf/water.py