Install test deps via setuptools/pip

3 files changed, 6 insertions, 2 deletions

diff --git a/.travis.yml b/.travis.yml
index 32806fe..d8f666c 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -19,7 +19,7 @@ addons:
       - jq
 
 before_install:
-  - pip install --quiet codecov
+  - pip install --quiet --upgrade pip codecov
 
 install:
   - make install
diff --git a/Makefile b/Makefile
index 08d0322..cc7ed16 100644
--- a/Makefile
+++ b/Makefile
@@ -1,5 +1,5 @@
 test_deps:
-	pip install coverage flake8 wheel
+	pip install .[test]
 
 lint: test_deps
 	./setup.py flake8
diff --git a/setup.py b/setup.py
index f819adc..c583e98 100755
--- a/setup.py
+++ b/setup.py
@@ -3,6 +3,8 @@
 import os, glob
 from setuptools import setup, find_packages
 
+tests_require = ["coverage", "flake8", "wheel"]
+
 setup(
     name="yq",
     version="2.0.1",
@@ -16,6 +18,8 @@ setup(
         "setuptools",
         "PyYAML >= 3.11"
     ],
+    tests_require=tests_require,
+    extras_require={"test": tests_require},
     packages=find_packages(exclude=["test"]),
     scripts=glob.glob("scripts/*"),
     include_package_data=True,