2

我有一些单元测试，并希望将每个测试运行的结果存储为 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)

文档说我可以为每个类制作代表，所以我尝试了一下并编写了以下代码：

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 与特定基类的所有实例一起工作，而只有一个代表注册。

4

1 回答 1

6

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

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

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

更新。多代表允许实现我想要实现的目标：为基类注册代表。

于 2013-11-02T23:40:26.510 回答