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


我有一段代码是这么处理的,先检查字典中是否有某个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')

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

python 单元测试

Kokami 11 years, 6 months ago

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

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

野生字幕菌 answered 11 years, 6 months ago

Your Answer