Python Unit Test Dictionary Assert 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')