我正在尝试测试一个缓存类,其中应该自动删除旧项.缓存可以容纳的项目数量由“常量”类变量控制,如下所示:
class Cache(dict): MAX_ITEMS = 100
出于测试目的,我想在单个测试用例的持续时间内更改该常量.使用unittest.mock.patch我可以用MagicMock替换类变量,但这不是我想要做的.我只需要用不同的号码替换它.
有没有一种优雅的方法在Python中执行此操作(如补丁),或者我应该在tearDown上自己重置变量?
我正在使用Python 3.4.3
解决方法
补丁可以做到这一点.尝试:
from unittest.mock import patch class A: val = 1 assert A.val == 1 with patch.object(A,"val","patched_value"): assert A.val == "patched_value" assert A.val == 1
当然你也可以使用patch.object作为装饰器.