use autotui-specific exception

seanbreckenridge · Mar 3, 2024 · 2ec10b5 · 2ec10b5
1 parent 471056e
commit 2ec10b5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_autotui.py b/tests/test_autotui.py
@@ -581,7 +581,7 @@ def test_enum_use_key() -> None:
 
 def test_enum_fails() -> None:
     ds = {"choice": "xx"}
-    with pytest.raises(ValueError, match=r"Could not find xx on Enumeration"):
+    with pytest.raises(AutoTUIException, match=r"Could not find xx on Enumeration"):
         autotui.deserialize_namedtuple(ds, DAT)
 
 
@@ -643,7 +643,7 @@ class UDAT(NamedTuple):
 def test_removing_enum_value() -> None:
     assert autotui.deserialize_namedtuple({"choice": "x"}, UDAT) == UDAT(choice=UEnum.x)
 
-    with pytest.raises(ValueError, match="Could not find z on Enumeration"):
+    with pytest.raises(AutoTUIException, match="Could not find z on Enumeration"):
         autotui.deserialize_namedtuple({"choice": "z"}, UDAT)
 
     with_none = UDAT(choice=None)  # type: ignore