【pytest系列】- assert断言的使用

unittest断言方式是用过框架自己实现的，即 self.assertEqual() 等，当我们使用pytest框架后，这种断言方式是不可用的，因为测试类不会再继承unittest.TestCase类，因此pytest使用的断言是使用python内置的断言assert，相对于unittest各种各样的断言方法，内置的断言会更简洁，它只有一种，详情如下。

常用断言

​ 关键代码： assert 表达式, 错误提示信息 。

def test_login():

    # 使用python内置的断言

    assert 1 == 2, '1是不等于2的'	# 错误信息也可以不写

test_login()

​ 运行结果：

C:\software\python\python.exe D:/learn/test.py

Traceback (most recent call last):

  File "D:/learn/test.py", line 13, in <module>

    test_login()

  File "D:/learn/test.py", line 11, in test_login

    assert 1 == 2, '1是不等于2的'

AssertionError: 1是不等于2的

Process finished with exit code 1

​ 常用的断言表达有以下几种 ：

• assert xx ：判断xx为真
• assert not xx ：判断xx不为真
• assert a in b ：判断b包含a
• assert a not in b ：判断b不包含a
• assert a == b ：判断a等于b
• assert a != b ：判断a不等于b

异常断言

​ 可以使用 pytest.raises 作为上下文管理器，当抛出异常时可以获取到对应的异常实例，然后断言它抛出的异常是不是预期的。

# 断言异常

def test_zero_division():

    with pytest.raises(ZeroDivisionError):

        1 / 0

​ 如任意一个自然数除以0时，由于0不能作为被除数，执行时就会抛出 ZeroDivisionError 异常，提示 division by zero 。

# 详细断言异常

def test_zero_division_long():

    with pytest.raises(ZeroDivisionError) as excinfo:

        1 / 0

    # 断言异常类型 type

    assert excinfo.type == ZeroDivisionError

    # 断言异常 value 值

    assert "division by zero" in str(excinfo.value)

​ 注意： 断言type的时候，异常类型是不需要加引号的，断言value值的时候需转str类型。

扩展

正则断言

​ 可以将 match 关键字参数传递给上下文管理器，以测试正则表达式与异常的字符串表示形式是否匹配

​ 注意： 这种方法只能断言value，不能断言type

# 自定义消息

def test_zero_division_long():

    with pytest.raises(ZeroDivisionError, match=".*zero.*") as excinfo:

        1 / 0

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

# 自定义消息

def test_zero_division_long():

    with pytest.raises(ZeroDivisionError, match="zero") as excinfo:

        1 / 0

​ 运行结果：

检查断言装饰器

​ 当代码抛出异常时，如果和raises指定的异常类相匹配，就不会断言失败。

​ 它相当于一个 检查异常装饰器 ，目的是检查是否有异常(不确定是否有异常)，不同使用场景：

• with pytest.raise(ZeroDivisionError) 对于故意测试异常代码的情况，使用可能会更好
• 而 @pytest.mark.xfail(raises=ZeroDivisionError) 对于检查未修复的错误（即可能会发生异常），使用检查断言可能会更好

# 断言装饰器

@pytest.mark.xfail(raises=ZeroDivisionError)

def test_f():

    1 / 0

​ 执行结果：