Follow comments, check exception message

8 years ago · ba4b0291ef
parent 680c20217e
commit ba4b0291ef
1 changed files with 7 additions and 2 deletions
--- a/python/paddle/v2/framework/tests/test_exception.py
+++ b/python/paddle/v2/framework/tests/test_exception.py
@ -4,8 +4,13 @@ import unittest

 class TestException(unittest.TestCase):
    def test_exception(self):
-        self.assertRaises(core.EnforceNotMet,
-                          lambda: core.__unittest_throw_exception__())
+        ex = None
+        try:
+            core.__unittest_throw_exception__()
+        except core.EnforceNotMet as ex:
+            self.assertIn("test exception", ex.message)
+
+        self.assertIsNotNone(ex)


 if __name__ == "__main__":