python - 如何正确测试程序 python 脚本

Question

我是 Python 的新手。但是，我正在编写一个脚本，从一个文件加载一些数据并生成另一个文件。我的脚本有几个功能，它还需要两个用户输入(路径)才能工作。

现在，我想知道是否有办法单独测试每个功能。由于没有类(class)，我不认为我可以通过单元测试来完成，对吗？

如果我不想一直运行整个脚本，测试脚本的常用方法是什么？其他人稍后必须维护脚本。因此，类似于单元测试的东西会很棒。

感谢您的投入!

Answer 1

如果您以对文件对象(流)进行操作的函数形式编写代码，或者如果数据足够小，接受并返回字符串，则可以轻松编写测试以提供适当的数据并检查结果.如果真实数据大到需要流，而测试数据不需要，则使用测试代码中的StringIO函数进行适配。

然后使用 __name__=="__main__" 技巧让您的单元测试驱动程序无需运行面向用户的脚本即可导入文件。