最近，我必须使用 pytest 为 python 模块编写单元测试。该模块包含一个类，其他类在其构造函数中初始化。
像往常一样，我为此类创建了一个固定装置，以便轻松为每个类方法编写测试。此时，当我尝试模拟构造函数中启动的不同类时，我遇到了一些问题。模拟不起作用，这些类的实例仍在创建中。
经过一些研究并结合我在网上找到的一些不同的解决方案后，我想分享我如何成功模拟课程。
解决方案
这是我尝试模拟的类的示例：
class classa: def __init__(self): self.class_b = classb() self.class_c = classc() self.count = 0
我们希望在测试期间为此类的每个字段设置一个值。该值可以是 none 或类模拟，但我们不希望启动 classb 和 classc 类。
立即学习“Python免费学习笔记（深入）”；
在我们的例子中，让我们决定 self.class_b 和 self.class_c 应该是模拟：
@pytest.fixturedef mock_class_b(): class_b = mock(spec=classb) return class_b@pytest.fixturedef mock_class_c(): class_c = mock(spec=classc) return class_c
因此，这个类别中满足我们目标的固定装置如下所示：
@pytest.fixturedef class_a_mock(mock_class_b, mock_class_c): with patch.object(target=classa, attribute="__init__", return_value=none) as mock_init: class_a = classa() class_a.class_b = mock_class_b class_a.class_c = mock_class_c class_b.count = 0 return class_a
重要的部分是如何使用 patch.object 函数，该函数来自 python 中的 unittest.mock 模块。它在测试中用于临时用模拟或其他值替换给定对象的属性。
争论
target=classa：我们要修补其属性的对象（通常是一个类）。attribute=”__init__”: 我们要修补的属性的名称。return_value=none：用一个不执行任何操作的函数替换 __init__ 方法
通过这种方式，我们可以在我们的装置中创建模拟变量。
阅读有关 patch.object 的更多信息
测试技巧
我写这个教程是为了这种情况，无论出于何种原因，我们都无法更改a类的代码。但是，我通常建议如果可能的话修改代码，不是为了改变逻辑，而是为了使其更易于测试。
以下是一些如何修改 a 类以使其更具可测试性的示例：
选项 1： 将 b 类和 c 类的实例作为参数传递。
这样，当我们编写固定装置时，我们可以传递模拟而不是实例。
class classa: def __init__(self, class_b_instance, class_c_instance): self.class_b = class_b_instance self.class_c = class_c_instance self.count = 0
选项2： 创建一个指示测试模式的布尔变量。
这样我们就可以决定a类的哪些字段在启动时会或不会获取值。
class classa: def __init__(self, test_mode=false): if not test_mode: self.class_b = classb() self.class_c = classc() self.count = 0
选项 3： 在单独的方法中进行类启动。
这种方法使我们可以选择避免在测试模块中调用 set_class_variables。
class ClassA: def __init__(self): self.class_b = None self.class_c = None self.count = None def set_class_variables(self): self.class_b = ClassB() self.class_c = ClassC() self.count = 0
希望这有帮助！ :)
以上就是模拟 Python 类的详细内容，更多请关注创想鸟其它相关文章！
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容， 请发送邮件至 chuangxiangniao@163.com 举报，一经查实，本站将立刻删除。
发布者：程序猿，转转请注明出处：https://www.chuangxiangniao.com/p/1349534.html
微信扫一扫 
支付宝扫一扫 
