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

前端之家收集整理的这篇文章主要介绍了如何在Python中单个测试的持续时间内替换类变量?前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。
我正在尝试测试一个缓存类,其中应该自动删除旧项.缓存可以容纳的项目数量由“常量”类变量控制,如下所示:
  1. class Cache(dict):
  2. MAX_ITEMS = 100

出于测试目的,我想在单个测试用例的持续时间内更改该常量.使用unittest.mock.patch我可以用MagicMock替换类变量,但这不是我想要做的.我只需要用不同的号码替换它.

有没有一种优雅的方法在Python中执行此操作(如补丁),或者我应该在tearDown上自己重置变量?

我正在使用Python 3.4.3

解决方法

补丁可以做到这一点.尝试:
  1. from unittest.mock import patch
  2.  
  3. class A:
  4. val = 1
  5.  
  6. assert A.val == 1
  7. with patch.object(A,"val","patched_value"):
  8. assert A.val == "patched_value"
  9.  
  10. assert A.val == 1

当然你也可以使用patch.object作为装饰器.

猜你在找的Python相关文章