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

发布于 2021-01-29 16:25:49

我在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")
关注者
0
被浏览
48
1 个回答
  • 面试哥
    面试哥 2021-01-29
    为面试而生,有面试问题,就找面试哥。

    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



知识点
面圈网VIP题库

面圈网VIP题库全新上线,海量真题题库资源。 90大类考试,超10万份考试真题开放下载啦

去下载看看