python - 计算 Python 2.4 中的警告

Question

我有一些测试需要计算函数引发的警告数量。在 Python 2.6 中这很简单，使用

with warnings.catch_warnings(record=True) as warn:
    ...
    self.assertEquals(len(warn), 2)

不幸的是，with 在 Python 2.4 中不可用，那么我还能使用什么？我不能简单地检查是否有一个警告(使用带有 action='error' 和 try/catch 的警告过滤器)，因为警告的数量很大。

Answer 1

我打算建议与 Ignacio 相同的解决方法，一个更完整的测试代码示例:

import warnings

def setup_warning_catcher():
    """ Wrap warnings.showwarning with code that records warnings. """


    caught_warnings = []
    original_showwarning = warnings.showwarning

    def custom_showwarning(*args,  **kwargs):
        caught_warnings.append(args[0])
        return original_showwarning(*args, **kwargs)

    warnings.showwarning = custom_showwarning
    return caught_warnings


caught_warnings_list = setup_warning_catcher()

# trigger warning here

assert len(caught_warnings_list) == 1