Use unittest2 for Python 2.6

sebastinas · sebastinas · commit 031991b10b9f · 2015-01-08T23:54:39.000+01:00
Signed-off-by: Sebastian Ramacher &lt;sebastian+dev@ramacher.at&gt;
diff --git a/bpython/test/test_repl.py b/bpython/test/test_repl.py
@@ -1,12 +1,10 @@
 import os
-import unittest
 from itertools import islice
 from mock import Mock
 try:
-    from unittest import skip
+    import unittest2 as unittest
 except ImportError:
-    def skip(f):
-        return lambda self: None
+    import unittest
 
 from bpython import config, repl, cli, autocomplete
 
@@ -245,7 +243,7 @@ def test_current_string(self):
         self.assertEqual(self.repl.current_string(), '')
 
     # TODO: figure out how to capture whether foobar is in globals
-    @skip('not working yet')
+    @unittest.skip('not working yet')
     def test_push(self):
         self.repl = FakeRepl()
         self.repl.push("foobar = 2")
diff --git a/setup.py b/setup.py
@@ -175,6 +175,10 @@ def initialize_options(self):
     install_requires.append('ndg-httpsclient')
     install_requires.append('pyasn1')
 
+tests_require = ['mock']
+if sys.version_info[0] == 2 and sys.version_info[1] < 7:
+    tests_require.append('unittest2')
+
 # translations
 mo_files = list()
 for language in os.listdir(translations_dir):
@@ -194,7 +198,7 @@ def initialize_options(self):
     interpreter for Unix-like operating systems.""",
     install_requires = install_requires,
     extras_require = extras_require,
-    tests_require = ['mock'],
+    tests_require = tests_require,
     packages = packages,
     data_files = data_files,
     package_data = {
