python - 检查功能是否会通过 Nose 测试发出警告

Question

我正在使用 nose 编写单元测试，并且我想检查函数是否引发警告(该函数使用 warnings.warn)。这是轻而易举就能做到的事情吗？

Answer 1

def your_code():
    # ...
    warnings.warn("deprecated", DeprecationWarning)
    # ...

def your_test():
    with warnings.catch_warnings(record=True) as w:
        your_code()
        assert len(w) > 1

当然，除了检查长度，您还可以深入检查:

断言 str(w.args[0]) == "已弃用"

在 python 2.7 或更高版本中，您可以使用最后一个检查来执行此操作:

断言 str(w[0].message[0]) == "已弃用"