前言
笔者平常运行pytest用例时，通常使用命令行方式，像这样
pytest -v pxl/test_dir/test_demo.py::TestDemo::test_my_var，执行某一条case，但每次命令行敲也挺麻烦的。那如何在python代码中调用pytest呢？带着疑问一起看。
解答疑问
在main()函数中调用pytest.main()就可以啦。
我们深入探讨一下 pytest.main 的用法和一些常见的应用场景。
什么是pytest.main？
pytest.main 是 Pytest 框架中一个非常实用的函数，它用于从命令行运行测试集或者以编程方式运行测试。通过调用 pytest.main() 函数，我们可以在代码中启动测试过程，并获取测试结果或者进行进一步的操作。
如何使用 pytest.main 运行测试用例？
我们看一个简单示例：
test_demo.py
import pytest
def test_add():
assert 1 + 1 == 2
if __name__ == "__main__":
pytest.main([__file__])
在这个示例中，我们定义了一个简单的测试用例 test_add，用于检查加法运算的正确性。然后，我们通过 pytest.main([__file__]) 来执行当前文件中的测试用例。
要运行这个测试用例，只需在命令行中执行以下命令：
python test_demo.py
Pytest 将会自动发现并执行 test_add 测试用例，并生成测试结果报告。
源码定义
def main(
args: Optional[Union[List[str], "os.PathLike[str]"]] = None,
plugins: Optional[Sequence[Union[str, _PluggyPlugin]]] = None,
) -> Union[int, ExitCode]:
"""Perform an in-process test run.
:param args: List of command line arguments.
:param plugins: List of plugin objects to be auto-registered during initialization.
:returns: An exit code.
"""
可以看到有两个参数，args 传一个list对象，list 里面是多个命令行的参数；plugins 传一个list对象，list 里面是初始化的时候需注册的插件
常用参数
-m或--mark=：只运行被标记为的测试用例。-k或--keyword=：只运行名称中包含的测试用例。-s或--capture=no：禁止捕获标准输出和标准错误，将其直接打印到控制台。-v或--verbose：以详细模式运行测试，显示更多的信息，如每个测试用例的名称、结果和错误信息。--collect-only：只收集测试用例但不执行它们，可以用于查看测试集中的所有可用测试。- --ignore：忽略某个测试模块
- -q: 简单输出模式, 不输出环境信息
- -x: 出现一条测试用例失败就退出测试。在调试阶段非常有用，当测试用例失败时，应该先调试通过，而不是继续执行测试用例。
这些参数可以单独使用，也可以组合使用。例如，可以同时使用 -m 和 -k 参数来指定运行特定标记和关键字的测试用例。像这样
pytest.main(["-s", "-v", "-m", "p0"])
举例说明
-m
import pytest
@pytest.mark.p0
def test_addition():
assert 2 + 2 == 4
@pytest.mark.p1
def test_subtraction():
assert 5 - 3 == 2
if __name__ == "__main__":
pytest.main(["-m", "p0"])
在这个示例中，我们使用 @pytest.mark 装饰器为两个测试用例标记了不同的标记。然后，我们在 pytest.main 函数调用时传递了 -m p0 参数，表示只运行被标记为 p0 的测试用例。这样就可以只运行 test_addition 测试用例。
-k
import pytest
def test_addition():
assert 2 + 2 == 4
def test_subtraction():
assert 5 - 3 == 2
def test_multiplication():
assert 2 * 3 == 6
if __name__ == "__main__":
pytest.main(["-k", "addition"])
在这个示例中，我们定义了三个测试用例，分别测试加法、减法和乘法运算的正确性。然后，我们在 pytest.main 函数调用时传递了 -k addition 参数，表示只运行名称中包含 "addition" 关键字的测试用例。这样就可以只运行 test_addition 测试用例。
--ignore
test_demo.py
import pytest
def test_addition():
assert 2 + 2 == 4
def test_subtraction():
assert 5 - 3 == 2
if __name__ == "__main__":
pytest.main(["--ignore=test_contract_controller.py"])
在这个示例中，我们定义了两个测试用例 test_addition 和 test_subtraction。然后，我们在 pytest.main 函数调用时传递了 --ignore=test_contract_controller.py 参数，表示忽略名为 test_contract_controller.py 的测试模块，不运行其中的测试用例。
--durations
import pytest
import time
def test_slow_function():
time.sleep(2)
assert True
def test_fast_function():
assert True
if __name__ == "__main__":
pytest.main(["--durations=5"])
在这个示例中，我们定义了两个测试用例，test_slow_function 和 test_fast_function，分别模拟了一个耗时较长和一个耗时较短的测试。然后，我们在 pytest.main 函数调用时传递了 --durations=5 参数，表示输出最慢的 5 个测试用例的执行时间。这样可以帮助我们识别出执行时间较长的测试用例，进行性能优化或者重点关注。
--collect-only
import pytest
def test_addition():
assert 2 + 2 == 4
def test_subtraction():
assert 5 - 3 == 2
if __name__ == "__main__":
pytest.main(["--collect-only"])
我们在 pytest.main 函数调用时传递了 --collect-only 参数，这样只会收集用例，而不会执行。
运行指定用例
运行指定目录下的case
pytest.main(["--collect-only", "/panda-test/peilian/test_dir/"])
运行指定模块下的case
pytest.main(["--collect-only", "/panda-test/peilian/test_dir/test_demo.py"])
运行指定的case
pytest.main(["--collect-only", "/panda-test/peilian/test_dir/test_demo.py::test_01"])
plugins参数
我们自定义的插件放到 conftest.py 会被pytest查找到，如果不是写到conftest.py 的插件内容，可以通过 plugins 参数指定加载。我们看个案例：
import pytest
class MyPlugin(object):
def pytest_sessionstart(self):
print("*** test run start ***")
def test_division():
assert 10 / 2 == 5
if __name__ == "__main__":
pytest.main(["/panda-test/peilian/test_dir/test_demo.py::test_division"], plugins=[MyPlugin()])
这里又涉及到一个新的钩子函数pytest_sessionstart，先不用管，之后写一篇文章来介绍。我们运行case，会发现在执行最开始会输出*** test run start ***。说明已经加载插件成功了。
plugins参数的作用就是指定需加载的插件，也可以指定多个。
最后
pytest.main 是一个非常有用的函数，它允许我们以编程的方式执行测试用例。通过灵活运用 pytest.main，我们可以更好地集成测试流程和自动化工具，提高测试效率和质量。
