Patch test_fittingfunc.py export_path

sarahshi · Feb 27, 2024 · dd7ed6f · dd7ed6f
1 parent 7bc6cb0
commit dd7ed6f
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/UnitTests/test_fittingfunc.py b/UnitTests/test_fittingfunc.py
@@ -204,9 +204,12 @@ def test_MCMC_exportpath(self):
             paths[dir_name] = full_path
             os.makedirs(full_path, exist_ok=True)
 
-        # Construct file paths
         fpath = os.path.join(paths["FIGURES"], "")
-        ppath = os.path.join(paths["PLOTFILES"], "")
+
+        add_dirs = ["TRACE", "HISTOGRAM", "PAIRWISE", "MODELFIT"]
+        ppath = paths["PLOTFILES"]
+        for add_dir in add_dirs:
+            os.makedirs(os.path.join(ppath, add_dir), exist_ok=True)
 
         Volatiles_DF, _ = pig.calculate_baselines(
             self.dfs_dict, temp_export_path)