一、pytest是什么
pytest是一个基于Python的测试框架,它可以轻松的创建单元测试、功能测试和集成测试。与其他Python测试框架相比,pytest具有简洁易懂的测试设计风格、跨平台支持、插件机制支持和丰富的API和钩子函数等优点。
在开始使用pytest之前,需要安装pytest模块。可以在终端中使用pip安装:
pip install pytest
安装成功后,就可以在项目目录下创建测试文件开始测试了。
二、pytest的基础用法
pytest最常用的两个参数是-m和-k。其中-m参数用于选择标记(mark)测试用例,-k参数用于根据表达式来选择测试用例。
-m参数
在pytest中,可以为测试用例添加标记,以便于测试时选择特定的测试用例。例如,可以为一组测试用例添加一个“slow”标记,表示测试用例运行较慢,需要更长的时间。可以使用-m参数来选择标记(mark)的测试用例:
pytest -m <mark>
其中<mark>是标记的名称,可以是一个字符串或多个字符串,多个字符串之间用“and”、“or”、“not”进行逻辑连接。例如,如果需要运行所有带有“slow”标记的测试用例,可以执行如下命令:
pytest -m slow
如果需要运行所有带有“slow”标记或带有“gui”标记的测试用例,可以执行如下命令:
pytest -m "slow or gui"
-k参数
在pytest中,可以通过表达式来选择特定的测试用例。例如,可以使用“test_login”来选择所有名称中包含“test_login”的测试用例。可以使用-k参数来根据表达式选取测试用例:
pytest -k <expression>
其中<expression>是一个表达式,用于选取测试用例。表达式可以包含测试用例名称、测试用例文件名或标记等信息。例如,如果需要运行所有测试用例名称中包含“login”的测试用例,可以执行如下命令:
pytest -k login
如果需要运行所有在tests文件夹下,名称中包含“login”字符的测试用例,可以执行如下命令:
pytest -k "login and tests"
三、pytest的进阶用法
pytest具有很多高级特性,如参数化、测试夹具、插件机制等。
参数化
参数化是pytest中非常强大和常用的特性之一,可以通过参数化来简化测试用例和测试数据的编写。在测试用例中使用pytest.mark.parametrize装饰器来标记参数化的参数,然后在测试函数的参数列表中使用这些参数。例如:
import pytest
@pytest.mark.parametrize("arg1, arg2, expected", [
(1, 2, 3),
(2, 3, 5),
(10, 20, 30)
])
def test_addition(arg1, arg2, expected):
result = arg1 + arg2
assert result == expected
以上代码可以传入不同的参数,执行三次函数运算,检测运算是否正确。
测试夹具
pytest支持各种类型的测试夹具,例如函数级别的夹具、模块级别的夹具、会话级别的夹具、自定义夹具等。使用夹具可以优化测试结构、减少代码冗余、提升测试效率和可读性。定义的夹具可以使用@pytest.fixture装饰器标记,表示当前函数是一个夹具函数,使用夹具时在测试函数的参数列表中使用夹具名即可。例如:
import pytest
@pytest.fixture()
def text_data():
return 'pytest'
def test_demo(text_data):
assert text_data == 'pytest'
以上代码利用fixture返回的数据,测试text_data是否等于'pytest',从而达到测试夹具的目的。
插件机制
pytest具有丰富的插件机制,支持各种类型的插件,如测试夹具、收集器、HTML测试报告、Junit测试报告、测试覆盖率等。可以在pytest.ini配置文件中指定插件,也可以在命令行中通过--install-plugin选项安装插件。例如,如果需要安装pytest-html插件,可以执行如下命令:
pip install pytest-html
安装成功后,在命令行中执行如下命令,可以生成HTML格式的测试报告:
pytest --html=report.html
四、pytest的实战案例
以下是一个简单的pytest测试代码,用于测试一个计算器程序:
import pytest
import calculator
def test_addition():
c = calculator.Calculator()
result = c.add(1, 2)
assert result == 3
def test_subtraction():
c = calculator.Calculator()
result = c.subtract(2, 1)
assert result == 1
@pytest.mark.parametrize("arg1, arg2, expected", [
(1, 2, 3),
(2, 3, 5),
(10, 20, 30)
])
def test_addition_with_parameters(arg1, arg2, expected):
c = calculator.Calculator()
result = c.add(arg1, arg2)
assert result == expected
@pytest.fixture()
def test_data():
data = {
'arg1': 1,
'arg2': 2,
'expected': 3
}
return data
def test_addition_with_fixture(test_data):
c = calculator.Calculator()
result = c.add(test_data['arg1'], test_data['arg2'])
assert result == test_data['expected']