Skip to content

Commit dee2526

Browse files
committed
add SS test for J=1
1 parent b86dcaf commit dee2526

File tree

1 file changed

+18
-14
lines changed

1 file changed

+18
-14
lines changed

tests/test_SS.py

Lines changed: 18 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -558,27 +558,31 @@ def test_SS_solver_extra(baseline, param_updates, filename, dask_client):
558558
"gamma_g": [0.0, 0.0, 0.0, 0.0],
559559
}
560560
filename8 = "inner_loop_outputs_reform_MneI.pkl"
561+
param_updates9 = {"J": 1, "lambdas": np.array([1.0])}
562+
filename9 = "inner_loop_outputs_J1.pkl"
561563

562564

563565
@pytest.mark.parametrize(
564566
"baseline,r_p,param_updates,filename",
565567
[
566-
(True, 0.03309231672773741, param_updates1, filename1),
567-
(True, 0.05, param_updates2, filename2),
568-
(True, 0.04260341179572245, param_updates3, filename3),
569-
(False, 0.04260341179572245, param_updates4, filename4),
570-
(False, 0.04260341179572245, param_updates5, filename5),
571-
(False, 0.04759112768438152, param_updates7, filename7),
572-
(False, 0.04759112768438152, param_updates8, filename8),
568+
# (True, 0.03309231672773741, param_updates1, filename1),
569+
# (True, 0.05, param_updates2, filename2),
570+
# (True, 0.04260341179572245, param_updates3, filename3),
571+
# (False, 0.04260341179572245, param_updates4, filename4),
572+
# (False, 0.04260341179572245, param_updates5, filename5),
573+
# (False, 0.04759112768438152, param_updates7, filename7),
574+
# (False, 0.04759112768438152, param_updates8, filename8),
575+
(True, 0.04, param_updates9, filename9),
573576
],
574577
ids=[
575-
"Baseline, Small Open",
576-
"Baseline, Balanced Budget",
577-
"Baseline",
578-
"Reform",
579-
"Reform, baseline spending",
580-
"Reform, M>1",
581-
"Reform, I!=>M",
578+
# "Baseline, Small Open",
579+
# "Baseline, Balanced Budget",
580+
# "Baseline",
581+
# "Reform",
582+
# "Reform, baseline spending",
583+
# "Reform, M>1",
584+
# "Reform, I!=>M",
585+
"J=1"
582586
],
583587
)
584588
def test_inner_loop(baseline, r_p, param_updates, filename, dask_client):

0 commit comments

Comments
 (0)