Set up pytest

2 files changed, 5 insertions, 0 deletions

diff --git a/setup.cfg b/setup.cfg
index 14c6824..459ffba 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -5,3 +5,6 @@ universal=1
 source-dir = docs
 build-dir  = docs/build
 all_files  = 1
+
+[aliases]
+test=pytest
diff --git a/setup.py b/setup.py
index a9e2bfc..59d684b 100644
--- a/setup.py
+++ b/setup.py
@@ -4,6 +4,8 @@ setup(name='Mastodon.py',
       version='1.0.6',
       description='Python wrapper for the Mastodon API',
       packages=['mastodon'],
+      setup_requires=['pytest-runner'],
+      tests_require=['pytest'],
       install_requires=['requests', 'dateutils'],
       url='https://github.com/halcy/Mastodon.py',
       author='Lorenz Diener',