Exemplos de main em Python

Exemplo n.º 1

def test_main(mock_main):
    import sys
    old_argv = sys.argv
    sys.argv = ["foo", "bar"]
    bm.main()
    assert mock_main.call_count == 1
    assert mock_main.call_args[0] == (["foo", "bar"], )
    assert mock_main.call_args[1] == {}
    sys.argv = old_argv

Exemplo n.º 2

Arquivo: test___main__.py Projeto: digitalsatori/Bokeh

def test_main(mock_main):
    import sys
    old_argv = sys.argv
    sys.argv = ["foo", "bar"]
    bm.main()
    assert mock_main.call_count == 1
    assert mock_main.call_args[0] == (["foo", "bar"],)
    assert mock_main.call_args[1] == {}
    sys.argv = old_argv