用命令行参数对Python进行单元测试？

Question

我正在对一个我拥有的应用程序进行单元测试，但没有使用该应用程序中的任何函数。它只使用命令行参数来知道要做什么。导入unittest的一般设置是什么？我对Python还比较陌生，正在尝试掌握它。你知道吗

例如，我的代码可以有如下命令应用程序类型配置alpha“，”应用程序类型配置测试版“，”应用程序类型信息alpha”-但我不完全确定如何在单元测试中引用它。你知道吗

Answer 1

如果您的程序没有函数或类，那么它就没有单元，因此单元测试不适用于它。如果引入如下分离，测试和维护应用程序将变得更加容易：

def configure(version):
    # do configure things

def info(version):
    # display info

def main(args):
    cmd, version = args[1:]  # add size checking
    if cmd == 'configure':
        configure(version)  
    elif cmd == 'info':
        info(version)

if __name__ == '__main__':
    main(sys.argv)

现在，您可以分别为configure()和info()编写单元测试。你知道吗