Дразнить ключ от os.environ

Question

Я пытаюсь протестировать функцию, которая должна возвращать OSError, если он не может найти определенный ключ в os.environ, но я не могу понять, как его смоделировать. Я в порядке с насмешкой над os.environ, также совершенно пустым.

with mock.patch.dict('os.environ', dict()): print(os.environ)

или

with mock.patch.dict(os.environ, dict()): print(os.environ)

Будет печатать всю мою среду независимо.

Answer 1

Я прибег к MonkeyPatching os.environ вместо этого:

old_os_environ = os.environ try: os.environ = {} self.assertRaises(OSError, my_function) finally: os.environ = old_os_environ