1. 前言

  • 与 unittest 不同，Pytest 使用的是 Python 自带的 assert 关键字来进行断言
  • assert 关键字后面可以接一个表达式，只要表达式的最终结果为 True，那么断言通过，用例执行成功，否则用例执行失败

2. assert 小栗子

想在抛出异常之后输出一些提示信息，执行之后就方便查看是什么原因了

  1. # 异常信息
  2. def f():
  3.     return 3
  5. def test_function():
  6.     a = f()
  7.     assert a % 2 == 0, "判断 a 为偶数，当前 a 的值为：%s" % a

2.1 执行结果

image.png

3. 常用断言

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

4. 异常断言

可以使用 pytest.raises 作为上下文管理器，当抛出异常时可以获取到对应的异常实例

  1. # 断言异常
  2. def test_zero_division():
  3.     with pytest.raises(ZeroDivisionError):
  4.         1 / 0

断言场景：断言它抛的异常是不是预期想要的
代码执行：1/0
预期结果：抛的异常是ZeroDivisionError: division by zero
如何断言：通常是断言异常的 type 和 value 值了
具体方式：这里 1/0 的异常类型是 ZeroDivisionError，异常的 value 值是 divisionby zero

  1. # 详细断言异常
  2. def test_zero_division_long():
  3.     with pytest.raises(ZeroDivisionError) as excinfo:
  4.         1 / 0
  6.     # 断言异常类型 type
  7.     assert excinfo.type == ZeroDivisionError
  8.     # 断言异常 value 值
  9.     assert "division by zero" in str(excinfo.value)

excinfo：是一个异常信息实例
主要属性：.type 、.value、.traceback
注意：断言 type 的时候，异常类型是不需要加引号的，断言 value 值的时候需转 str

5. 拓展一：match

可以将 match 关键字参数传递给上下文管理器，以测试正则表达式与异常的字符串表示形式是否匹配
注意：这种方法只能断言value，不能断言type
image.png

  1. # 自定义消息
  2. def test_zero_division_long():
  3.     with pytest.raises(ZeroDivisionError, match=".*zero.*") as excinfo:
  4.         1 / 0

该 match 方法的regexp参数与 re.search 函数匹配，因此在上面的示例中 match=’zero’ 也可以使用
image.png
image.png

6. 拓展二：检查断言装饰器

  1. # 断言装饰器
  2. @pytest.mark.xfail(raises=ZeroDivisionError)
  3. def test_f():
  4.     1 / 0

6.1 执行结果

image.png

6.2 知识点

  • 代码抛出异常，但是和raises指定的异常类相匹配，所以不会断言失败
  • 它相当于一个检查异常装饰器，功能：检查是否有异常，不确定是否有异常
  • with Pytest.raise(ZeroDivisionError)对于故意测试异常代码的情况，使用可能会更好
  • 而@Pytest.mark.xfail(raises=ZeroDivisionError)对于检查未修复的错误（即，可能会发生异常），使用检查断言可能会更好