通过pytest在python中传递命令行参数

Question

我可以在运行时传递命令行参数

python <filename>.py arg1

但是，当我尝试传递命令行参数以运行 pytest 时，它会失败并给出如下错误。能否请教。

pytest <filename>.py arg1
ERROR: file not found: arg1

编辑：

例如，假设我已经传递了一个参数并正在通过 sys.argv 读取它，我正在考虑以这种方式使用它：

import sys
arg = sys.argv[3]
def f():
    return 3

def test_function():
    assert f() == arg

Answer 1

您的pytest <filename>.py arg1 命令试图在<filename>.py 和arg1 两个模块上调用pytest，但是没有模块arg1。

如果您想在运行 pytest 之前传递一些参数，请在提取变量后从 python 脚本运行 pytest。

正如其他人建议的那样，尽管您可能希望以其他方式参数化您的测试，请尝试：Parameterized pytest。

# run.py
import pytest
import sys

def main():
    # extract your arg here
    print('Extracted arg is ==> %s' % sys.argv[2])
    pytest.main([sys.argv[1]])

if __name__ == '__main__':
    main()

使用python run.py filename.py arg1调用它

Answer 2

这是我刚刚通过阅读parameterized pytest docs 并进行了一段时间的黑客攻击...我不知道它的整体稳定性或良好程度，因为我刚刚开始工作。 然而，我确实检查了 HTML 覆盖率生成是否适用于这种方法。

• 为configuring the command-line args you want to pass添加一个文件到你的测试目录：
  • tests\conftest.py

# this is just so we can pass --server and --port from the pytest command-line
def pytest_addoption(parser):
    ''' attaches optional cmd-line args to the pytest machinery '''
    parser.addoption("--server", action="append", default=[], help="real server hostname/IP")
    parser.addoption("--port", action="append", default=[], help="real server port number")

• 然后添加一个测试文件，用这个特殊的pytest_generate_tests函数which is called when collecting a test function
  • tests\test_junk.py

def pytest_generate_tests(metafunc):
    ''' just to attach the cmd-line args to a test-class that needs them '''
    server_from_cmd_line = metafunc.config.getoption("server")
    port_from_cmd_line = metafunc.config.getoption("port")
    print('command line passed for --server ({})'.format(server_from_cmd_line))
    print('command line passed for --port ({})'.format(port_from_cmd_line))
    # check if this function is in a test-class that needs the cmd-line args
    if server_from_cmd_line and port_from_cmd_line and hasattr(metafunc.cls, 'real_server'):
        # now set the cmd-line args to the test class
        metafunc.cls.real_server = server_from_cmd_line[0]
        metafunc.cls.real_port = int(port_from_cmd_line[0])


class TestServerCode(object):
    ''' test-class that might benefit from optional cmd-line args '''
    real_server=None
    real_port = None

    def test_valid_string(self):
        assert self.real_server!=None
        assert self.real_port!=None

    def test_other(self):
        from mypackage import my_server_code
        if self.real_server !=  None:
            assert "couldn\'t find host" not in my_server_code.version(self.real_server, self.real_port)

• 然后运行（例如使用 HTML 覆盖）：

  • pytest tests\test_junk.py --server="abc" --port=123 --cov-report html --cov=mypackage