Exemplos de main em Python

Exemplo n.º 1

Arquivo: test_main.py Projeto: smartmob-project/smartmob

def test_main_sys_argv(capsys):
    with pytest.raises(SystemExit) as error:
        main()
    assert error.value.args[0] == 0
    stdout, _ = capsys.readouterr()
    assert stdout.strip() == version

Exemplo n.º 2

Arquivo: test_main.py Projeto: smartmob-project/smartmob

def test_main_explicit_args(capsys):
    with pytest.raises(SystemExit) as error:
        main(['--version'])
    assert error.value.args[0] == 0
    stdout, _ = capsys.readouterr()
    assert stdout.strip() == version