black linting and RegEx alteration

HopkinsIDD · Nov 8, 2024 · e2a87c3 · e2a87c3
1 parent a6c3585
commit e2a87c3
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/flepimop/gempyor_pkg/tests/seir/test_model_info.py b/flepimop/gempyor_pkg/tests/seir/test_model_info.py
@@ -44,7 +44,10 @@ def test_ModelInfo_init_tf_is_ahead_of_ti_fail(self):
         config.read(user=False)
         config.set_file(f"{DATA_DIR}/config_test.yml")
         config["start_date"] = "2022-01-02"
-        with pytest.raises(ValueError, match=r"`tf`\s?\(time to finish\)\s?is less than or equal to\s?`ti`\s?\(time to start\)\."):
+        with pytest.raises(
+            ValueError,
+            match=r"`tf`\s?\(time to finish\)\s?is less than or equal to\s?`ti`\s?\(time to start\)\.",
+        ):
             s = ModelInfo(
                 config=config,
                 seir_modifiers_scenario=None,

diff --git a/flepimop/gempyor_pkg/tests/seir/test_seir.py b/flepimop/gempyor_pkg/tests/seir/test_seir.py
@@ -128,7 +128,7 @@ def test_constant_population_legacy_integration():
 
 
 def test_constant_population_rk4jit_integration_fail():
-    with pytest.raises(ValueError, match=r".*with.*method.*integration.*"):
+    with pytest.raises(ValueError, match=r".*With.*method.*integration.*"):
         config.set_file(f"{DATA_DIR}/config.yml")
 
         first_sim_index = 1