Add unittest2 for py2

Author: gmuraru

friendlylog/simple_logger.py

@@ -57,7 +57,6 @@ def set_fmt(fmt_string):
 _logger.setLevel(logging.DEBUG)
 
 
-
 # Export functions and objects.
 inner_logger = _logger  # Don't use this except if you know what you are doing.
 inner_stream_handler = _stream_handler  # Same thing for this object.

tests/test_colored_logger.py

@@ -8,6 +8,12 @@
 from friendlylog import colored_logger as logger
 from threading import Thread
 
+if six.PY2:
+    import unittest2 as unittest
+else:
+    import unittest
+
+
 
 # Tests cannot be executed in parallel due to the hack in the setUp method.
 class TestColoredLogger(unittest.TestCase):

tests/test_simple_logger.py

@@ -3,12 +3,15 @@
 import random
 import six
 import string
-import unittest
 
 from friendlylog import simple_logger as logger
-
 from threading import Thread
 
+if six.PY2:
+    import unittest2 as unittest
+else:
+    import unittest
+
 
 # Tests cannot be executed in parallel due to the hack in the setUp method.
 class TestSimpleLogger(unittest.TestCase):