[WIP] Use `unittest` assert methodsadd-unittest-assertion

Changed assertion in unit tests from Python built-in
to `unittest` provided. The motivation is to have better
reporting on test failures.

1 files changed, 4 insertions, 12 deletions

diff --git a/tests/test_debug.py b/tests/test_debug.py
index 1cd4c05..1225aae 100644
--- a/tests/test_debug.py
+++ b/tests/test_debug.py
@@ -7,10 +7,10 @@
 import sys
 import unittest
 
-from tests.base import BaseTestCase
-
 from pyasn1 import debug
 from pyasn1 import error
+from tests.base import BaseTestCase
+
 
 class DebugCaseBase(BaseTestCase):
     def testKnownFlags(self):
@@ -19,16 +19,8 @@ class DebugCaseBase(BaseTestCase):
         debug.setLogger(0)
 
     def testUnknownFlags(self):
-        try:
-            debug.setLogger(debug.Debug('all', 'unknown', loggerName='xxx'))
-
-        except error.PyAsn1Error:
-            debug.setLogger(0)
-            return
-
-        else:
-            debug.setLogger(0)
-            assert 0, 'unknown debug flag tolerated'
+        self.assertRaises(
+            error.PyAsn1Error, debug.Debug, 'all', 'unknown', loggerName='xxx')
 
 
 suite = unittest.TestLoader().loadTestsFromModule(sys.modules[__name__])