时间戳

首页 > 解决方案 > 在 Python 中模拟数据类时如何使用规范

python - 在 Python 中模拟数据类时如何使用规范

问题描述

我正在尝试使用 backport 包将我们的 namedtuple 类移植到 Python 3.6 中的数据类中。但是，我注意到在模拟数据类类时，您不能再使用“spec”关键字了。我认为这是因为数据类代码是自动生成的。

from dataclasses import dataclass
import mock

@dataclass
class A:
    aaa: str
    bbb: int


m = mock.Mock(spec=A)

m.aaa

这是我得到的错误：

AttributeError: Mock object has no attribute 'aaa'

知道是否有任何方法可以自动将原始对象的所有属性设置为模拟对象？我有很多包含大量数据的数据类。如果我尝试手动设置值，那将非常乏味。

标签： pythonattributesmockingpython-dataclasses

解决方案

我最终使用这个通用辅助函数来实现规范对常规类的作用：

import mock
from dataclasses import fields


def create_dataclass_mock(obj):
    return mock.Mock(spec=[field.name for field in fields(obj)])

推荐阅读