Question

我有一些单元测试，并希望将每个测试运行的结果存储为YAML文件以供进一步分析。 YAML格式的转储数据以多种方式满足我的需求。但是测试属于不同的套装，结果有不同的父类，这就是我所拥有的例子：

>>> rz # shorthand for result
[<upstream_suite.fftest.SimpleTest testMethod=test_is_fsType_not_set>,
<upstream_suite.openfolder.UfoOpenTest testMethod=test_is_A>,
<upstream_suite.openfolder.UfoOpenTest testMethod=test_is_A_a_glyph_instance>,
<upstream_suite.openfolder.UfoOpenTest testMethod=test_is_ended_ufo>,
<upstream_suite.openfolder.UfoOpenTest testMethod=test_is_folder>,
<upstream_suite.openfolder.UfoOpenTest testMethod=test_is_fsType_eq_1>,
<upstream_suite.openfolder.UfoOpenTest testMethod=test_it_exists>]
# etc.

每个测试都可以有自己的父类。

>>> type(rz[0]) == type(rz[1])
False

我已经使用__getstate__方法为所有类添加了自己的基类，但由于某种原因它无法正常工作：

>>> rz[0].__getstate__()
{'targets': ['upstream'], 'methodDoc': 'Is the OS/2 table fsType set to 0?', 'methodName': 'test_is_fsType_not_set', 'tool': 'FontForge', 'name': 'upstream_suite.fftest'}
>>> yaml.safe_dump(rz[0])
*** RepresenterError: cannot represent an object: test_is_fsType_not_set (upstream_suite.fftest.SimpleTest)

Documentations说我可以为每个班级制作代表，所以我试了一下并编写了这段代码：

import yaml
from somepath import MyTestCase

def repr_testcase(dumper, data):
    return dumper.represent_mapping(u'tag:yaml.org,2002:map', {
        'methodDoc': data._testMethodDoc,
        'tool': data.tool,
        'name': data.name,
        'methodName': data._testMethodName,
        'targets': data.targets,
        })

# Register representer 
yaml.SafeDumper.add_representer(MyTestCase, repr_testcase)

同一时间：

>>> isinstance(rz[0], MyTestCase)
True

即使在safe_dump无效之后也是如此。但如果我做了这个技巧，那么它可以工作，我可以得到结果：

>>> yaml.SafeDumper.add_representer(type(rz[0]), repr_testcase)
>>> print(yaml.safe_dump(rz[0]))
methodDoc: Is the OS/2 table fsType set to 0?
methodName: test_is_fsType_not_set
name: checker.upstream_suite.fontforge_suite.fftest
targets: [upstream]
tool: FontForge

但这意味着我需要注册所有可能的事件，这是......至少，看起来并不聪明。实际上我的真实result是更复杂的字典，从不同的地方收集的测试等等。

>>> for x in rz: yaml.SafeDumper.add_representer(type(x), repr_testcase)
>>> yaml.safe_dump(rz)
... here is expected result

所以，问题是，我对__getstate__做错了什么，或者我如何使用只有一个代表注册的特定基类的所有实例来创建工作pyyaml。

Answer 1

我找到了如何使这项工作的方式。简短的回答是，还有另一种方法来注册名为add_multi_representer的代表。所以修复是：

# Change this line:
# yaml.SafeDumper.add_representer(MyTestCase, repr_testcase)

# to this line:
yaml.SafeDumper.add_multi_representer(MyTestCase, repr_testcase)

<强> UPD 即可。多代表允许我做我想要实现的目标：为基类注册代表。

在pyyaml中表示具有相同基类的不同类的实例

1 个答案: