add strict guard

PaddlePaddle · Oct 9, 2023 · ee30b93 · ee30b93
1 parent 6420ae0
commit ee30b93
Showing 1 changed file with 6 additions and 2 deletions.
diff --git 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__":