Python 异常处理与单元测试结合实践

异常处理与单元测试结合能提升代码健壮性，需用pytest.raises或unittest.assertRaises测试异常类型、消息及处理逻辑，避免过度捕获和静默失败，确保正常与异常路径均被覆盖。

Python的异常处理和单元测试，在我看来，它们就像是代码健壮性的左右手。异常处理确保程序在运行时遇到预料之外或错误情况时能优雅地应对，不至于直接崩溃；而单元测试则是在开发阶段就对这些“应对措施”进行验证，确保它们真的能按我们设想的方式工作。两者结合起来，不仅能提前发现潜在问题，还能显著提升软件的稳定性和可维护性，让我们的代码在面对各种“不确定性”时，都能表现得更加可靠。

解决方案

将异常处理和单元测试结合起来，核心在于我们不仅要测试代码在“正常路径”（happy path）下的行为，更要积极地测试它在“异常路径”（unhappy path）下的表现。这意味着我们要编写专门的测试用例，来触发代码中预期的异常，并验证这些异常是否被正确地捕获、处理，或者以预期的形式被抛出。

具体来说，这包括几个层面：

1. 验证异常是否被正确抛出： 当输入不合法或条件不满足时，我们的函数应该按照设计抛出特定类型的异常。单元测试需要确认，在这些情况下，确实是正确的异常类型被抛出，并且异常消息也符合预期。
2. 验证异常处理逻辑： 如果我们的代码中包含了

   try...except

   登录后复制
   块来捕获并处理异常（例如，记录日志、返回默认值、重新抛出更高层次的异常），单元测试就需要验证这些处理逻辑是否有效。比如，当一个

   FileNotFoundError

   登录后复制
   被捕获后，程序是否真的记录了日志，或者是否返回了一个预设的空列表，而不是崩溃。
3. 验证异常的副作用： 异常处理可能会导致一些状态变化或者产生一些副作用（比如数据库回滚、资源关闭）。单元测试也应该验证这些副作用是否按预期发生。

我个人觉得，很多人在写单元测试时，往往只关注“功能是否正常”，却忘了“功能在异常情况下是否依然正常”。但实际上，很多生产环境的问题，恰恰是出在这些被忽视的异常路径上。通过有意识地将两者结合，我们能更全面地评估代码的鲁棒性。

立即学习“Python免费学习笔记（深入）”；

为什么在Python中，对异常处理进行单元测试是不可或缺的？

回想一下，我们有多少次在生产环境遇到莫名其妙的崩溃，结果发现是某个本该被捕获的异常漏掉了，或者被捕获后却什么都没做（比如一个空的

except:

首先，它确保了代码的防御性。我们写异常处理，是为了让程序在面对错误时有“自愈”或“优雅降级”的能力。如果没有单元测试，我们怎么知道这些防御机制真的起作用了？比如，一个文件读取函数，如果文件不存在，我们希望它抛出

FileNotFoundError

其次，它验证了错误信息的准确性与一致性。一个好的异常信息，能大大帮助我们调试问题，甚至能给用户提供有用的反馈。通过单元测试，我们可以确认抛出的异常信息是否清晰、准确，是否包含了所有必要的信息。这避免了在调试时，我们面对一个只有“Something went wrong”的模糊错误信息而束手无策。

再者，它提升了开发者的信心。当我为一个关键模块编写了详尽的异常处理测试后，我知道即使未来系统在某些极端条件下运行，我的代码也能尽可能地保持稳定。这种信心，在面对复杂系统和快速迭代时，是极其宝贵的。它让我可以更放心地进行重构，因为我知道，即使不小心引入了新的bug，测试也能帮我揪出来。

如何有效地利用

pytest.raises

登录后复制

或

unittest.assertRaises

登录后复制

来测试异常？

在Python中，测试异常抛出行为，

pytest

unittest

pytest.raises

使用

pytest.raises

青柚面试

简单好用的日语面试辅助工具

57

查看详情

pytest.raises

with

import pytest

def divide(a, b):
    if b == 0:
        raise ValueError("Cannot divide by zero!")
    if not isinstance(a, (int, float)) or not isinstance(b, (int, float)):
        raise TypeError("Inputs must be numbers.")
    return a / b

def test_divide_by_zero():
    with pytest.raises(ValueError) as excinfo:
        divide(10, 0)
    # 验证异常类型
    assert excinfo.type is ValueError
    # 验证异常消息
    assert "Cannot divide by zero!" in str(excinfo.value)

def test_divide_with_non_numeric_input():
    with pytest.raises(TypeError, match="Inputs must be numbers."):
        divide("a", 2)
    # 也可以不使用as excinfo，直接匹配消息
    with pytest.raises(TypeError, match="Inputs must be numbers."):
        divide(10, "b")

def test_divide_success():
    # 确保在正常情况下不会抛出异常
    assert divide(10, 2) == 5.0

