Updated test_inner_loop in test_TPI.py

PSLmodels · Sep 13, 2024 · 9130a0a · 9130a0a
1 parent 5f64455
commit 9130a0a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/tests/test_TPI.py b/tests/test_TPI.py
@@ -192,10 +192,11 @@ def test_inner_loop():
     r_p = outer_loop_vars_old[2]
     w = outer_loop_vars_old[1]
     BQ = outer_loop_vars_old[3]
-    TR = outer_loop_vars_old[4]
-    theta = outer_loop_vars_old[5]
+    RM = outer_loop_vars_old[4]
+    TR = outer_loop_vars_old[5]
+    theta = outer_loop_vars_old[6]
     p_m = np.ones((p.T + p.S, p.M))
-    outer_loop_vars = (r_p, r, w, p_m, BQ, TR, theta)
+    outer_loop_vars = (r_p, r, w, p_m, BQ, RM, TR, theta)
     test_tuple = TPI.inner_loop(
         guesses, outer_loop_vars, initial_values, ubi, j, ind, p
     )

diff --git a/tests/test_io_data/tpi_inner_loop_inputs.pkl b/tests/test_io_data/tpi_inner_loop_inputs.pkl