如何在使用 argparse 的文件上使用指定的测试目录运行 pytest？

• 我正在使用 Python 库编写单元测试pytest
• 我需要指定一个目录用于测试文件以避免自动发现测试文件，因为有一个很大的子目录结构，包括库中的许多文件名称中包含“_test”或“test_”但不适合pytest
• 库中的一些文件使用argparse用于指定命令行选项
• 问题在于，将 pytest 的目录指定为命令行参数似乎会干扰使用 argparse 的命令行选项

举个例子，我的根目录中有一个文件，名为script_with_args.py如下：

import argparse

def parse_args():
    parser = argparse.ArgumentParser(description="description")

    parser.add_argument("--a", type=int, default=3)
    parser.add_argument("--b", type=int, default=5)

    return parser.parse_args()

我还有一个名为tests在根目录中，包含一个名为的测试文件test_file.py:

import script_with_args

def test_script_func():
    args = script_with_args.parse_args()
    assert args.a == 3

如果我打电话python -m pytest从命令行来看，测试顺利通过。如果我从命令行指定测试目录python -m pytest tests，返回以下错误：

============================= test session starts =============================
platform win32 -- Python 3.6.5, pytest-3.5.1, py-1.5.3, pluggy-0.6.0
rootdir: C:\Users\Jake\CBAS\pytest-tests, inifile:
plugins: remotedata-0.2.1, openfiles-0.3.0, doctestplus-0.1.3, arraydiff-0.2
collected 1 item

tests\test_file.py F                                                     [100%]

================================== FAILURES ===================================
______________________________ test_script_func _______________________________

    def test_script_func():
        # a = 1
        # b = 2
>       args = script_with_args.parse_args()

tests\test_file.py:13:
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _
script_with_args.py:9: in parse_args
    return parser.parse_args()
..\..\Anaconda3\lib\argparse.py:1733: in parse_args
    self.error(msg % ' '.join(argv))
..\..\Anaconda3\lib\argparse.py:2389: in error
    self.exit(2, _('%(prog)s: error: %(message)s\n') % args)
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _

self = ArgumentParser(prog='pytest.py', usage=None, description='description', f
ormatter_class=<class 'argparse.HelpFormatter'>, conflict_handler='error', add_h
elp=True)
status = 2, message = 'pytest.py: error: unrecognized arguments: tests\n'

    def exit(self, status=0, message=None):
        if message:
            self._print_message(message, _sys.stderr)
>       _sys.exit(status)
E       SystemExit: 2

..\..\Anaconda3\lib\argparse.py:2376: SystemExit
---------------------------- Captured stderr call -----------------------------
usage: pytest.py [-h] [--a A] [--b B]
pytest.py: error: unrecognized arguments: tests
========================== 1 failed in 0.19 seconds ===========================

我的问题是，如何指定 pytest 的测试文件目录，而不干扰 argparse 的命令行选项？

parse_args()不带参数读取sys.argv[1:]列表。这将包括“测试”字符串。

pytests也用那个sys.argv[1:]有自己的解析器。

使你的解析器可测试的一种方法是提供一个可选的argv:

def parse_args(argv=None):
    parser = argparse.ArgumentParser(description="description")

    parser.add_argument("--a", type=int, default=3)
    parser.add_argument("--b", type=int, default=5)

    return parser.parse_args(argv)

然后你可以用以下方法测试它：

parse_args(['-a', '4'])

并真正使用它

parse_args()

改变sys.argv也是个好办法。但是，如果您打算将解析器放入这样的函数中，那么您不妨为其提供额外的灵活性。