如何在 python 中使用 unittest2 setup.py test

Question

如何强制python setup.py test使用 unittest2 包而不是内置的 unittest 包进行测试？

Answer 1

假设

你有一个名为tests的目录，其中包含一个__init__.py文件，该文件定义了一个名为suite的函数，该函数返回一个测试套件。

我的解决方案是将默认的python setup.py test命令替换为我自己的test命令，该命令使用unittest2：

from setuptools import Command
from setuptools import setup
class run_tests(Command):
    """Runs the test suite using the ``unittest2`` package instead of the     
    built-in ``unittest`` package.                                            
    This is necessary to override the default behavior of ``python setup.py   
    test``.                                                                   
    """
    #: A brief description of the command.                                    
    description = "Run the test suite (using unittest2)."
    #: Options which can be provided by the user.                             
    user_options = []
    def initialize_options(self):
        """Intentionally unimplemented."""
        pass
    def finalize_options(self):
        """Intentionally unimplemented."""
        pass
    def run(self):
        """Runs :func:`unittest2.main`, which runs the full test suite using  
        ``unittest2`` instead of the built-in :mod:`unittest` module.         
        """
        from unittest2 import main
        # I don't know why this works. These arguments are undocumented.      
        return main(module='tests', defaultTest='suite',
                    argv=['tests.__init__'])
setup(
  name='myproject',
  ...,
  cmd_class={'test': run_tests}
)

现在运行python setup.py test运行我的自定义test命令。