我正在为一些检查 os.environ
中的值的代码编写测试(我知道这不是最佳的,但我必须接受它)。我想在测试期间从 os.environ 中删除一个条目。我不确定 mock 是否支持这个。我知道 patch.dict
可用于修改项目,但我希望删除键/值对。我想要这样的东西:
print os.environ
{ ... , 'MY_THING': 'foo', ... }
with mock.patch.dict.delete('os.environ', 'MY_THING'):
# run the test
# ( 'MY_THING' in os.environ ) should return False
# everything back to normal now
print os.environ
{ ... , 'MY_THING': 'foo', ... }
有没有办法实现这样的壮举?
最佳答案
mock.patch.dict
不像您的示例所需代码那样工作。 patch.dict
是一个需要参数的函数。您可能想像这样使用它:
>>> import os
>>> import mock
>>> with mock.patch.dict('os.environ'):
... del os.environ['PATH']
... print 'PATH' in os.environ
...
False
>>> print 'PATH' in os.environ
True
关于python - 使用 python 的 mock 临时从 dict 中删除一个对象,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/29243269/