如何将命令行参数从 pytest 传递到代码

我正在尝试将参数从 pytest 测试用例传递到正在测试的模块。例如，使用 main.pyPython 样板 https://www.python-boilerplate.com/py3+argparse+pytest，我可以从命令行运行它：

$ python3 main.py
usage: main.py [-h] [-f] [-n NAME] [-v] [--version] arg
main.py: error: the following arguments are required: arg
$ python3 main.py xx
hello world
Namespace(arg='xx', flag=False, name=None, verbose=0)

现在我尝试用 pytest 做同样的事情，使用以下 test_sample.py

(NOTE:main.py 需要命令行参数。但这些参数需要在特定测试中进行硬编码，它们不应该是 pytest 的命令行参数。 pytest 测试用例只需将这些值作为命令行参数发送给 main.main()。）

import main
def test_case01():
    main.main()
    # I dont know how to pass 'xx' to main.py,
    # so for now I just have one test with no arguments

并运行测试：

pytest -vs test_sample.py

此操作失败并显示错误消息。我尝试查看其他答案以寻求解决方案，但无法使用它们。例如，42778124 https://stackoverflow.com/questions/42778124建议创建一个单独的文件 run.py 这不是一个可取的事情。和48359957 https://stackoverflow.com/questions/48359957 and 40880259 https://stackoverflow.com/questions/40880259似乎更多地处理 pytest 的命令行参数，而不是将命令行参数传递给主代码。

我不需要 pytest 来获取命令行参数，这些参数可以在特定测试中进行硬编码。但这些参数需要作为参数传递给主代码。你能给我一个 test_sample.py，它用一些参数调用 main.main() 吗？

如果您无法修改签名main方法，您可以使用猴子补丁 https://docs.pytest.org/en/latest/monkeypatch.html用测试数据暂时替换参数的技术。示例：想象一下为以下程序编写测试：

import argparse


def main():
    parser = argparse.ArgumentParser(description='Greeter')
    parser.add_argument('name')
    args = parser.parse_args()
    return f'hello {args.name}'


if __name__ == '__main__':
    print(main())

从命令行运行它时：

$ python greeter.py world
hello world

测试main带有一些自定义数据的函数，monkeypatchsys.argv:

import sys
import greeter

def test_greeter(monkeypatch):
    with monkeypatch.context() as m:
        m.setattr(sys, 'argv', ['greeter', 'spam'])
        assert greeter.main() == 'hello spam'

当与参数化 https://docs.pytest.org/en/latest/parametrize.html技术，这允许轻松测试不同的参数而无需修改测试函数：

import sys
import pytest
import greeter

@pytest.mark.parametrize('name', ['spam', 'eggs', 'bacon'])
def test_greeter(monkeypatch, name):
    with monkeypatch.context() as m:
        m.setattr(sys, 'argv', ['greeter', name])
        assert greeter.main() == 'hello ' + name

现在您将获得三个测试，每个测试对应一个参数：

$ pytest -v test_greeter.py

...

test_greeter.py::test_greeter[spam] PASSED
test_greeter.py::test_greeter[eggs] PASSED
test_greeter.py::test_greeter[bacon] PASSED