Updated test_TPI.py and test_run_example.py

rickecon · Sep 19, 2024 · 796a779 · 796a779
1 parent 3169b8f
commit 796a779
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/tests/test_TPI.py b/tests/test_TPI.py
@@ -222,7 +222,7 @@ def test_inner_loop():
 filename4 = os.path.join(
     CUR_PATH, "test_io_data", "run_TPI_outputs_reform_baseline_spend.pkl"
 )
-param_updates5 = {"zeta_K": [1.0], "initial_guess_r_SS": 0.10}
+param_updates5 = {"zeta_K": [1.0], "initial_guess_r_SS": 0.04}
 filename5 = os.path.join(
     CUR_PATH, "test_io_data", "run_TPI_outputs_baseline_small_open.pkl"
 )

diff --git a/tests/test_run_example.py b/tests/test_run_example.py
@@ -75,7 +75,10 @@ def test_run_ogcore_example_output(f=call_run_ogcore_example):
     )
     # Delete directory created by run_ogcore_example.py
     roe_output_dir = os.path.join(
-        path.parent, "run_examples", "OUTPUT_BASELINE"
+        path.parent,
+        "run_examples",
+        "OG-Core-Example",
+        "OUTPUT_BASELINE"
     )
     shutil.rmtree(roe_output_dir)