Merge pull request #5 from ggorlen/add-assertion-error

handle and display AssertionErrors
codewars · Jun 12, 2020 · e5a0195 · e5a0195
2 parents b3821b0 + b4bebd6
commit e5a0195
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/codewars_test/test_framework.py b/codewars_test/test_framework.py
@@ -109,6 +109,8 @@ def wrapper(func):
             time = timer()
             try:
                 func()
+            except AssertionError as e:
+                display('FAILED', str(e))
             except Exception:
                 fail('Unexpected exception raised')
                 tb_str = ''.join(format_exception(*exc_info()))