Only depend on unittest2 for Python 2.X with X < 7

sebastinas · sebastinas · commit d6de41b9506f · 2015-01-06T20:58:43.000+01:00
Signed-off-by: Sebastian Ramacher &lt;sebastian+dev@ramacher.at&gt;
diff --git a/setup.py b/setup.py
@@ -174,6 +174,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):
@@ -193,7 +197,7 @@ def initialize_options(self):
     interpreter for Unix-like operating systems.""",
     install_requires = install_requires,
     extras_require = extras_require,
-    tests_require = ['mock', 'unittest2'],
+    tests_require = tests_require,
     packages = packages,
     data_files = data_files,
     package_data = {
