test_formatting.py

import stackprinter


def test_frame_formatting():
    """ pin plaintext output """
    msg = stackprinter.format()
    lines = msg.split('\n')

    expected = ['File "test_formatting.py", line 6, in test_frame_formatting',
                '    4    def test_frame_formatting():',
                '    5        """ pin plaintext output """',
                '--> 6        msg = stackprinter.format()',
                "    7        lines = msg.split('\\n')",
                '    ..................................................',
                "     stackprinter.format = <function 'format' __init__.py:17>",
                '    ..................................................',
                '',
                '']

    for k, (our_line, expected_line) in enumerate(zip(lines[-len(expected):], expected)):
        if k == 0:
            assert our_line[-52:] == expected_line[-52:]
        elif k == 6:
            assert our_line[:58] == expected_line[:58]
        else:
            assert our_line == expected_line

    # for scheme in stackprinter.colorschemes.__all__:
    #     stackprinter.format(style=scheme, suppressed_paths=[r"lib/python.*"])


def test_exception_formatting():
    from source import Hovercraft

    try:
        Hovercraft().eels
    except:
        msg_plain = stackprinter.format()
        msg_color = stackprinter.format(style='darkbg')

    lines = msg_plain.split('\n')

    assert lines[0].endswith('eels')
    assert lines[-1] == 'Exception: ahoi!'

    print(msg_plain)
    print(msg_color)


def test_none_tuple_formatting():
    output = stackprinter.format((None, None, None))
    assert output == "NoneType: None"


def test_none_value_formatting():
    output = stackprinter.format((TypeError, None, None))
    assert output == "TypeError: None"