Python单元测试字典断言KeyError

Question

我不确定如何创建 Python unittest 来检查字典是否返回 KeyError。 我认为单元测试会调用字典键，所以它看起来像这样：

def test_dict_keyerror_should_appear(self):
    my_dict = {'hey': 'world'}
    self.assertRaises(KeyError, my_dict['some_key'])

但是，我的测试只会出现KeyError错误，而不是断言发生了 KeyError。

Answer 1

为了解决这个问题，我使用了一个lambda来调用字典键来引发错误。

def test_dict_keyerror_should_appear(self):
    my_dict = {'hey': 'world'}
    self.assertRaises(KeyError, lambda: my_dict['some_key'])

Answer 2

另一种选择是使用operator.getitem ：

from operator import getitem

self.assertRaises(KeyError, getitem, my_dict, 'some_key')