pytest多进程或多线程执行测试实例

下面是关于pytest多进程或多线程执行测试实例的完整攻略。

什么是pytest？

pytest是Python的一个单元测试框架，是Python标准库中unittest的一个替代方案。

pytest多进程或多线程执行测试实例有什么优劣？

pytest支持多进程或多线程执行测试实例，这样可以有效提高测试效率，提升测试覆盖率，但也有一定的缺点，例如可能会带来一些测试结果不稳定的问题。需要在实际的使用中根据具体的情况进行选择。

怎么实现pytest多进程或多线程执行测试实例？

多线程执行测试实例

使用pytest-xdist插件可以很方便地实现pytest多线程执行测试实例的功能。

1. 安装pytest-xdist插件

在命令行界面中执行以下命令：

pip install pytest-xdist

1. 使用pytest-xdist插件进行测试

在命令行界面中执行以下命令：

pytest -n 4

上述命令中的“-n 4”表示使用4个进程执行测试。

多进程执行测试实例

使用pytest-forked插件可以很方便地实现pytest多进程执行测试实例的功能。

1. 安装pytest-forked插件

在命令行界面中执行以下命令：

pip install pytest-forked

1. 使用pytest-forked插件进行测试

在命令行界面中执行以下命令：

pytest --forked -n 4

上述命令中的“--forked -n 4”表示使用4个进程执行测试。

示例说明

示例1

下方是(test_foo.py)，一个简单的测试样例：

def test_sum():
    assert sum([1, 2, 3]) == 6
def test_len():
    assert len([4, 5]) == 2

在命令行执行以下命令：

pytest -n 2 test_foo.py

表示使用2个进程运行测试，会得到以下输出：

=============== test session starts ===============
platform darwin -- Python 3.x.y, pytest-3.x.y, py-1.x.y, pluggy-0.x.y -- /path/to/bin/python3
cachedir: .pytest_cache
[pytestfork] pytest-forked: process <ParentProcess 97725>: forked N workers
[pytestfork] live output for process <ParentProcess 97725>:
[pytestfork]  plugin.py:144 [INFO] running first test suite in main process Test directory is /mytests/tests.
[pytestfork]  plugin.py:225 [INFO] running next available worker pid 97726.
[pytestfork]  runner.py:126 [INFO] worker 97726 started.
[pytestfork]  runner.py:126 [INFO] worker 97727 started.
[pytestfork]  runner.py:126 [INFO] worker 97728 started.
[pytestfork]  runner.py:126 [INFO] worker 97729 started.
test_foo.py .. [100%]

============== 2 passed in x.x seconds =============

可以看到，测试用例已经非常快速地执行完毕。

示例2

下方是两个(test_foo.py)分别进行耗时操作的测试样例：

import time
def test_one():
    time.sleep(5)
    assert 1
def test_two():
    time.sleep(5)
    assert 1

在命令行中执行以下命令：

pytest -n 2 test_foo.py

这样的话，两个测试用例就会被安排在两个进程中分别运行。
因为这里运行时间为5s，因此两个用例都应该被执行完成但需要10s。

运行的输出结果如下所示：

=============== test session starts ===============
platform darwin -- Python 3.8.10, pytest-6.1.2, py-1.9.0, pluggy-0.13.1
cachedir: .pytest_cache
[pytestfork] pytest-forked: running 2 parallel tests using 2 free CPUs

[pytestfork] RUNNING tests/test_foo.py::test_one (1/2)
[pytestfork] RUNNING tests/test_foo.py::test_two (2/2)
[pytestfork] worker finished: exitstatus=0

[pytestfork] worker finished: exitstatus=0

================ 2 passed in 5.05s ===================

从上述输出结果可以看到，两个用例都被成功地执行了。
第一个用例被分配到了一个进程中，而第二个用例被分配到了另一个进程中。 这样可以大大缩短测试用例的执行时间。

这就是使用pytest实现多进程或多线程执行测试实例的方法。