如何在python setup.py中使用unittest2測試

Question

如何強制python setup.py test使用unittest2包進行測試，而不是使用內置的unittest包？

Answer 1

讓我們假設你有一個名爲tests目錄中包含一個__init__.py文件，該文件定義了一個名爲suite函數返回一個測試套件。

我的解決方案與使用unittest2我自己test命令替換默認python setup.py test命令：

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命令。