Add mock as a dependency under Python < 3.0, fixes #30

KushalP · Aug 17, 2016 · a3fb997 · a3fb997
1 parent c02a219
commit a3fb997
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -26,6 +26,14 @@ def run_tests(self):
 except:
     long_description = None
 
+test_requires = [
+    'pytest >= 2.5.2',
+    'pytest-cov >= 1.6',
+]
+
+if sys.version_info[:2] < (3, 0):
+    test_requires.append('mock')
+
 setup(
     name='serfclient',
     version='1.0.0',
@@ -40,7 +48,6 @@ def run_tests(self):
     license='MIT',
     packages=['serfclient'],
     install_requires=['msgpack-python >= 0.4.0'],
-    tests_require=['pytest >= 2.5.2',
-                   'pytest-cov >= 1.6'],
-    cmdclass={'test': PyTest}
+    tests_require=test_requires,
+    cmdclass={'test': PyTest},
 )