fixing unit test for suffix

StochSS · Aug 20, 2024 · 30267b2 · 30267b2
1 parent 252d895
commit 30267b2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/gillespy2/core/results.py b/gillespy2/core/results.py
@@ -283,7 +283,7 @@ def build_from_solver_results(cls, solver, live_output_options):
             return results
         raise ValueError("number_of_trajectories must be non-negative and non-zero")
 
-    def to_csv(self, path=".", nametag=None, stamp=None, postfix=".odf", verbose=False):
+    def to_csv(self, path=".", nametag=None, stamp=None, suffix=".odf", verbose=False):
         """
         Outputs the Results to one or more .csv files in a new directory.
 
@@ -311,7 +311,7 @@ def to_csv(self, path=".", nametag=None, stamp=None, postfix=".odf", verbose=Fal
         else:
             identifier = nametag
 
-        directory = os.path.join(path, f"{identifier}-{stamp}{postfix}")
+        directory = os.path.join(path, f"{identifier}-{stamp}{suffix}")
         if verbose:
             print(f"Writing data to: {directory}")
         # multiple trajectories

diff --git a/test/test_results.py b/test/test_results.py
@@ -120,8 +120,8 @@ def test_to_csv_single_result_directory_exists(self):
 
         with tempfile.TemporaryDirectory() as tempdir:
             result.to_csv(stamp = test_stamp, nametag = test_nametag, path=tempdir)
-            test_path = tempdir+"/"+test_nametag+test_stamp
-            assert os.path.isdir(test_path)
+            test_path = tempdir+"/"+test_nametag+test_stamp+".odf"
+            self.assertTrue(os.path.isdir(test_path))
 
     def test_to_csv_single_result_file_exists(self):
         test_data = {'time':[0]}