Skip to content

Commit e7fa843

Browse files
committed
Add test
1 parent 724749b commit e7fa843

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

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

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -163,6 +163,12 @@ def test_autoregressive_shared_and_not_shared():
163163
observed_state_names=["data_1", "data_2", "data_3"],
164164
share_states=True,
165165
)
166+
167+
# make sure param_info is correct
168+
# shound't have endog state when share_states is True
169+
assert "endog_" not in shared.param_info["params_shared_ar"]["dims"][0]
170+
assert shared.param_info["sigma_shared_ar"]["dims"] is None
171+
166172
individual = st.AutoregressiveComponent(
167173
order=3,
168174
name="individual_ar",

0 commit comments

Comments
 (0)