Skip to content

Commit 4dcfc99

Browse files
committed
Fix cycle tests after param renaming
1 parent 753823c commit 4dcfc99

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

tests/statespace/models/structural/components/test_cycle.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -122,9 +122,9 @@ def test_multivariate_cycle_with_shared(rng):
122122

123123
assert cycle.state_names == ["Cos_cycle[shared]", "Sin_cycle[shared]"]
124124
assert cycle.shock_names == []
125-
assert cycle.param_names == ["cycle"]
125+
assert cycle.param_names == ["params_cycle"]
126126

127-
params = {"cycle": np.array([1.0, 2.0], dtype=config.floatX)}
127+
params = {"params_cycle": np.array([1.0, 2.0], dtype=config.floatX)}
128128
x, y = simulate_from_numpy_model(cycle, rng, params, steps=12 * 12)
129129

130130
np.testing.assert_allclose(y[:, 0], y[:, 1], atol=ATOL, rtol=RTOL)
@@ -351,9 +351,9 @@ def test_add_multivariate_shared_and_not_shared():
351351
assert mod.shock_names == expected_states[:2]
352352

353353
assert mod.param_names == [
354-
"shared_cycle",
354+
"params_shared_cycle",
355355
"sigma_shared_cycle",
356-
"individual_cycle",
356+
"params_individual_cycle",
357357
"length_individual_cycle",
358358
"dampening_factor_individual_cycle",
359359
"P0",
@@ -364,17 +364,17 @@ def test_add_multivariate_shared_and_not_shared():
364364
assert mod.coords["state_individual_cycle"] == ["Cos_individual_cycle", "Sin_individual_cycle"]
365365
assert mod.coords["endog_individual_cycle"] == ["gdp", "inflation", "unemployment"]
366366

367-
assert mod.param_info["shared_cycle"]["dims"] == ("state_shared_cycle",)
368-
assert mod.param_info["shared_cycle"]["shape"] == (2,)
367+
assert mod.param_info["params_shared_cycle"]["dims"] == ("state_shared_cycle",)
368+
assert mod.param_info["params_shared_cycle"]["shape"] == (2,)
369369

370370
assert mod.param_info["sigma_shared_cycle"]["dims"] is None
371371
assert mod.param_info["sigma_shared_cycle"]["shape"] == ()
372372

373-
assert mod.param_info["individual_cycle"]["dims"] == (
373+
assert mod.param_info["params_individual_cycle"]["dims"] == (
374374
"endog_individual_cycle",
375375
"state_individual_cycle",
376376
)
377-
assert mod.param_info["individual_cycle"]["shape"] == (3, 2)
377+
assert mod.param_info["params_individual_cycle"]["shape"] == (3, 2)
378378

379379
params = {
380380
"length_individual_cycle": 12.0,

0 commit comments

Comments
 (0)