Python测试框架pytest（13）assert断言

2022-01-22 10:00:04

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、运行结果：

抛出异常之后输出提示信息，方便查看是什么原因。

1、常用断言

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、运行结果：

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) 对于检查未修复的错误（即可能会发生异常），使用检查断言可能会更好。

码农公寓

1、常用断言

2、异常断言

相关文章