Function assert_output

tests/utils.py:362–399

(output, expected_entries, prefix=None, normalize=False)

def assert_output(output, expected_entries, prefix=None, normalize=False):
    lines = tuple(filter(None, output.split('\n')))
    if expected_entries and not lines:
        raise OutputFailure("Output is empty")

    if prefix is not None:
        for line in lines:
            if not line.startswith(prefix):
                raise OutputFailure(line)

    if normalize:
        verify_normalize(lines, prefix)

    # Filter only entries compatible with the current Python
    filtered_expected_entries = []
    for expected_entry in expected_entries:
        if isinstance(expected_entry, _BaseEntry):
            if expected_entry.is_compatible_with_current_python_version():
                filtered_expected_entries.append(expected_entry)
        else:
            filtered_expected_entries.append(expected_entry)

    expected_entries_count = len(filtered_expected_entries)
    any_mismatch = False
    result = ''
    template = u'\n{line!s:%s}   {expected_entry}  {arrow}' % max(map(len, lines))
    for expected_entry, line in zip_longest(filtered_expected_entries, lines, fillvalue=""):
        mismatch = not (expected_entry and expected_entry.check(line))
        any_mismatch |= mismatch
        arrow = '<===' * mismatch
        result += template.format(**locals())

    if len(lines) != expected_entries_count:
        result += '\nOutput has {} lines, while we expect {} lines.'.format(
                len(lines), len(expected_entries))

    if any_mismatch:
        raise OutputFailure(result)


def assert_sample_output(module):