TST: captured ParseIqTreeError

YapengLang · Oct 18, 2024 · fbb3394 · fbb3394
1 parent 84b723b
commit fbb3394
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/test_iqtree/test_tree_yaml.py b/tests/test_iqtree/test_tree_yaml.py
@@ -143,15 +143,15 @@ 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(KeyError):
+    with pytest.raises(ParseIqTreeError):
         _ = _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(KeyError):
+    with pytest.raises(ParseIqTreeError):
         _ = _process_tree_yaml(non_lie_dna_with_rate_model, ["a", "b", "c", "d"])
 
 
@@ -171,5 +171,5 @@ 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(KeyError):
+    with pytest.raises(ParseIqTreeError):
         _ = _process_tree_yaml(lie_dna_model, ["a", "b", "c", "d"])