Updated setup.py test to use PyTest

Author: mark-adams

.coveragerc

@@ -3,3 +3,4 @@ omit =
     tests/*
     .tox/*
     setup.py
+    *.egg/*

setup.py

@@ -6,6 +6,26 @@
 import jwt
 
 from setuptools import setup
+from setuptools.command.test import test
+
+
+class PyTest(test):
+    user_options = [('pytest-args=', 'a', "Arguments to pass to py.test")]
+
+    def initialize_options(self):
+        test.initialize_options(self)
+        self.pytest_args = []
+
+    def finalize_options(self):
+        test.finalize_options(self)
+        self.test_args = []
+        self.test_suite = True
+
+    def run_tests(self):
+        # import here, cause outside the eggs aren't loaded
+        import pytest
+        errno = pytest.main(['.'] + self.pytest_args)
+        sys.exit(errno)
 
 
 with open(os.path.join(os.path.dirname(__file__), 'README.md')) as readme:
@@ -57,5 +77,7 @@ def get_packages(package):
         'Programming Language :: Python :: 3.4',
         'Topic :: Utilities',
     ],
-    test_suite='tests'
+    test_suite='tests',
+    tests_require=['pytest', 'pytest-cov'],
+    cmdclass={'test': PyTest},
 )