Merge with a0fec27963ebb52d39cdad42d5a7fd428f989784

1 files changed, 8 insertions, 2 deletions

diff --git a/setup.py b/setup.py
index 0015f0d9..4c5ff054 100644
--- a/setup.py
+++ b/setup.py
@@ -1,7 +1,7 @@
 #!/usr/bin/env/python
 # -*- coding: utf-8 -*-
 
-from distutils.core import setup
+from setuptools import setup
 
 setup(
     name = "djangorestframework",
@@ -14,9 +14,11 @@ setup(
     author_email = 'tom@tomchristie.com',
     packages = ['djangorestframework',
                 'djangorestframework.templatetags',
-                'djangorestframework.tests'],
+                'djangorestframework.tests',
+                'djangorestframework.runtests'],
     package_dir={'djangorestframework': 'djangorestframework'},
     package_data = {'djangorestframework': ['templates/*', 'static/*']},
+    test_suite = 'djangorestframework.runtests.runtests.main',
     classifiers = [
         'Development Status :: 4 - Beta',
         'Environment :: Web Environment',
@@ -29,3 +31,7 @@ setup(
     ]
 )
 
+import os, shutil
+shutil.rmtree(os.path.join(os.path.dirname(__file__), 'djangorestframework.egg-info'), True)
+
+