pyhamcrest修改"得到"描述

时间:2014-02-17 16:27:22

标签: python hamcrest

我编写了一个自定义的hamcrest匹配器,用于检查列表中的文件是否已被复制。列表可能很长(1000个文件+),所以如果缺少一个文件,我不希望匹配器打印出整个列表。

我可以为丢失的文件制作自定义说明,但有没有办法修改Got: <list of files>部分?

完整代码:

class FilesHaveBeenCopied(BaseMatcher):
    def __init__(self):
        self.missing = None

    def _matches(self, source_files):
        try:            
            self.missing = next(f for f in source_files if not os.path.exists(target_of(f)))
        except StopIteration:
            return True
        return False

    def describe_to(self, description):            
        description.append_text("file to be copied '{0}'".format(self.missing)) 

def have_been_copied():
    return FilesHaveBeenCopied()

用法:

assert_that(self.source_files, have_been_copied())

1 个答案:

答案 0 :(得分:1)

覆盖describe_mismatch以覆盖完整的实际值:

def describe_mismatch(self, actual, description):
    description.append(self.missing)

您的describe_to应描述预期值 - 而不是缺失值。或者它应该只报告文件的数量,如“21个现有文件的列表”。