如何在Python中单个测试的持续时间内替换类变量?

前端之家收集整理的这篇文章主要介绍了如何在Python中单个测试的持续时间内替换类变量?前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。
我正在尝试测试一个缓存类,其中应该自动删除旧项.缓存可以容纳的项目数量由“常量”类变量控制,如下所示:
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作为装饰器.

猜你在找的Python相关文章