Nose and Mock as tests_require deps, test command

HolgerPeters · HolgerPeters · commit 6ab2392ca359 · 2015-03-04T12:39:41.000+01:00
- require nose and mock for tests, not for normal installation
- implement a test command in setup.py for invoking tests
diff --git a/setup.py b/setup.py
@@ -121,6 +121,49 @@
     'Topic :: Scientific/Engineering :: Visualization',
     ]
 
+from setuptools.command.test import test as TestCommand
+class NoseTestCommand(TestCommand):
+    def finalize_options(self):
+        TestCommand.finalize_options(self)
+        self.test_args = []
+        self.test_suite = True
+
+    def run_tests(self):
+        try:
+            import matplotlib
+            matplotlib.use('agg')
+            import nose
+            from matplotlib.testing.noseclasses import KnownFailure
+            from matplotlib import default_test_modules
+            from matplotlib import font_manager
+            import time
+            # Make sure the font caches are created before starting any possibly
+            # parallel tests
+            if font_manager._fmcache is not None:
+                while not os.path.exists(font_manager._fmcache):
+                    time.sleep(0.5)
+            plugins = [KnownFailure]
+
+            # Nose doesn't automatically instantiate all of the plugins in the
+            # child processes, so we have to provide the multiprocess plugin
+            # with a list.
+            from nose.plugins import multiprocess
+            multiprocess._instantiate_plugins = plugins
+
+            if '--no-pep8' in sys.argv:
+                default_test_modules.remove('matplotlib.tests.test_coding_standards')
+                sys.argv.remove('--no-pep8')
+            elif '--pep8' in sys.argv:
+                default_test_modules = ['matplotlib.tests.test_coding_standards']
+                sys.argv.remove('--pep8')
+            nose.main(addplugins=[x() for x in plugins],
+                      defaultTest=default_test_modules,
+                      argv=['nosetests'],
+                      exit=False)
+        except ImportError:
+            sys.exit(-1)
+
+
 # One doesn't normally see `if __name__ == '__main__'` blocks in a setup.py,
 # however, this is needed on Windows to avoid creating infinite subprocesses
 # when using multiprocessing.
@@ -135,6 +178,7 @@
     package_dir = {'': 'lib'}
     install_requires = []
     setup_requires = []
+    tests_require = []
     default_backend = None
 
     # Go through all of the packages and figure out which ones we are
@@ -195,6 +239,7 @@
             package_data[key] = list(set(val + package_data[key]))
         install_requires.extend(package.get_install_requires())
         setup_requires.extend(package.get_setup_requires())
+        tests_require.extend(package.get_tests_require())
 
     # Write the default matplotlibrc file
     if default_backend is None:
@@ -254,11 +299,13 @@
         # List third-party Python packages that we require
         install_requires=install_requires,
         setup_requires=setup_requires,
+        tests_require=tests_require,
 
         # matplotlib has C/C++ extensions, so it's not zip safe.
         # Telling setuptools this prevents it from doing an automatic
         # check for zip safety.
         zip_safe=False,
+        cmdclass={'test': NoseTestCommand},
 
         **extra_args
     )
diff --git a/setupext.py b/setupext.py
@@ -415,6 +415,12 @@ def get_setup_requires(self):
         """
         return []
 
+    def get_tests_require(self):
+        """
+        Get a list of Python packages that we require for executing tests.
+        """
+        return []
+
     def _check_for_pkg_config(self, package, include_file, min_version=None,
                               version=None):
         """
@@ -692,7 +698,7 @@ def get_package_data(self):
                 'sphinxext/tests/tinypages/_static/*',
             ]}
 
-    def get_install_requires(self):
+    def get_tests_require(self):
         requires = ['nose>=%s' % self.nose_min_version]
         if not sys.version_info >= (3, 3):
             requires += ['mock']
diff --git a/tox.ini b/tox.ini
@@ -13,4 +13,5 @@ commands =
     {envpython} {toxinidir}/tests.py --processes=-1 --process-timeout=300
 deps =
     nose
+    mock
     numpy

-Original file line number
+Diff line change
     {envpython} {toxinidir}/tests.py --processes=-1 --process-timeout=300
 deps =
     nose
 +    mock
     numpy