Enable run tests with full dependency including extras_require

dvecerdea · Dec 21, 2018 · 25cf54c · 25cf54c
1 parent 7d48556
commit 25cf54c
Showing 1 changed file with 16 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -24,6 +24,18 @@
         f.read().decode('utf-8')).group(1)))
 
 
+kerberos_require = ['requests_kerberos']
+
+all_require = [kerberos_require]
+
+tests_require = all_require + [
+    'httpretty',
+    'pytest',
+    'pytest-runner'
+]
+
+py27_require = ['ipaddress', 'typing']
+
 setup(
     name='presto-python-client',
     author='Presto Team',
@@ -61,8 +73,9 @@
         'six',
     ],
     extras_require={
-        'kerberos': ['requests_kerberos'],
-        'tests': ['httpretty', 'pytest', 'pytest-runner'],
-        ':python_version=="2.7"': ['ipaddress', 'typing'],
+        'all': all_require,
+        'kerberos': kerberos_require,
+        'tests': tests_require,
+        ':python_version=="2.7"': py27_require,
     },
 )