python - 为主要功能测试设置命令行参数

Question

我在 python 中有一个 main() 函数，它可以获取命令行参数。
有没有办法为这个函数编写 pytest 测试并在代码中定义参数？

例如

def main():
    # argparse code
    args, other = arg_parser.parse_known_args()
    return args.first_arg


def test_main():
    res = main() # call with first_arg = "abc"
    assert(res == "abc")

Answer 1

添加到以前的答案，而不是修改 sys.argv使用可以掩盖和保护底层对象的上下文管理器更安全。一个例子是

with unittest.mock.patch('sys.argv', ['program_name', '--option1', 'inputFile']):
    main()

这仅适用于 python3。对于python2 Mock图书馆可以解决问题。
我在不同的 stackoverflow 帖子 here 上找到了这个解决方案.