Source

whoosh / setup.py

Diff from to
 
 import os.path, sys
 from setuptools import setup, find_packages
+from setuptools.command.test import test as TestCommand
+
+try:
+    import pytest
+except ImportError:
+    pytest = None
 
 sys.path.insert(0, os.path.abspath("src"))
 from whoosh import __version__, versionstring
 
+
+class PyTest(TestCommand):
+    def finalize_options(self):
+        TestCommand.finalize_options(self)
+        self.test_args = []
+        self.test_suite = True
+
+    def run_tests(self):
+        #import here, cause outside the eggs aren't loaded
+        import pytest
+        pytest.main(self.test_args)
+
+
 if __name__ == "__main__":
     setup(
         name="Whoosh",
         url="http://bitbucket.org/mchaput/whoosh",
 
         zip_safe=True,
-        test_suite="nose.collector",
+        tests_require=['pytest'],
+        cmdclass = {'test': pytest},
 
         classifiers=[
         "Development Status :: 5 - Production/Stable",