|
14 | 14 |
|
15 | 15 |
|
16 | 16 | @pytest.mark.parametrize("s", [10, 25, 50]) |
17 | | -@pytest.mark.parametrize("d", [1, 2, 3]) |
| 17 | +@pytest.mark.parametrize("d", [1, 3]) |
18 | 18 | @pytest.mark.parametrize("innovations", [True, False]) |
19 | 19 | @pytest.mark.parametrize("remove_first_state", [True, False]) |
20 | 20 | @pytest.mark.filterwarnings( |
@@ -54,7 +54,7 @@ def random_word(rng): |
54 | 54 | assert mod.coords["state_season"] == state_names[test_slice] |
55 | 55 |
|
56 | 56 |
|
57 | | -@pytest.mark.parametrize("d", [1, 2, 3]) |
| 57 | +@pytest.mark.parametrize("d", [1, 3]) |
58 | 58 | @pytest.mark.parametrize( |
59 | 59 | "remove_first_state", [True, False], ids=["remove_first_state", "keep_first_state"] |
60 | 60 | ) |
@@ -146,8 +146,7 @@ def test_time_seasonality_multiple_observed(rng, d, remove_first_state): |
146 | 146 | np.testing.assert_allclose(matrix, expected) |
147 | 147 |
|
148 | 148 |
|
149 | | -@pytest.mark.parametrize("d1", [1, 2, 3]) |
150 | | -@pytest.mark.parametrize("d2", [1, 2, 3]) |
| 149 | +@pytest.mark.parametrize("d1, d2", [(1, 1), (1, 3), (3, 1), (3, 3)]) |
151 | 150 | def test_add_two_time_seasonality_different_observed(rng, d1, d2): |
152 | 151 | mod1 = st.TimeSeasonality( |
153 | 152 | season_length=3, |
|
0 commit comments