Integrate pytest and flake8 into setuptools

1 files changed, 10 insertions, 1 deletions

diff --git a/setup.py b/setup.py
index 466b10e..8f866b7 100644
--- a/setup.py
+++ b/setup.py
@@ -12,11 +12,20 @@ setup(
                       open('LICENSE.rst').read() + '\n\n' +
                       open('HISTORY.rst').read()),
     license='ISC',
-    py_modules=['dotfiles'],
+    py_modules=[
+        'dotfiles'
+    ],
+    setup_requires=[
+        'pytest-runner',
+        'flake8',
+    ],
     install_requires=[
         'click',
         'py',
     ],
+    tests_require=[
+        'pytest'
+    ],
     entry_points={
         'console_scripts': [
             'dotfiles=dotfiles:cli',