temp change for testing

Author: jdebacker

tests/test_TPI.py

@@ -507,30 +507,30 @@ def test_inner_loop():
 @pytest.mark.parametrize(
     "baseline,param_updates,filename",
     [
-        (True, param_updates2, filename2),
+        # (True, param_updates2, filename2),
         (True, {"initial_guess_r_SS": 0.035}, filename1),
-        (False, {}, filename3),
-        (False, param_updates4, filename4),
-        (True, param_updates5, filename5),
-        (True, param_updates6, filename6),
-        (True, param_updates7, filename7),
-        (True, param_updates8, filename8),
-        (True, param_updates9, filename9),
-        (True, param_updates10, filename10),
-        (True, param_updates11, filename11),
+        # (False, {}, filename3),
+        # (False, param_updates4, filename4),
+        # (True, param_updates5, filename5),
+        # (True, param_updates6, filename6),
+        # (True, param_updates7, filename7),
+        # (True, param_updates8, filename8),
+        # (True, param_updates9, filename9),
+        # (True, param_updates10, filename10),
+        # (True, param_updates11, filename11),
     ],
     ids=[
-        "Baseline, balanced budget",
+        # "Baseline, balanced budget",
         "Baseline",
-        "Reform",
-        "Reform, baseline spending",
-        "Baseline, small open",
-        "Baseline, small open some periods",
-        "Baseline, delta_tau = 0",
-        "Baseline, Kg > 0",
-        "Baseline, M=3 non-zero Kg",
-        "Baseline, M=3 zero Kg",
-        "Baseline, M!=I",
+        # "Reform",
+        # "Reform, baseline spending",
+        # "Baseline, small open",
+        # "Baseline, small open some periods",
+        # "Baseline, delta_tau = 0",
+        # "Baseline, Kg > 0",
+        # "Baseline, M=3 non-zero Kg",
+        # "Baseline, M=3 zero Kg",
+        # "Baseline, M!=I",
     ],
 )
 def test_run_TPI_full_run(
@@ -580,8 +580,8 @@ def test_run_TPI_full_run(
 
     # Need to run SS first to get results
     SS.ENFORCE_SOLUTION_CHECKS = True
-    # ss_outputs = SS.run_SS(p, client=dask_client)
-    ss_outputs = SS.run_SS(p, client=None)
+    ss_outputs = SS.run_SS(p, client=dask_client)
+    # ss_outputs = SS.run_SS(p, client=None)
 
     if p.baseline:
         utils.mkdirs(os.path.join(p.baseline_dir, "SS"))
@@ -622,8 +622,6 @@ def test_run_TPI_full_run(
                     - v[: p.T, :, :]
                 ).max(),
             )
-    # save test results to disk
-    pickle.dump(test_dict, open("TPI_TESTING_vars.pkl", "wb"))
 
     for k, v in expected_dict.items():
         print("Testing, ", k)
@@ -632,6 +630,15 @@ def test_run_TPI_full_run(
                 "Diff = ",
                 np.abs(test_dict[VAR_NAME_MAPPING[k]][: p.T] - v[: p.T]).max(),
             )
+            if np.abs(
+                test_dict[VAR_NAME_MAPPING[k]][: p.T] - v[: p.T]
+            ).max() > 1e-5:
+                # print the indices where the difference is large
+                indices = np.where(
+                    np.abs(test_dict[VAR_NAME_MAPPING[k]][: p.T] - v[: p.T])
+                    > 1e-5
+                )
+                print("Indices with large difference:", indices)
             assert np.allclose(
                 test_dict[VAR_NAME_MAPPING[k]][: p.T],
                 v[: p.T],
@@ -646,6 +653,16 @@ def test_run_TPI_full_run(
                     - v[: p.T, :, :]
                 ).max(),
             )
+            if np.abs(
+                test_dict[VAR_NAME_MAPPING[k]][: p.T, :, :] - v[: p.T, :, :]
+            ).max() > 1e-5:
+                # print the indices where the difference is large
+                indices = np.where(
+                    np.abs(
+                        test_dict[VAR_NAME_MAPPING[k]][: p.T, :, :] - v[: p.T, :, :]
+                    ) > 1e-5
+                )
+                print("Indices with large difference:", indices)
             assert np.allclose(
                 test_dict[VAR_NAME_MAPPING[k]][: p.T, :, :],
                 v[: p.T, :, :],