handle and display AssertionErrors

Author: ggorlen

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()))