python unittest 怎么处理异常抛出多个error

Question

我有一段代码是这么处理的，先检查字典中是否有某个key，如果没有，则抛出一个我定义的ERROR。

def fn(key):
    try:
        return dict[key]
    except KeyError:
        return NOKEYERROR('BLABLA')

代码倒是没有什么问题。但是测试的时候，我想要编写用例测试fn可以正确抛出NOKEYERROR，但是用assertRaises一直跑不通，结果是error。

def test_fn(self):
    self.assertRaises(NOKEYERROR, fn('123'))

unittest运行后，结果是：

> KeyError: 'bla'
During handling of the above exception, another exception occurred:
.... raise NOKEYERROR('BALBLA')

我的测试用例代码错在哪里呢？

Answer 1

self.assertRaises(NOKEYERROR, fn, '123')

你写成 fn('123') 的话，在传入 assertRaises 之前函数就已经被调用了。所以 assertRaises 的参数是 callable 以及它的参数。