如何编写Python单元测试以捕获已处理异常并使测试失败？

你的问题核心在于：被测试函数内部用try-except捕获并吞掉了异常，导致外部测试代码无法感知到异常的发生，所以常规的try-except + self.fail方案无效。要实现“只要函数执行过程中出现过异常（无论是否被内部处理）就测试失败”的需求，可以通过追踪函数执行过程中的异常事件来实现。

方案：利用sys.settrace追踪内部异常 #

Python的sys.settrace可以设置一个全局追踪函数，该函数会在代码执行的关键事件（比如函数调用、异常抛出等）时被触发。我们可以利用这个机制，记录函数执行期间是否有任何异常被抛出，哪怕异常被内部捕获了。

完整测试代码如下：

import sys
import unittest

def do_something():
    try:
        # 模拟可能抛出异常的代码
        raise ValueError("测试异常")
    except:
        print("An exception was raised")

class TestDoSomething(unittest.TestCase):
    def test_do_something(self):
        exception_occurred = False

        # 定义追踪函数：捕获异常事件
        def trace_func(frame, event, arg):
            nonlocal exception_occurred
            # 当发生异常时，标记异常已出现
            if event == 'exception':
                exception_occurred = True
            # 返回自身以继续追踪后续代码
            return trace_func

        # 保存原追踪函数，设置自定义追踪
        original_trace = sys.gettrace()
        sys.settrace(trace_func)
        
        try:
            do_something()
        finally:
            # 恢复原追踪函数，避免影响其他测试
            sys.settrace(original_trace)

        # 如果检测到异常，触发测试失败
        if exception_occurred:
            self.fail("函数内部发生了未被抛出的异常")

原理说明 #

• sys.settrace设置的trace_func会在每一个代码执行事件被调用，其中event == 'exception'表示当前上下文抛出了异常，无论这个异常是否会被后续的except捕获。
• 我们通过nonlocal关键字在追踪函数中修改外部的exception_occurred标记，记录异常是否发生。
• 测试结束后必须恢复原追踪函数，避免干扰其他测试用例的执行。

可选优化：封装为上下文管理器 #

如果需要在多个测试用例中复用这个逻辑，可以封装成上下文管理器，让代码更简洁：

from contextlib import contextmanager
import sys

@contextmanager
def detect_internal_exceptions():
    exception_occurred = False
    original_trace = sys.gettrace()

    def trace_func(frame, event, arg):
        nonlocal exception_occurred
        if event == 'exception':
            exception_occurred = True
        return trace_func

    sys.settrace(trace_func)
    try:
        yield lambda: exception_occurred
    finally:
        sys.settrace(original_trace)

# 测试用例中使用
class TestDoSomething(unittest.TestCase):
    def test_do_something(self):
        with detect_internal_exceptions() as has_exception:
            do_something()
        if has_exception():
            self.fail("函数内部检测到异常")

内容的提问来源于stack exchange，提问作者talking_corndog