python - 如何使用测试文件和预期的输出文件测试 Python

Question

我有一个 Python 脚本。
一种特定的方法需要一个文本文件，并且可能会创建 0-3 个文件。
我有示例文本文件，以及每个文本文件的预期输出文件。

我将如何设置此脚本进行测试？
我使用 unittest 来测试当前没有文件 i/o 的函数。

• SampleInputFile1 -> 预计生成 0 个文件。
• SampleInputFile2 -> 期望生成 1 个具有特定输出的文件。
• SampleInputFile3 -> 期望生成 3 个文件，每个文件都有特定的输出。

我想确保所有三个示例文件，预期的文件都是用预期的内容生成的。测试脚本问题

标签： pythontestingio

Answer 1

也许我不明白这个问题，但是您使用程序将文本分成不同的潜台词？

为什么不使用 sklearn 中的 train_test_split 来获取测试和训练文件，

sklearn.model_selection.train_test_split(*arrays, test_size=None, train_size=None, random_state=None, shuffle=True, stratify=None)