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

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')

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


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

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

【热门文章】
【热门文章】