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
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