- fixed setup.py test so that it only finds and executes netaddr tests

1 files changed, 8 insertions, 1 deletions

diff --git a/setup.py b/setup.py
index 9b6f3d9..4bd1f41 100644
--- a/setup.py
+++ b/setup.py
@@ -27,9 +27,16 @@ class PyTest(Command):
     def run(self):
         import subprocess
         import sys
-        errno = subprocess.call([sys.executable, 'runtests.py'])
+        import os
+        saved_cwd = os.getcwd()
+        try:
+            os.chdir(os.path.join(os.path.dirname(__file__), 'test'))
+            errno = subprocess.call([sys.executable, '../runtests.py'])
+        finally:
+            os.chdir(saved_cwd)
         raise SystemExit(errno)
 
+
 def main():
     if sys.version_info[:2] < (2, 4):
         sys.stderr.write("netaddr requires Python version 2.4 or higher.\n")