[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

tests/statespace/core/test_statespace.py

@@ -1241,9 +1241,9 @@ def test_param_dims_coords(ss_mod_multi_component):
             assert dims is None
             continue
         for i, s in zip(shape, dims):
-            assert i == len(ss_mod_multi_component.coords[s]), (
-                f"Mismatch between shape {i} and dimension {s}"
-            )
+            assert i == len(
+                ss_mod_multi_component.coords[s]
+            ), f"Mismatch between shape {i} and dimension {s}"
 
 
 @pytest.mark.filterwarnings("ignore:Provided data contains missing values")

tests/statespace/filters/test_kalman_filter.py

@@ -73,9 +73,9 @@ def test_output_shapes_one_state_one_observed(filter_func, rng):
 
     for output_idx, name in enumerate(output_names):
         expected_output = get_expected_shape(name, p, m, r, n)
-        assert outputs[output_idx].shape == expected_output, (
-            f"Shape of {name} does not match expected"
-        )
+        assert (
+            outputs[output_idx].shape == expected_output
+        ), f"Shape of {name} does not match expected"
 
 
 @pytest.mark.parametrize("filter_func", filter_funcs, ids=filter_names)
@@ -86,9 +86,9 @@ def test_output_shapes_when_all_states_are_stochastic(filter_func, rng):
     outputs = filter_func(*inputs)
     for output_idx, name in enumerate(output_names):
         expected_output = get_expected_shape(name, p, m, r, n)
-        assert outputs[output_idx].shape == expected_output, (
-            f"Shape of {name} does not match expected"
-        )
+        assert (
+            outputs[output_idx].shape == expected_output
+        ), f"Shape of {name} does not match expected"
 
 
 @pytest.mark.parametrize("filter_func", filter_funcs, ids=filter_names)
@@ -99,9 +99,9 @@ def test_output_shapes_when_some_states_are_deterministic(filter_func, rng):
     outputs = filter_func(*inputs)
     for output_idx, name in enumerate(output_names):
         expected_output = get_expected_shape(name, p, m, r, n)
-        assert outputs[output_idx].shape == expected_output, (
-            f"Shape of {name} does not match expected"
-        )
+        assert (
+            outputs[output_idx].shape == expected_output
+        ), f"Shape of {name} does not match expected"
 
 
 @pytest.fixture
@@ -161,9 +161,9 @@ def test_output_shapes_with_time_varying_matrices(f_standard_nd, rng):
 
     for output_idx, name in enumerate(output_names):
         expected_output = get_expected_shape(name, p, m, r, n)
-        assert outputs[output_idx].shape == expected_output, (
-            f"Shape of {name} does not match expected"
-        )
+        assert (
+            outputs[output_idx].shape == expected_output
+        ), f"Shape of {name} does not match expected"
 
 
 @pytest.mark.parametrize("filter_func", filter_funcs, ids=filter_names)
@@ -175,9 +175,9 @@ def test_output_with_deterministic_observation_equation(filter_func, rng):
 
     for output_idx, name in enumerate(output_names):
         expected_output = get_expected_shape(name, p, m, r, n)
-        assert outputs[output_idx].shape == expected_output, (
-            f"Shape of {name} does not match expected"
-        )
+        assert (
+            outputs[output_idx].shape == expected_output
+        ), f"Shape of {name} does not match expected"
 
 
 @pytest.mark.parametrize(
@@ -190,9 +190,9 @@ def test_output_with_multiple_observed(filter_func, filter_name, rng):
     outputs = filter_func(*inputs)
     for output_idx, name in enumerate(output_names):
         expected_output = get_expected_shape(name, p, m, r, n)
-        assert outputs[output_idx].shape == expected_output, (
-            f"Shape of {name} does not match expected"
-        )
+        assert (
+            outputs[output_idx].shape == expected_output
+        ), f"Shape of {name} does not match expected"
 
 
 @pytest.mark.parametrize(
@@ -206,9 +206,9 @@ def test_missing_data(filter_func, filter_name, p, rng):
     outputs = filter_func(*inputs)
     for output_idx, name in enumerate(output_names):
         expected_output = get_expected_shape(name, p, m, r, n)
-        assert outputs[output_idx].shape == expected_output, (
-            f"Shape of {name} does not match expected"
-        )
+        assert (
+            outputs[output_idx].shape == expected_output
+        ), f"Shape of {name} does not match expected"
 
 
 @pytest.mark.parametrize("filter_func", filter_funcs, ids=filter_names)

tests/statespace/models/structural/test_against_statsmodels.py

@@ -67,16 +67,16 @@ def _assert_coord_shapes_match_matrices(mod, params):
     n_shocks = max(1, len(mod.coords[SHOCK_DIM]))
     n_obs = len(mod.coords[OBS_STATE_DIM])
 
-    assert x0.shape[-1:] == (n_states,), (
-        f"x0 expected to have shape (n_states, ), found {x0.shape[-1:]}"
-    )
+    assert x0.shape[-1:] == (
+        n_states,
+    ), f"x0 expected to have shape (n_states, ), found {x0.shape[-1:]}"
     assert P0.shape[-2:] == (
         n_states,
         n_states,
     ), f"P0 expected to have shape (n_states, n_states), found {P0.shape[-2:]}"
-    assert c.shape[-1:] == (n_states,), (
-        f"c expected to have shape (n_states, ), found {c.shape[-1:]}"
-    )
+    assert c.shape[-1:] == (
+        n_states,
+    ), f"c expected to have shape (n_states, ), found {c.shape[-1:]}"
     assert d.shape[-1:] == (n_obs,), f"d expected to have shape (n_obs, ), found {d.shape[-1:]}"
     assert T.shape[-2:] == (
         n_states,
@@ -107,9 +107,9 @@ def _assert_keys_match(test_dict, expected_dict):
     assert len(key_diff) == 0, f"{', '.join(key_diff)} were not found in the test_dict keys."
 
     key_diff = set(param_keys) - set(expected_keys)
-    assert len(key_diff) == 0, (
-        f"{', '.join(key_diff)} were keys of the tests_dict not in expected_dict."
-    )
+    assert (
+        len(key_diff) == 0
+    ), f"{', '.join(key_diff)} were keys of the tests_dict not in expected_dict."
 
 
 def _assert_param_dims_correct(param_dims, expected_dims):