设置主要功能测试的命令行参数
我在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-
argvhttps://docs.python.org/3/library/argparse.html#partial-
parsing