如何使用覆盖率 API 运行单元测试

Question

我正在尝试使用 coverage-API (https://coverage.readthedocs.io/en/6.3.2/api.html#api）。

链接页面上描述的简单用例告诉我将执行的代码包装在他们提供的代码片段中。我正在使用 unittest.main() 来执行测试。下面的代码运行时没有错误，但既没有创建任何报告信息，也没有执行 print("Done.")。

我猜 unittest.main() 在某个地方调用了 sys.exit() ？如何使用 API 来执行所有单元测试？

例子

import coverage
import unittest


def func(input):
    return input


class testInput(unittest.TestCase):
    def test_func(self):
        self.assertEqual(func(1), 1)

    
if __name__ == '__main__':
    cov = coverage.Coverage()
    cov.start()

    unittest.main()

    cov.stop()
    cov.save()

    cov.html_report()
    print("Done.")

Answer 1

是的，看起来unittest.main()正在调用sys.exit()。您确定需要使用覆盖率 API 吗？您可以跳过覆盖范围和大部分单元测试：

# foo.py
import unittest


def func(input):
    return input


class testInput(unittest.TestCase):
    def test_func(self):
        self.assertEqual(func(1), 1)

然后：

$ python -m coverage run -m unittest foo.py
.
----------------------------------------------------------------------
Ran 1 test in 0.000s

OK

$ python -m coverage report -m
Name     Stmts   Miss  Cover   Missing
--------------------------------------
foo.py       6      0   100%
--------------------------------------
TOTAL        6      0   100%

Answer 2

要使示例正常工作，您可以简单地将对 unittest.main() 的调用放在 try/ except 语句中，这样一旦单元测试已完成。

# testInput.py
import coverage
import unittest

def func(input):
    return input

class testInput(unittest.TestCase):
    def test_func(self):
        self.assertEqual(func(1), 1)

if __name__ == '__main__':
    cov = coverage.Coverage()
    cov.start()

    try:
        unittest.main()
    except:  # catch-all except clause
        pass

    cov.stop()
    cov.save()

    cov.html_report()
    print("Done.")

这将根据需要运行，包括在最后打印 Done.：

python3 testInput.py 
.
----------------------------------------------------------------------
Ran 1 test in 0.000s

OK
Done.

HTML 覆盖率报告将在当前工作目录的 htmlcov 目录中创建。