python - 如何使用 argparse 删除 CLI 参数，以便 unittest 接受 arg 列表

Question

我想将我自己的参数传递到为 unittest 设置的文件中。所以像这样从命令行调用它应该可以工作:

python Test.py --c keith.ini SomeTests.test_one

目前我遇到了两个问题。

1) Arg 解析不允许未知参数

用法:Test.py [-h] [--c CONFILE]
Test.py:错误:无法识别的参数:SomeTests.test_one

2) 单元测试不允许未知参数。所以 --c fileName 不被 unittest 接受并返回:

AttributeError: 'module' 对象没有属性 'keith'

所以这个想法是在调用 unittest runner 之前收集我的参数并删除它们。

import unittest
import argparse

myArgs = None

def getArgs( allArgs ):
    parser = argparse.ArgumentParser( )
    parser.add_argument('--c', dest='conFile', type=str, default=None, help='Config file')
    args = parser.parse_args()

    if ( args.conFile == None ):
        parser.print_help()

    return args


class SomeTests(unittest.TestCase):
    def test_one(self):
        theTest( 'keith' )

    def test_two(self):
        otherTest( 'keith' )

if __name__ == '__main__':
    myArgs = getArgs( sys.argv )
    print 'Config File: ' + myArgs.conFile
    unittest.main( argv=sys.argv, testRunner = unittest.TextTestRunner(verbosity=2))

有趣的是我刚刚发现了 parse_known_args() 所以我将解析行更改为:

args = parser.parse_known_args(['--c']). 

我认为这会解决我的问题并给我一些东西传递给单元测试。不幸的是我得到:

Test.py: error: argument --c: expected one argument. 

这不应该工作吗？

Answer 1

OK花了一些努力，但想通了。这是完全可能的。 argparse 的文档不正确。函数 parse_known_args() 不应包含已知参数列表。 argparse 还删除了 arg[0]，这对于返回很重要，因此其他命令可以看到有效的参数列表。我认为这次删除是一个错误。我已经包含了最终的示例代码。

import unittest
import argparse
import sys

myArgs = None

def getArgs( allArgs ):
    parser = argparse.ArgumentParser( )
    parser.add_argument('--c', dest='conFile', type=str, default=None, help='Configuration file. (Required)')
    args, addArgs = parser.parse_known_args( )

    if ( args.conFile == None ):
        parser.print_help()
        sys.exit(2)

    # argparse strips argv[0] so prepend it
    return args, [ sys.argv[0]] + addArgs

def verify( expected, actual ):
    assert expected == actual, 'Test Failed: '

# Reusable Test
def theTest( exp ):
    print 'myargs: ' + str( myArgs )
    verify( exp, 'keith' )

def otherTest( exp ):
    theTest( exp )

class SomeTests(unittest.TestCase):
    def test_one(self):
        theTest( 'keith' )

    def test_two(self):
        otherTest( 'keith2' )

if __name__ == '__main__':
    myArgs, addArgs = getArgs( sys.argv )
    unittest.main( argv=addArgs, testRunner = unittest.TextTestRunner(verbosity=2))

将其保存到文件后，您可以像下面的示例一样调用它，它会正常工作。

python Test.py                     # Requires config file
python Test.py --c keith.ini       # Runs all tests
python Test.py --c keith.ini SomeTests          # Runs Class
python Test.py --c keith.ini SomeTests.test_one # Runs test

HTH，享受