Only import unittest2 if Python is < 2.7

See also: celery/case#1

Author: tony

tmuxp/testsuite/__init__.py

@@ -12,15 +12,15 @@
 import pkgutil
 import sys
 
-try:
-    import unittest2 as unittest
-except ImportError:  # Python 2.7
-    import unittest
-
 from tmuxp import log
 from tmuxp._compat import string_types, PY2, reraise
 from tmuxp.server import Server
 
+if sys.version_info < (2, 7):
+    import unittest2 as unittest
+else:
+    import unittest
+
 t = Server()
 t.socket_name = 'tmuxp_test'
 

tmuxp/testsuite/helpers.py

@@ -21,12 +21,11 @@
 from tmuxp import exc
 from tmuxp.testsuite import t
 
-try:
+if sys.version_info < (2, 7):
     import unittest2 as unittest
-except ImportError:  # Python 2.7
+else:
     import unittest
 
-
 logger = logging.getLogger(__name__)
 
 TEST_SESSION_PREFIX = 'test tmuxp_'