这里我发现很多人刚开始用这个功能时，可能会直接写

pytest.raises(Exception)

使用

unittest.assertRaises

unittest

assertRaises

import unittest

def divide_unittest(a, b):
    if b == 0:
        raise ValueError("Cannot divide by zero!")
    if not isinstance(a, (int, float)) or not isinstance(b, (int, float)):
        raise TypeError("Inputs must be numbers.")
    return a / b

class TestDivideUnittest(unittest.TestCase):
    def test_divide_by_zero(self):
        with self.assertRaises(ValueError) as cm:
            divide_unittest(10, 0)
        self.assertIn("Cannot divide by zero!", str(cm.exception))

    def test_divide_with_non_numeric_input(self):
        # unittest.assertRaisesRegex 可以用来匹配异常消息
        with self.assertRaisesRegex(TypeError, "Inputs must be numbers."):
            divide_unittest("a", 2)

    def test_divide_success(self):
        self.assertEqual(divide_unittest(10, 2), 5.0)

if __name__ == '__main__':
    unittest.main()

无论是

pytest

unittest

结合异常处理与单元测试的最佳实践与常见误区

结合异常处理和单元测试，并不是简单地写几个

pytest.raises

最佳实践：

1. 具体化异常类型： 总是尝试抛出或捕获最具体的异常类型。比如，是

   ValueError

   登录后复制
   就不要用

   Exception

   登录后复制
   。在测试时，也应该测试具体的异常类型。这能让你的代码和测试都更加精确，避免掩盖其他潜在的问题。
2. 测试异常消息： 异常消息是调试的关键。确保你的测试不仅验证了异常类型，还验证了异常消息是否清晰、准确，是否包含了所有必要的信息。
3. 测试异常处理逻辑的副作用： 如果你的

   except

   登录后复制
   块中包含了日志记录、资源清理、状态重置等逻辑，那么单元测试应该验证这些逻辑是否被正确执行。例如，可以使用

   unittest.mock

   登录后复制
   来模拟日志记录器，并断言日志函数是否被调用，以及调用时的参数是否正确。
4. 测试“不抛出异常”的情况： 除了测试异常情况，也要确保在正常输入下，代码不会意外地抛出异常。这看起来很基础，但有时为了覆盖所有分支，我们可能会无意中引入一些在正常路径下触发异常的逻辑。
5. 模拟外部依赖的异常： 很多时候，异常是由外部服务、数据库或文件系统引起的。在单元测试中，我们应该使用

   mock

   登录后复制
   来模拟这些外部依赖抛出异常的情况，从而测试我们代码的健壮性。
6. 优先测试行为，而非实现： 你的测试应该关注代码在异常情况下的行为表现（比如，抛出了什么异常，或者执行了什么恢复逻辑），而不是它内部是如何实现的。这样，即使未来重构了异常处理的内部机制，只要外部行为不变，测试依然有效。

常见误区：

1. 过度捕获（

   except Exception:

   登录后复制
   ）： 这是一个非常常见的错误。捕获

   Exception

   登录后复制
   会捕获所有异常，包括

   SystemExit

   登录后复制
   和

   KeyboardInterrupt

   登录后复制
   ，这可能会导致程序无法正常退出，或者掩盖了真正的问题。在测试中，如果你的代码过度捕获，那么测试特定的异常类型就会变得困难。
2. “静默失败”的异常处理（

   except SomeError: pass

   登录后复制
   ）： 捕获了异常却不做任何处理，这比直接崩溃更糟糕，因为它让问题隐蔽化，导致更难发现和调试。你的测试应该能够识别出这种“静默失败”的情况。
3. 只测试抛出，不测试处理： 很多开发者会测试他们的函数是否在特定条件下抛出了异常，但却忘了测试当这个异常被上层代码捕获后，上层代码的处理逻辑是否正确。例如，一个函数内部抛出

   ValueError

   登录后复制
   ，上层

   try...except

   登录后复制
   捕获后，是返回

   None

   登录后复制
   ，还是记录日志，这些都需要测试。
4. 测试过于宽泛的异常： 就像前面提到的，用

   pytest.raises(Exception)

   登录后复制
   来测试所有异常是不好的习惯。它可能会导致你的测试通过，但实际上代码抛出了一个你意料之外的、更具体的异常。
5. 忘记测试自定义异常： 如果你定义了自定义异常，那么务必为这些自定义异常编写测试，确保它们在正确的场景下被抛出，并且包含了正确的属性和信息。

我曾经踩过一个坑，就是测试一个文件读取函数，只测试了文件存在的情况，结果上线后遇到文件不存在，直接崩溃了。后来才意识到，对

FileNotFoundError

以上就是Python 异常处理与单元测试结合实践的详细内容，更多请关注php中文网其它相关文章！