pytest 在 main 函数中执行测试用例的 3 种常用方法

在 Python 脚本的if __name__ == '__main__':主函数中调用pytest，可以直接运行测试用例，无需手动敲命令行，非常适合 IDE 直接运行、自动化脚本集成。

下面给你最实用、可直接复制的写法：

方法 1：最简写法（执行当前文件所有用例）

python

运行

import pytest # 测试用例示例 def test_case1(): assert 1 + 1 == 2 def test_case2(): assert "pytest" in "hello pytest" if __name__ == '__main__': # 执行当前文件的所有测试用例 pytest.main()

直接运行这个 Python 文件，就会自动执行所有test_开头的函数。

方法 2：指定运行某个文件 / 某个用例（最常用）

python

运行

import pytest if __name__ == '__main__': # 1. 运行当前文件所有用例 # pytest.main(["test_demo.py"]) # 2. 只运行当前文件（推荐写法） pytest.main([__file__]) # 3. 只运行文件中指定的某个测试函数 # pytest.mai