1、与 unittest 不同,pytest 使用的是 Python 自带的 assert 关键字来进行断言。
2、assert 关键字后面可以接一个表达式,只要表达式的最终结果为 True,那么断言通过,用例执行成功,否则用例执行失败。
示例:
1、创建test_assert.py文件
脚本代码:
#!/usr/bin/env python # -*- coding: utf-8 -*- """ 微信公众号:AllTests软件测试 """ # 异常信息 def f(): return 3 def test_function(): a = f() assert a % 2 == 0, "判断 a 为偶数,当前 a 的值为:%s" % a
2、运行结果:
抛出异常之后输出提示信息,方便查看是什么原因。
assert xx :判断 xx 为真
assert not xx :判断 xx 不为真
assert a in b :判断 b 包含 a
assert a == b :判断 a 等于 b
assert a != b :判断 a 不等于 b
1、创建test_assert2.py文件
脚本代码:
#!/usr/bin/env python # -*- coding: utf-8 -*- """ 微信公众号:AllTests软件测试 """ # 功能:用于计算a与b相加的和 def add(a,b): return a + b # 功能:用于判断素数 def is_prime(n): if n < 1: return False for i in range(2,n): if n % i == 0: return False return True # 断言:测试相等 def test_add_1(): assert add(3,4) == 7 # 断言:测试不相等 def test_add_2(): assert add(17,22) != 50 # 断言:测试小于或者等于 def test_add_3(): assert add(17,22) <= 50 # 断言:测试大于或者等于 def test_add_4(): assert add(17,22) >= 38 # 断言:测试包含 def test_in(): a = 'Hello' b = 'he' assert b in a # 断言:测试不包含 def test_not_in(): a = 'Hello' b = 'hi' assert b not in a # 断言:判断是否为True def test_true_1(): assert is_prime(13) # 断言:判断是否为True def test_true_2(): assert is_prime(7) is True # 断言:判断是否不为True def test_true_3(): assert is_prime(4) # 断言:判断是否不为True def test_true_4(): assert is_prime(6) is not True # 断言:判断是否为False def test_false_1(): assert is_prime(8) is False
2、运行结果:
为了编写有关引发的异常的断言,可以使用 pytest.raises() 作为这样的上下文管理器。
1、创建test_assert3.py文件
脚本代码:
#!/usr/bin/env python # -*- coding: utf-8 -*- """ 微信公众号:AllTests软件测试 """ import pytest # 异常断言 def test_zero_division(): with pytest.raises(ZeroDivisionError): 1 / 0
2、进一步完善异常断言的用法,创建test_zero_division_long函数。
脚本代码:
#!/usr/bin/env python # -*- coding: utf-8 -*- """ 微信公众号:AllTests软件测试 """ import pytest # 异常断言(断言它抛的异常是不是预期想要的) # 抛的异常是ZeroDivisionError: division by zero # excinfo:是一个异常信息实例 # 注意:断言 type 的时候,异常类型是不需要加引号的,断言 value 的时候需转 str def test_zero_division_long(): with pytest.raises(ZeroDivisionError) as excinfo: 1 / 0 # 断言异常 type 是 ZeroDivisionError assert excinfo.type == ZeroDivisionError # 断言异常 value 是 division by zero assert "division by zero" in str(excinfo.value)
3、match 关键字
可以将match关键字参数传递给上下文管理器,以测试正则表达式与异常的字符串表示形式是否匹配。
注意:这种方法只能断言value,不能断言type。
创建test_zero_division_match、test_zero_division_match2函数。
脚本代码:
#!/usr/bin/env python # -*- coding: utf-8 -*- """ 微信公众号:AllTests软件测试 """ import pytest # 异常断言-match def test_zero_division_match(): with pytest.raises(ZeroDivisionError, match=".*zero.*") as excinfo: 1 / 0 # 异常断言-match def test_zero_division_match2(): with pytest.raises(ZeroDivisionError, match="zero") as excinfo: 1 / 0
该match方法的regexp参数与re.search函数匹配,因此在上面的示例中match='zero'也可以使用。
运行结果:
4、检查断言装饰器
创建test_xfail函数。
脚本代码:
#!/usr/bin/env python # -*- coding: utf-8 -*- """ 微信公众号:AllTests软件测试 """ import pytest # 断言装饰器 @pytest.mark.xfail(raises=ZeroDivisionError) def test_xfail(): 1 / 0
运行结果:
代码抛出异常,但是和 raises 指定的异常类相匹配,所以不会断言失败。
它相当于一个检查异常装饰器,功能:检查是否有异常,不确定是否有异常。
with pytest.raise(ZeroDivisionError) 对于故意测试异常代码的情况,使用可能会更好。
而 @pytest.mark.xfail(raises=ZeroDivisionError) 对于检查未修复的错误(即可能会发生异常),使用检查断言可能会更好。