Add logging test

Author: rossant

phylib/__init__.py

@@ -39,13 +39,15 @@
 
 
 class _Formatter(logging.Formatter):
+    color_codes = {'L': '94', 'D': '90', 'I': '0', 'W': '33', 'E': '31'}
+
     def format(self, record):
         # Only keep the first character in the level name.
         record.levelname = record.levelname[0]
         filename = op.splitext(op.basename(record.pathname))[0]
         record.caller = '{:s}:{:d}'.format(filename, record.lineno).ljust(20)
         message = super(_Formatter, self).format(record)
-        color_code = {'D': '90', 'I': '0', 'W': '33', 'E': '31'}.get(record.levelname, '7')
+        color_code = self.color_codes.get(record.levelname, '90')
         message = '\33[%sm%s\33[0m' % (color_code, message)
         return message
 

phylib/utils/tests/test_testing.py

@@ -13,19 +13,30 @@
 
 from ..testing import captured_output, captured_logging, _assert_equal
 
+logger = logging.getLogger('phylib')
+
 
 #------------------------------------------------------------------------------
 # Tests
 #------------------------------------------------------------------------------
 
+def test_logging_1():
+    print()
+    logger.setLevel(5)
+    logger.log(5, "level 5")
+    logger.log(10, "debug")
+    logger.log(20, "info")
+    logger.log(30, "warning")
+    logger.log(40, "error")
+
+
 def test_captured_output():
     with captured_output() as (out, err):
         print('Hello world!')
     assert out.getvalue().strip() == 'Hello world!'
 
 
 def test_captured_logging():
-    logger = logging.getLogger('phylib')
     handlers = logger.handlers
     with captured_logging() as buf:
         logger.debug('Hello world!')