Restore setup.py test command

Author: nicoddemus

setup.py

@@ -1,14 +1,34 @@
+import sys
 import re
 
 from setuptools import setup
+from setuptools.command.test import test as TestCommand
+
+
+class PyTest(TestCommand):
+    """
+    Overrides setup "test" command, taken from here:
+    http://pytest.org/latest/goodpractises.html
+    """
+
+    def finalize_options(self):
+        TestCommand.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([])
+        sys.exit(errno)
 
 
 with open('pytestqt/__init__.py') as f:
     m = re.search("version = '(.*)'", f.read())
     assert m is not None
     version = m.group(1)
 
-
 setup(
     name="pytest-qt",
     version=version,
@@ -40,4 +60,5 @@
         'Topic :: Software Development :: User Interfaces',
     ],
     tests_requires=['pytest'],
+    cmdclass={'test': PyTest},
 )