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

发布时间：2020-12-16 23:45:49 所属栏目：Python 来源：网络整理

导读：我正在尝试测试一个缓存类,其中应该自动删除旧项.缓存可以容纳的项目数量由“常量”类变量控制,如下所示： class Cache(dict): MAX_ITEMS = 100 出于测试目的,我想在单个测试用例的持续时间内更改该常量.使用unittest.mock.patch我可以用MagicMock替换类变量

我正在尝试测试一个缓存类,其中应该自动删除旧项.缓存可以容纳的项目数量由“常量”类变量控制,如下所示：

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作为装饰器.

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!