diff --git a/tests/test_error_handling.py b/tests/test_error_handling.py index 860c29e9..f0a39827 100644 --- a/tests/test_error_handling.py +++ b/tests/test_error_handling.py @@ -1,5 +1,7 @@ import unittest +from test_case_base import TestCaseBase, strict_mode_guard + import sot @@ -10,11 +12,13 @@ def fn_with_try_except(): raise ValueError("ValueError") except ValueError: print("catch ValueError") + return True -class TestErrorHandling(unittest.TestCase): +class TestErrorHandling(TestCaseBase): + @strict_mode_guard(0) def test_fn_with_try_except(self): - sot.symbolic_translate(fn_with_try_except)() + self.assert_results(fn_with_try_except) if __name__ == "__main__":