正确处理 ZeroDivisionError：Pytest 中的异常测试指南

心靈之曲

发布时间：2025-08-16 19:12:01

854人浏览过

来源于php中文网

原创

正确处理 zerodivisionerror：pytest 中的异常测试指南

本文旨在帮助开发者理解并正确处理使用 Pytest 进行单元测试时遇到的 ZeroDivisionError 异常。我们将深入探讨如何编写测试用例，以验证特定函数在遇到除零情况时是否按预期抛出 ZeroDivisionError 异常，并提供代码示例和注意事项，确保测试的准确性和可靠性。

在编写单元测试时，验证代码在特定情况下是否抛出预期的异常至关重要。当涉及到 ZeroDivisionError 时，尤其需要确保函数能够正确处理除数为零的情况。Pytest 提供了 pytest.raises() 上下文管理器，可以优雅地处理这种情况。关键在于理解 pytest.raises() 期望被测试的代码显式地抛出异常，而不是仅仅捕获并处理异常。

让我们通过一个实际的例子来理解这个概念。假设我们有一个 fuel.py 文件，其中包含 convert 函数，该函数将分数转换为百分比：

def convert(fraction):
    try:
        numerator, denominator = fraction.split('/')
        numerator = int(numerator)
        denominator = int(denominator)
        percentage = round((numerator/denominator)*100)
        if percentage > 100:
            raise ValueError
        else:
            return percentage
    except ValueError:
        return -1
    except ZeroDivisionError:
        return -1

def gauge(percentage):
    if percentage <= 1:
        return 'E'
    elif percentage >= 99:
        return 'F'
    else:
        return f'{percentage}%'

现在，我们编写一个 test_fuel.py 文件来测试 convert 函数，特别是针对 ZeroDivisionError 的情况。

from fuel import convert, gauge

import pytest

def test_gauge():
    assert gauge(1) == 'E'
    assert gauge(50) == '50%'
    assert gauge(99) == 'F'

def test_convert():
    assert convert('5/3') == -1
    assert convert('1/2') == 50
    with pytest.raises(ZeroDivisionError):
        convert('5/0')
    with pytest.raises(ValueError):
        convert('cat/dog')

关键点在于 with pytest.raises(ZeroDivisionError): 语句。 pytest.raises() 上下文管理器会执行其内部的代码块，并断言是否抛出了指定的异常（在本例中为 ZeroDivisionError）。如果 convert('5/0') 没有抛出 ZeroDivisionError，测试将会失败，并显示 "DID NOT RAISE " 错误。

正确处理 ZeroDivisionError 的方法:

HIX.AI

HIX.AI是一个多功能的一体化AI写作助手，集成了120多种AI写作工具，支持50多种语言，能够满足各种写作需求。

下载

要使测试通过，convert 函数需要显式地抛出 ZeroDivisionError，而不是仅仅捕获它。修改 fuel.py 中的 convert 函数，使其在分母为零时抛出 ZeroDivisionError：

def convert(fraction):
    try:
        numerator, denominator = fraction.split('/')
        numerator = int(numerator)
        denominator = int(denominator)
        if denominator == 0:
            raise ZeroDivisionError
        percentage = round((numerator/denominator)*100)
        if percentage > 100:
            raise ValueError
        else:
            return percentage
    except ValueError:
        return -1
    except ZeroDivisionError:
        raise ZeroDivisionError # 重新抛出异常

或者，移除 convert 函数中的 except ZeroDivisionError 块，让异常直接抛出：

def convert(fraction):
    try:
        numerator, denominator = fraction.split('/')
        numerator = int(numerator)
        denominator = int(denominator)
        percentage = round((numerator/denominator)*100)
        if percentage > 100:
            raise ValueError
        else:
            return percentage
    except ValueError:
        return -1

总结与注意事项：