热门文章
 当前位置：技术文章 > 编程面试题 >

设置主要功能测试的命令行参数

问题内容：

我在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")

问题答案：

parse_args接受一个argv参数。文档在示例中反复使用

parser = argparse.ArgumentParser()
parser.add_argument('--foo', action='store_true')
parser.add_argument('bar')
parser.parse_known_args(['--foo', '--badger', 'BAR', 'spam'])

字符串列表复制sys.argv[1:]到命令行的地方。如果参数是None（或省略），则解析器使用sys.argv[1:]。

因此，如果

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

你可以用

main(['foo', '-f','v'])

unittesting用于的文件argparse.py既使用此方法，也使用您的sys.argv直接修改方法。

https://docs.python.org/3/library/argparse.html#beyond-sys-
argv

https://docs.python.org/3/library/argparse.html#partial-
parsing