本文旨在帮助开发者理解并正确处理使用 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 <class 'ZeroDivisionError'>" 错误。
正确处理 ZeroDivisionError 的方法:
要使测试通过，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总结与注意事项：
- pytest.raises() 期望被测试的代码块显式地抛出指定的异常。
- 如果函数内部捕获了异常但没有重新抛出，pytest.raises() 测试将会失败。
- 确保你的代码在遇到预期异常情况时，能够正确地抛出异常，以便 pytest.raises() 能够捕获并验证。
- 在某些情况下，移除不必要的异常处理块，让异常自然地向上冒泡，可能是更简洁和有效的解决方案。
- 仔细阅读 Pytest 的官方文档，深入理解 pytest.raises() 的用法和行为。
通过理解 pytest.raises() 的工作原理，并确保你的代码能够正确地抛出异常，你可以编写更可靠和准确的单元测试，从而提高代码的质量和稳定性。
