main_test.py 文件源码

def test_obama_example(resource, capsys):
    main.main(resource('obama_wikipedia.txt'))
    stdout, _ = capsys.readouterr()
    lines = stdout.split('\n')
    assert re.match(
        r'.*Obama\b.*\| received\b.*\| national attention\b',
        lines[1])