前言

在 pytest 中，斷言（assertion）是測(cè)試用例中最核心的部分之一。斷言用于驗(yàn)證函數(shù)或方法的執(zhí)行結(jié)果是否符合預(yù)期。正確的斷言不僅能夠幫助我們發(fā)現(xiàn)錯(cuò)誤，還能提供詳細(xì)的錯(cuò)誤信息，便于快速定位問(wèn)題。

1. 基本斷言

1.1 示例代碼

假設(shè)我們有一個(gè)簡(jiǎn)單的函數(shù) add，我們想要編寫一個(gè)測(cè)試用例來(lái)驗(yàn)證這個(gè)函數(shù)的行為。

# 文件名：mylib.py

def add(a, b):
    return a + b

接下來(lái)，我們編寫一個(gè)測(cè)試用例來(lái)驗(yàn)證 add 函數(shù)。

# 文件名：test_mylib.py

def test_add():
    assert add(1, 2) == 3
    assert add(-1, 1) == 0
    assert add(0, 0) == 0

2. 多重?cái)嘌?/h3>

有時(shí)候我們需要在一個(gè)測(cè)試用例中驗(yàn)證多個(gè)條件?？梢允褂枚鄠€(gè) assert 語(yǔ)句來(lái)實(shí)現(xiàn)這一點(diǎn)。

2.1 示例代碼

# 文件名：test_mylib.py

def test_add_multiple_conditions():
    assert add(1, 2) == 3
    assert add(-1, 1) == 0
    assert add(0, 0) == 0
    assert add(100, 200) == 300

3. 斷言異常

有時(shí)我們需要驗(yàn)證函數(shù)是否會(huì)拋出特定的異常。可以使用 with pytest.raises() 上下文管理器來(lái)實(shí)現(xiàn)這一點(diǎn)。

3.1 示例代碼

假設(shè)我們有一個(gè)函數(shù) divide，它可能會(huì)拋出 ZeroDivisionError。

# 文件名：mylib.py

def divide(a, b):
    return a / b

接下來(lái)，我們編寫一個(gè)測(cè)試用例來(lái)驗(yàn)證 divide 函數(shù)是否會(huì)拋出異常。

# 文件名：test_mylib.py

def test_divide_zero():
    with pytest.raises(ZeroDivisionError):
        divide(1, 0)

4. 斷言特定異常消息

有時(shí)我們需要驗(yàn)證拋出的異常是否包含特定的消息?？梢允褂?match 參數(shù)來(lái)實(shí)現(xiàn)這一點(diǎn)。

4.1 示例代碼

# 文件名：test_mylib.py

def test_divide_zero_message():
    with pytest.raises(ZeroDivisionError, match="division by zero"):
        divide(1, 0)

5. 斷言列表和字典

在驗(yàn)證數(shù)據(jù)結(jié)構(gòu)時(shí)，經(jīng)常需要斷言列表或字典是否符合預(yù)期。

5.1 示例代碼

假設(shè)我們有一個(gè)函數(shù) process_data，它返回一個(gè)列表。

# 文件名：mylib.py

def process_data():
    return [1, 2, 3]

接下來(lái)，我們編寫一個(gè)測(cè)試用例來(lái)驗(yàn)證 process_data 函數(shù)的返回值。

# 文件名：test_mylib.py

def test_process_data():
    assert process_data() == [1, 2, 3]

5.2 斷言字典

假設(shè)我們有一個(gè)函數(shù) get_user_info，它返回一個(gè)字典。

# 文件名：mylib.py

def get_user_info():
    return {"name": "Alice", "age": 25}

接下來(lái)，我們編寫一個(gè)測(cè)試用例來(lái)驗(yàn)證 get_user_info 函數(shù)的返回值。

# 文件名：test_mylib.py

def test_get_user_info():
    assert get_user_info() == {"name": "Alice", "age": 25}

6. 斷言浮點(diǎn)數(shù)

在處理浮點(diǎn)數(shù)時(shí)，由于浮點(diǎn)數(shù)的精度問(wèn)題，直接使用 == 可能會(huì)導(dǎo)致誤判?？梢允褂?pytest.approx 來(lái)實(shí)現(xiàn)這一點(diǎn)。

6.1 示例代碼

假設(shè)我們有一個(gè)函數(shù) calculate_pi，它返回一個(gè)近似值。

