python,unittest,用命令行参数测试脚本

Question

我编写了一个python脚本,它接受命令行参数并播放一轮Tic Tac Toe.
运行它看起来像这样......

运行ttt o#xo ## x ## x 0 1

如果移动是合法的,那么它将打印新的电路板布局以及是否有人赢得了比赛

我必须使用unittest为它编写测试.我不知道如何使用各种命令行参数测试整个脚本,我看到的所有示例似乎只是测试脚本中的各个函数.脚本也用于argparse解析参数

谢谢!

Answer 1

重构你的程序,使其主要动作(减去argparsing)发生在"main"函数中:

def main(args):
    ...

if __name__ == '__main__':
    args = parse_args()
    main(args)

然后,你可以写的行为测试main和parse_args.

PS.可以使用该subprocess模块将您的程序作为外部进程调用,然后解析输出,但我认为这将是更加丑陋和不必要的.

PPS.作为以这种方式编写程序的额外好处,您将能够将程序作为模块导入,并在其他脚本中调用其主函数.例如,如果您有一天想为它构建GUI,那么这可能很有用.