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, 11 months ago

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

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

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

Your Answer