- added support for running the test suite to setup.py

1 files changed, 17 insertions, 3 deletions

diff --git a/setup.py b/setup.py
index 5e4d17c..9b6f3d9 100644
--- a/setup.py
+++ b/setup.py
@@ -8,13 +8,28 @@ A distutils Python setup file. For setuptools support see setup_egg.py.
 import os
 import sys
 
-from distutils.core import setup
+from distutils.core import setup, Command
 
 if os.path.exists('MANIFEST'):
     os.remove('MANIFEST')
 
 import release
 
+
+class PyTest(Command):
+    user_options = []
+    def initialize_options(self):
+        pass
+
+    def finalize_options(self):
+        pass
+
+    def run(self):
+        import subprocess
+        import sys
+        errno = subprocess.call([sys.executable, 'runtests.py'])
+        raise SystemExit(errno)
+
 def main():
     if sys.version_info[:2] < (2, 4):
         sys.stderr.write("netaddr requires Python version 2.4 or higher.\n")
@@ -27,6 +42,7 @@ def main():
         author           = release.author,
         author_email     = release.author_email,
         classifiers      = release.classifiers,
+        cmdclass         = {'test': PyTest},
         description      = release.description,
         download_url     = release.download_url,
         keywords         = release.keywords,
@@ -39,8 +55,6 @@ def main():
         scripts          = release.scripts,
         url              = release.url,
         version          = release.version,
-        install_requires = release.install_requires,
-        setup_requires   = release.setup_requires,
         options = {
             'build_scripts': {
                 'executable': '/usr/bin/env python',