# 文件名：mylib.py

def calculate_pi():
    return 3.14159265358979323846

接下來(lái)，我們編寫一個(gè)測(cè)試用例來(lái)驗(yàn)證 calculate_pi 函數(shù)的返回值。

# 文件名：test_mylib.py

def test_calculate_pi():
    assert calculate_pi() == pytest.approx(3.141592653589793, abs=1e-10)

7. 斷言字符串

在驗(yàn)證字符串時(shí)，可以使用 assert 來(lái)檢查字符串是否包含特定的子串。

7.1 示例代碼

假設(shè)我們有一個(gè)函數(shù) greet，它返回一個(gè)問(wèn)候字符串。

# 文件名：mylib.py

def greet(name):
    return f"Hello, {name}!"

接下來(lái)，我們編寫一個(gè)測(cè)試用例來(lái)驗(yàn)證 greet 函數(shù)的返回值。

# 文件名：test_mylib.py

def test_greet():
    assert "Hello, Alice!" == greet("Alice")
    assert "Hello, Bob!" == greet("Bob")

8. 斷言元組

在驗(yàn)證元組時(shí)，可以使用 assert 來(lái)檢查元組是否符合預(yù)期。

8.1 示例代碼

假設(shè)我們有一個(gè)函數(shù) get_coordinates，它返回一個(gè)坐標(biāo)元組。

# 文件名：mylib.py

def get_coordinates():
    return (1.0, 2.0)

接下來(lái)，我們編寫一個(gè)測(cè)試用例來(lái)驗(yàn)證 get_coordinates 函數(shù)的返回值。

# 文件名：test_mylib.py

def test_get_coordinates():
    assert get_coordinates() == (1.0, 2.0)

9. 斷言列表和字典的子集

在驗(yàn)證列表或字典的子集時(shí)，可以使用 Python 的內(nèi)置函數(shù) all 或 any 來(lái)實(shí)現(xiàn)這一點(diǎn)。

9.1 示例代碼

假設(shè)我們有一個(gè)函數(shù) get_students，它返回一個(gè)學(xué)生列表。

# 文件名：mylib.py

def get_students():
    return [{"name": "Alice", "age": 25}, {"name": "Bob", "age": 30}]

接下來(lái)，我們編寫一個(gè)測(cè)試用例來(lái)驗(yàn)證 get_students 函數(shù)的返回值。

# 文件名：test_mylib.py

def test_get_students():
    students = get_students()
    assert all(student["age"] >= 18 for student in students)
    assert any(student["name"] == "Alice" for student in students)

10. 斷言日志

在驗(yàn)證日志輸出時(shí)，可以使用 caplog 固定夾（fixture）來(lái)捕獲日志輸出。

10.1 示例代碼

假設(shè)我們有一個(gè)函數(shù) log_error，它記錄一條錯(cuò)誤日志。

# 文件名：mylib.py

import logging
logger = logging.getLogger(__name__)
def log_error(message):
    logger.error(message)

接下來(lái)，我們編寫一個(gè)測(cè)試用例來(lái)驗(yàn)證 log_error 函數(shù)的日志輸出。

# 文件名：test_mylib.py

import logging
import pytest
def test_log_error(caplog):
    caplog.set_level(logging.ERROR)
    log_error("An error occurred")
    assert "An error occurred" in caplog.text

11. 總結(jié)

通過(guò)以上示例，我們?cè)敿?xì)介紹了 pytest 中常用的斷言方法，并通過(guò)具體的示例代碼展示了它們的使用方法：

基本斷言：驗(yàn)證函數(shù)的返回值。

多重?cái)嘌裕候?yàn)證多個(gè)條件。

斷言異常：驗(yàn)證函數(shù)是否會(huì)拋出特定的異常。

斷言特定異常消息：驗(yàn)證拋出的異常是否包含特定的消息。

斷言列表和字典：驗(yàn)證數(shù)據(jù)結(jié)構(gòu)是否符合預(yù)期。

斷言浮點(diǎn)數(shù)：驗(yàn)證浮點(diǎn)數(shù)是否近似相等。

斷言字符串：驗(yàn)證字符串是否包含特定的子串。

斷言元組：驗(yàn)證元組是否符合預(yù)期。

斷言列表和字典的子集：驗(yàn)證列表或字典的子集。

斷言日志：驗(yàn)證日志輸出。