setup.py test

1 files changed, 12 insertions, 1 deletions

diff --git a/setup.py b/setup.py
index 34ff57a..bbf2724 100644..100755
--- a/setup.py
+++ b/setup.py
@@ -1,11 +1,19 @@
+# -*- coding: utf-8 -*-
+
+import os
+
 from os.path import dirname, join
-from setuptools import setup, find_packages
+from setuptools import setup, find_packages, Command
+# Hack because logging in setuptools sucks.
+import multiprocessing
 
 
 def fread(fn):
     with open(join(dirname(__file__), fn), 'r') as f:
         return f.read()
 
+tests_require = ['nose', 'unittest2', 'pycrypto']
+
 setup(
     name = 'oauthlib',
     version = '0.0.1',
@@ -16,5 +24,8 @@ setup(
     url = 'https://github.com/idangazit/oauthlib',
     license = fread('LICENSE'),
     packages = find_packages(exclude=('tests', 'docs')),
+    test_suite = 'nose.collector',
+    tests_require=tests_require,
+    extras_require={'test': tests_require},
 )