TST: add match parameter to capture ParseIqTreeError explicitly

YapengLang · Nov 8, 2024 · cca1aa8 · cca1aa8
1 parent 9262900
commit cca1aa8
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/tests/test_iqtree/test_tree_yaml.py b/tests/test_iqtree/test_tree_yaml.py
@@ -186,18 +186,19 @@ def test_non_lie_dna_model_motif_absent(
     non_lie_dna_with_rate_model: dict[str, Any],
 ) -> None:
     non_lie_dna_with_rate_model["ModelDNA"].pop("state_freq")
-    with pytest.raises(ParseIqTreeError):
+    with pytest.raises(ParseIqTreeError, match=re.escape("IQ-TREE output malformated, motif parameters not found.")):
         _ = _process_tree_yaml(non_lie_dna_with_rate_model, ["a", "b", "c", "d"])
 
 
 def test_non_lie_dna_model_rate_absent(
     non_lie_dna_with_rate_model: dict[str, Any],
 ) -> None:
     non_lie_dna_with_rate_model["ModelDNA"].pop("rates")
-    with pytest.raises(ParseIqTreeError):
+    with pytest.raises(ParseIqTreeError,match=re.escape("IQ-TREE output malformated, rate parameters not found.")):
         _ = _process_tree_yaml(non_lie_dna_with_rate_model, ["a", "b", "c", "d"])
 
 
+
 def test_lie_dna_model(
     lie_dna_model: dict[str, Any],
 ) -> None:
@@ -214,7 +215,7 @@ def test_lie_dna_model_motif_absent(
     lie_dna_model: dict[str, Any],
 ) -> None:
     lie_dna_model["ModelLieMarkovRY2.2b"].pop("state_freq")
-    with pytest.raises(ParseIqTreeError):
+    with pytest.raises(ParseIqTreeError, match=re.escape("IQ-TREE output malformated, motif parameters not found.")):
         _ = _process_tree_yaml(lie_dna_model, ["a", "b", "c", "d"])