Python 2.6：正确使用 unittest.TestSuite

Question

使用Python 2.6，我在一个目录下的python文件中做了一个非常简单的测试：

#mytest.py
import unittest

class BasicTests(unittest.TestCase):
    def test_ok(self):
        self.assertTrue(True)

suite = unittest.TestLoader().loadTestsFromTestCase(BasicTests)

我切换到目录并运行python -m unittest mytest.suite，我收到以下错误：

Traceback (most recent call last):
  File "/usr/lib/python2.6/runpy.py", line 122, in _run_module_as_main
    "__main__", fname, loader, pkg_name)
  File "/usr/lib/python2.6/runpy.py", line 34, in _run_code
    exec code in run_globals
  File "/usr/lib/python2.6/unittest.py", line 875, in <module>
    main(module=None)
  File "/usr/lib/python2.6/unittest.py", line 816, in __init__
    self.parseArgs(argv)
  File "/usr/lib/python2.6/unittest.py", line 843, in parseArgs
    self.createTests()
  File "/usr/lib/python2.6/unittest.py", line 849, in createTests
    self.module)
  File "/usr/lib/python2.6/unittest.py", line 613, in loadTestsFromNames
    suites = [self.loadTestsFromName(name, module) for name in names]
  File "/usr/lib/python2.6/unittest.py", line 598, in loadTestsFromName
    test = obj()
  File "/usr/lib/python2.6/unittest.py", line 464, in __call__
    return self.run(*args, **kwds)
TypeError: run() takes exactly 2 arguments (1 given)

我尝试了几种变体（例如unittest.makeSuite 和unittest.LoadTestFromNames），但它们都给了我相同的基本错误。我错过了什么？我保留reading the documentation，我似乎遵循规范。

Answer 1

我通常不在命令行上使用unittest，但我有自己的测试运行脚本。

你需要在模块中添加一个函数suite

def suite():
    return unittest.TestLoader().loadTestsFromTestCase(BasicTests)

然后将其称为python -m unittest mytest.suite。但是后来我遇到了以下问题：

TypeError: calling <function suite at 0x00C1FB70> returned <unittest.TestSuite tests=[<mysite.BasicTests testMethod=test_ok>]>, not a test

这是因为unittest 使用了类似isinstance(mytest.suite(), TestSuite) 的东西，但是通过使用-m 执行，你会得到TestSuite 类的两个不同版本（一个是__main__.TestSuite，另一个是unittest.TestSuite），所以isinstance 返回错误。
对我来说，这看起来像一个错误。通过在loadTestsFromName 的开头插入from unittest import TestSuite, TestCase 来修补unittest.py 解决了isinstance 问题。抱歉，我无法为您提供“正确”解决方案（如果有的话）。

Answer 2

更新：看起来我上面的内容实际上是正确的。当我安装 Python 2.7 的副本时，一切正常。我翻遍了unittest.py 源代码，发现这行代码并没有按照预期的方式运行：

    elif isinstance(obj, TestSuite):
        return obj
    elif hasattr(obj, '__call__'):
        test = obj()

第一个elif 条件失败，因此它落入引发异常之后的条件。我仍然不确定那是如何远程发生的——也许是一个糟糕的编译——但我​​已经继续前进并与提供者联系filed a bug。

Answer 3

你可能想试试：

mytest.py：

import unittest

class BasicTests(unittest.TestCase):
    def test_ok(self):
        self.assertTrue(True)

if __name__ == '__main__':
    unittest.sys.argv.insert(1,'--verbose')
    unittest.main(argv = unittest.sys.argv)    

那么，

% python mytest.py

在TestCase 的所有子类中运行所有测试，并且

% python mytest.py BasicTests.test_ok

只运行test_ok。当您知道某个测试存在问题并且不想运行每个测试时，该命令的第二种形式很有用。