unittest2 if python 2.6

Author: tony

tmuxp/testsuite/__init__.py

@@ -30,6 +30,9 @@
 
 
 def suite():
-    import unittest
+    try:
+        import unittest2
+    except ImportError: # Python 2.7
+        import unittest
 
     return unittest.TestLoader().discover('.', pattern="test_*.py")

tmuxp/testsuite/helpers.py

@@ -1,6 +1,9 @@
 # -*- coding: utf-8 -*-
 from __future__ import absolute_import, division, print_function, with_statement
-import unittest
+try:
+    import unittest2
+except ImportError: # Python 2.7
+    import unittest
 import time
 from . import t
 

tmuxp/testsuite/test_cli.py

@@ -3,13 +3,15 @@
 
 import os
 import shutil
-import unittest
+try:
+    import unittest2
+except ImportError:  # Python 2.7
+    import unittest
 import kaptan
 import tempfile
 from .. import config, cli
 from ..util import tmux
 
-from .. import log
 import logging
 
 logger = logging.getLogger(__name__)

tmuxp/testsuite/test_config.py

@@ -3,12 +3,14 @@
 
 import os
 import shutil
-import unittest
+try:
+    import unittest2
+except ImportError: # Python 2.7
+    import unittest
 import kaptan
 from .. import config, exc
 from ..util import tmux
 
-from .. import log
 import logging
 
 logger = logging.getLogger(__name__)

tmuxp/testsuite/test_config_teamocil.py

@@ -2,13 +2,14 @@
 from __future__ import absolute_import, division, print_function, with_statement
 
 import os
-import shutil
-import unittest
+try:
+    import unittest2
+except ImportError:  # Python 2.7
+    import unittest
 import kaptan
 from .. import config, exc
 from ..util import tmux
 
-from .. import log
 import logging
 
 logger = logging.getLogger(__name__)

tmuxp/testsuite/test_config_tmuxinator.py

@@ -2,13 +2,14 @@
 from __future__ import absolute_import, division, print_function, with_statement
 
 import os
-import shutil
-import unittest
+try:
+    import unittest2
+except ImportError:  # Python 2.7
+    import unittest
 import kaptan
 from .. import config, exc
 from ..util import tmux, basestring
 
-from .. import log
 import logging
 
 logger = logging.getLogger(__name__)

tmuxp/testsuite/test_pane.py

@@ -1,11 +1,14 @@
 # -*- coding: utf-8 -*-
 from __future__ import absolute_import, division, print_function, with_statement
 
-import unittest
+try:
+    import unittest2
+except ImportError:  # Python 2.7
+    import unittest
+
 from .helpers import TmuxTestCase
 from . import t
 
-from .. import log
 import logging
 
 logger = logging.getLogger(__name__)

tmuxp/testsuite/test_server.py

@@ -1,14 +1,15 @@
 # -*- coding: utf-8 -*-
 from __future__ import absolute_import, division, print_function, with_statement
 
-import unittest
+try:
+    import unittest2
+except ImportError: # Python 2.7
+    import unittest
 from random import randint
 from .. import Server
-from ..util import tmux
-from .helpers import TmuxTestCase, TEST_SESSION_PREFIX
+from .helpers import TmuxTestCase
 from . import t
 
-from .. import log
 import logging
 
 logger = logging.getLogger(__name__)

tmuxp/testsuite/test_session.py

@@ -1,15 +1,17 @@
 # -*- coding: utf-8 -*-
 from __future__ import absolute_import, division, print_function, with_statement
 
-import unittest
+try:
+    import unittest2
+except ImportError:  # Python 2.7
+    import unittest
 from random import randint
 from time import sleep
 from .. import Session, Window, Pane
 from ..util import tmux
 from .helpers import TmuxTestCase, TEST_SESSION_PREFIX
 from . import t
 
-from .. import log
 import logging
 
 logger = logging.getLogger(__name__)

tmuxp/testsuite/test_tmuxobject.py

@@ -1,13 +1,16 @@
 # -*- coding: utf-8 -*-
 from __future__ import absolute_import, division, print_function, with_statement
 
-import unittest
 import random
+try:
+    import unittest2
+except ImportError:  # Python 2.7
+    import unittest
+
 from .. import Pane, Window, Session
 from .helpers import TmuxTestCase
 from . import t
 
-from .. import log
 import logging
 
 logger = logging.getLogger(__name__)