Pytest测试用例使用小结

基础使用

Pytest 测试用例实现代码

import pytest
from server.service import [e-mail]
def service():return Service(logger)class TestService:@classmethoddef setup_class(cls):"""初始化设置一次:return:"""logger.info("{}----类初始化设置".format(cls.__name__))@classmethoddef teardown_class(cls):"""结束后，清除设置:return:"""logger.info("{}----类清除设置".format(cls.__name__))def setup_method(self):"""执行每个方法前，初始化:return:"""logger.info("{}----执行每个方法前，初始化".format(self.__module__))def teardown_method(self):"""执行每个方法后，清除设置:return:"""logger.info("{}----执行每个方法后，清除设置".format(self.__module__)

可在类中添加类前后和方法前后固定执行的操作。

定义一个测试实例

@pytest.mark.parametrize(['params', 'expected'],  # 也可使用字符串传入'params, expected'传入[({"title": "测试用例"},Code.OK),  # 第一组测试参数]
)
def test_add(self, service, params, expected):result = dialogue_service.add_dialogue(**params)assert result.code == Code.OK

假设TestService类在test_service.py中，运行该类，添加main方法，之后在该文件目录下运行命令python test.service.py

if __name__ == '__main__':pytest.main()

运行测试类中一个测试用例

可在测试函数函数上添加标识装饰器@pytest.mark.[运行名称]，如pytest.marks.add

@pytest.mark.add
def test_add(self, service, params, expected):result = service.add(**params)assert result.code == Code.OK

如果过需要添加参数，则标识按如下方式添加：

@pytest.mark.parametrize(['params', 'expected'],[pytest.param({"title": "测试用例",},Code.OK,marks=pytest.mark.add),]
)
def test_add(self, service, params, expected):result = service.add(**params)assert result.code == Code.OK

运行命令：pytest -m add

或直接使用命令，指定要运行的测试用例：pytest test_service.py::TestService:test_add

项目包引入路径问题

import os
import sys
current_path = os.getcwd()
sys.path.append(os.path.join(current_path.split('project_name')[0], 'project_name'))  # 不添加测试时无法找到正确路径

将当前项目路径添加到系统中。

参考：

Python自动化测试框架unittest与pytest的区别_pytest和unittest哪个好-CSDN博客