use pytest instead of nose

.travis.yml

@@ -11,6 +11,6 @@ install:
     - 'pip install -e ".[test]"'
     - 'pip install coveralls'
 script:
-    - "python ./setup.py nosetests"
+    - "python ./setup.py test"
 after_success:
     - coveralls

setup.py

@@ -6,19 +6,43 @@
 import sys
 
 from setuptools import setup, find_packages
+from setuptools.command.test import test as TestCommand
 
 here = os.path.abspath(os.path.dirname(__file__))
 README = open(os.path.join(here, 'README.md')).read()
 CHANGES = open(os.path.join(here, 'CHANGES.rst')).read()
 
 
+class PyTest(TestCommand):
+    user_options = [('pytest-args=', 'a', "Arguments to pass into py.test")]
+
+    def initialize_options(self):
+        TestCommand.initialize_options(self)
+        try:
+            from multiprocessing import cpu_count
+            self.pytest_args = ['-n', str(cpu_count()), '--boxed']
+            self.pytest_args = []
+        except (ImportError, NotImplementedError):
+             self.pytest_args = ['-n', '1', '--boxed']
+
+    def finalize_options(self):
+        TestCommand.finalize_options(self)
+        self.test_args = []
+        self.test_suite = True
+
+    def run_tests(self):
+        import pytest
+
+        errno = pytest.main(self.pytest_args)
+        sys.exit(errno)
+
 with open(os.path.join(here, 'gandi', 'cli', '__init__.py')) as v_file:
     version = re.compile(r".*__version__ = '(.*?)'",
                          re.S).match(v_file.read()).group(1)
 
 requires = ['setuptools', 'pyyaml', 'click>=3.1', 'requests', 'IPy']
 
-tests_require = ['nose', 'coverage', 'tox']
+tests_require = ['pytest', 'pytest-cov', 'tox']
 if sys.version_info < (2, 7):
     tests_require += ['unittest2', 'importlib']
 
@@ -51,11 +75,11 @@
     ],
     url='https://github.com/Gandi/gandi.cli',
     packages=find_packages(),
+    cmdclass={'test': PyTest},
     include_package_data=True,
     zip_safe=False,
     install_requires=requires,
     tests_require=tests_require,
-    test_suite='nose.collector',
     extras_require=extras_require,
     entry_points={
         'console_scripts': [