如何使用 python 库 unittest 断言异常

Question

我有一个虚拟函数来尝试异常：

def fun(n):
    try:
        if n <1:
            raise ValueError
        return 1
        except:
            pass

在我的单元测试中我使用：

import  unittest
class TestFibonnacci(unittest.TestCase):
    def test_values(self):
        self.assertEqual(fun(1),1)
        self.assertRaises(ValueError, fun(-1))

但是我无法得到我实际得到的答案：

发生异常 乙…… 错误：test_values (ma​​in.TestFibonnacci)

Traceback（最近一次调用最后一次）： 这里的追溯

TypeError: 'NoneType' 对象不可调用

在 0.001 秒内运行 1 次测试

我做错了什么？

Answer 1

您正在立即调用fun(-1)，而不是让self.assertRaises 调用它，以便它可以捕获异常。

您必须分别传递函数及其参数。

self.assertRaises(ValueError, fun, -1)

或者，您可以使用assertRaises 作为上下文管理器。

with self.assertRaises(ValueError):
    fun(-1)

with 语句捕获fun 引发的异常并将其提供给assertRaises 返回值的__exit__ 方法。