reshape arrays for test tax

PSLmodels · Jan 18, 2024 · a710dad · a710dad
1 parent e2f060d
commit a710dad
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/tests/test_tax.py b/tests/test_tax.py
@@ -17,6 +17,7 @@
     "J": 1,
     "T": 4,
     "eta": (np.ones((4, 1)) / (4 * 1)),
+    "e": np.ones((4, 1)),
 }
 p.update_specifications(new_param_values)
 p.retire = [3, 3, 3, 3, 3, 3, 3, 3]
@@ -78,6 +79,7 @@ def test_replacement_rate_vals(n, w, factor, j, p_in, expected):
     "lambdas": [1.0],
     "J": 1,
     "T": 3,
+    "e": np.ones((3, 1)),
     "eta": (np.ones((3, 1)) / (3 * 1)),
     "h_wealth": [2],
     "p_wealth": [3],
@@ -94,6 +96,7 @@ def test_replacement_rate_vals(n, w, factor, j, p_in, expected):
     "lambdas": [1.0],
     "J": 1,
     "T": 3,
+    "e": np.ones((3, 1)),
     "eta": (np.ones((3, 1)) / (3 * 1)),
     "h_wealth": [1.2, 1.1, 2.3],
     "p_wealth": [2.2, 2.3, 1.8],
@@ -127,6 +130,7 @@ def test_ETR_wealth(b, p, expected):
     "lambdas": [1.0],
     "J": 1,
     "T": 3,
+    "e": np.ones((3, 1)),
     "eta": (np.ones((3, 1)) / (3 * 1)),
     "h_wealth": [3],
     "p_wealth": [4],
@@ -144,6 +148,7 @@ def test_ETR_wealth(b, p, expected):
     "lambdas": [1.0],
     "J": 1,
     "T": 3,
+    "e": np.ones((3, 1)),
     "eta": (np.ones((3, 1)) / (3 * 1)),
     "h_wealth": [1.2, 1.1, 2.3],
     "p_wealth": [2.2, 2.3, 1.8],
@@ -878,6 +883,7 @@ def test_MTR_income(etr_params, mtr_params, params, mtr_capital, expected):
     "inv_tax_credit": [[0.02]],
     "T": 3,
     "S": 3,
+    "e": np.ones((3, p1.J)),
     "rho": rho_vec.tolist(),
     "eta": (np.ones((3, p1.J)) / (3 * p1.J)),
     "labor_income_tax_noncompliance_rate": [[0.0]],
@@ -1206,6 +1212,7 @@ def test_get_biz_tax(w, Y, L, K, p_m, p, m, method, expected):
     "T": 3,
     "S": 3,
     "J": 2,
+    "e": np.ones((3, 2)),
     "rho": rho_vec.tolist(),
     "lambdas": [0.65, 0.35],
     "eta": (np.ones((3, 2)) / (3 * 2)),