2013-11-02 89 views
2

我有一些unittests集合,并希望将每个测试运行的结果存储为YAML文件以供进一步分析。 YAML格式的转储数据在几个方面符合我的需求。但测试属于不同的花色和结果有不同的父类,这是一个例子我有什么:在pyyaml中表示具有相同基类的不同类的实例

>>> 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只有一个代表注册。

回答

5

我找到了如何使这项工作。简单的回答是,还有另一种方式来注册代理商,它叫做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。多代理人允许做我想要实现的功能:为基类注册代理人。

相关问题