Question

我在pwd中保存了一组单元测试脚本。我希望能够计算将要执行的单元测试（nosetests）的数量（不实际执行它们）并将该数字返回到python变量中，如下所示：

>>> number_of_unit_tests = count_unit_tests('.')
>>> number_of_unit_tests

400

我知道我可以从命令行收集这样的内容：

nosetests --collect-only

但是可以在脚本中执行此操作吗？

Answer 1

你可以运行任何鼻子命令形式python脚本，如basic nose usage中所述，唯一的技巧是提取测试数量。我看了functional tests in nose并认为这样的事情应该有效，但你可以进一步削减它：

import sys
import unittest
from cStringIO import StringIO

import nose
from nose.result import _TextTestResult


class TestRunner(unittest.TextTestRunner):
    def _makeResult(self):
        self.result = _TextTestResult(
            self.stream, self.descriptions, self.verbosity)
        return self.result


def count_unit_tests(module_name):
    stream = StringIO()
    runner = TestRunner(stream=stream)

    result = nose.run(
        testRunner=runner,
        argv=[sys.argv[0],
              module_name,
              '-s',
              '-v',
              '--collect-only'
        ]
    )

    return runner.result.testsRun

if __name__ == '__main__':
    print count_unit_tests('.')

计算并返回测试次数

1 个答案: