adding functionality for user to use `python setup.py test`

diff --git a/setup.py b/setup.py
index 3dae71fd173a46115f6b1cc13a3ba1eeb8862a17..7a7eb8605e949c7ce651d8c0eb2b9948d4da202b 100644 (file)
--- a/setup.py
+++ b/setup.py
@@ -1,4 +1,5 @@
 from setuptools import setup
+import sys
 
 setup(
     name='xkcdpass',
@@ -17,6 +18,8 @@ setup(
             'xkcdpass = xkcdpass.xkcd_password:main',
         ],
     },
+    tests_require=['mock'] if sys.version_info[0] == 2 else None,
+    test_suite = 'tests',
     classifiers=[
         'Development Status :: 5 - Production/Stable',
         'Programming Language :: Python :: 2',