X-Git-Url: https://git.ayoreis.com/Adafruit_Blinka-hackapet.git/blobdiff_plain/e307ee8331fabcb9393a4110a595bae58dad3bf3..3f379525a51a3b34e001ca20dacfb27d64ba4c70:/python/unittest.py

diff --git a/python/unittest.py b/python/unittest.py
index d986705..aab8192 100644
--- a/python/unittest.py
+++ b/python/unittest.py
@@ -1,7 +1,10 @@
-"""Copied from https://raw.githubusercontent.com/micropython/micropython-lib/cfa1b9cce0c93a3115bbff3886c9bbcddd9e8047/unittest/unittest.py """
+"""Based on https://raw.githubusercontent.com/micropython/micropython-lib/cfa1b9cce0c93a3115bbff3886c9bbcddd9e8047/unittest/unittest.py """
+import sys
 class SkipTest(Exception):
     pass
 
+raiseException = False
+raiseBaseException = True
 
 class AssertRaisesContext:
 
@@ -195,12 +198,22 @@ def run_class(c, test_result):
             except SkipTest as e:
                 print(" skipped:", e.args[0])
                 test_result.skippedNum += 1
-            except:
+            except Exception as e: # user exception
                 print(" FAIL")
-                test_result.failuresNum += 1
-                # Uncomment to investigate failure in detail
-                #raise
-                continue
+                if raiseException:
+                    raise
+                else:
+                    print(e)
+                    test_result.failuresNum += 1
+                    continue
+            except BaseException as e: # system exception
+                print(" FAIL")
+                if raiseBaseException:
+                    raise
+                else:
+                    print(e)
+                    test_result.failuresNum += 1
+                    continue
             finally:
                 tear_down()
 
@@ -212,7 +225,7 @@ def main(module="__main__"):
             if isinstance(c, object) and isinstance(c, type) and issubclass(c, TestCase):
                 yield c
 
-    m = __import__(module, fromlist=['']) # changed to permit non-top-level testing modules
+    m = __import__(module) # changed to permit non-top-level testing modules
     suite = TestSuite()
     for c in test_cases(m):
         suite.addTest(c)