一、前言
- 与unittest断言不同,pytest使用的是python自带的assert关键字来进行断言
- assert关键字后面可以接一个表达式,只要表达式的最终结果为True,那么断言通过,用例执行成功,否则用例执行失败
二、常用断言
pytest 里面断言实际上就是 python 里面的 assert 断言方法,常用的有以下几种
- assert xx :判断 xx 为真
- assert not xx :判断 xx 不为真
- assert a in b :判断 b 包含 a
- assert a == b :判断 a 等于 b
- assert a != b :判断 a 不等于 b
三、异常后输出提示信息
抛出异常之后输出一些提示信息,执行之后就方便查看是什么原因了
# 异常信息
def func():
return 6
def test_04():
a = func()
assert a % 2 == 0, "判断a为偶数,当前a的值为:%s" % a
执行结果如下:
四、异常断言
可以使用 pytest.raises 作为上下文管理器,当抛出异常时可以获取到对应的异常实例
# 断言异常
def test_zero_division():
with pytest.raises(ZeroDivisionError):
1 / 0
说明:
- 断言场景:断言它抛的异常是不是预期想要的
- 代码执行:1/0
- 预期结果:抛的异常是ZeroDivisionError: division by zero
- 如何断言:通常是断言异常的 type 和 value 值了
- 具体方式:这里 1/0 的异常类型是 ZeroDivisionError,异常的 value 值是 divisionby zero
断言异常信息和类型如下:
# 断言异常
import pytest
def test_zero_division():
with pytest.raises(ZeroDivisionError) as info:
1 / 0
# 断言异常类型 type
assert info.type == ZeroDivisionError
# 断言异常value值
assert "division by zero" in str(info.value)
注意:
- info :是一个异常信息实例,主要属性有: .type 、 .value 、 .traceback
- 断言 type 的时候,异常类型是不需要加引号的,断言 value值的时候需转 str
执行结果如下图:
五、其余方式
5.1.match
可以将 match 关键字参数传递给上下文管理器,以测试正则表达式与异常的字符串表示形式是否匹配
# 自定义消息
import pytest
def test_zero_division02():
with pytest.raises(ZeroDivisionError, match=".*zero.*") as info:
1 / 0
注意:
- 这种方法只能断言value,不能断言type
-
ZeroDivisionError
:指定期望捕获的异常类型为 ZeroDivisionError。 -
match="zero"
:指定异常消息必须匹配正则表达式zero
,也就是异常消息中必须包含 “zero” 字符串
# 自定义消息
import pytest
def test_zero_division02():
with pytest.raises(ZeroDivisionError, match=".*zero.*") as info:
1 / 0
def test_zero_division03():
with pytest.raises(ZeroDivisionError, match="zero") as info:
1 / 0
执行后结果如下:
5.2.检查断言装饰器
可以通过装饰器进行检查断言
# 断言装饰器
import pytest
@pytest.mark.xfail(raises=ZeroDivisionError)
def test_error():
1/0
执行结果如下:
注意:
- 代码抛出异常,由于和raises指定的异常类相匹配,所以不会断言失败
- 相当于一个检查异常装饰器,功能就是检查是否有异常,但是不确定是否有异常
- with pytest.raise(ZeroDivisionError) 对于故意测试异常代码的情况,效果比较好
- 而@pytest.mark.xfail(raises=ZeroDivisionError) 对于检查未修复的错误(即可能会发生异常),使用检查断言可能会更好