diff --git a/setup.py b/setup.py index dee5365..d3ceddd 100755 --- a/setup.py +++ b/setup.py @@ -2,11 +2,16 @@ import sys ispy3 = sys.version_info[0] == 3 +kwargs = {} if ispy3: from distutils.core import setup # NOQA else: try: from setuptools import setup # NOQA + kwargs = { + 'tests_require': ['nose'], + 'test_suite': 'nose.collector', + } except ImportError: from distutils.core import setup # NOQA @@ -32,4 +37,5 @@ setup( "Programming Language :: Python", "Topic :: Software Development", "Topic :: Utilities", - ]) + ], + **kwargs)