python-3.x - python中的单元测试 - 在测试中创建数据帧

Question

我正在为方法编写单元测试 - 经常操作数据帧。

我的数据来自 API 调用，我陷入了在测试中使用 API 调用的陷阱 - 我觉得这不能准确地测试特定组件，因为 API 调用可能存在问题。

在每个测试中创建一个虚拟数据帧，并分别测试 API 调用是否返回预期格式的数据帧是更好的做法吗？

手动制作 DataFrame 很痛苦，是否有实用程序可以将事件控制台中的 DataFrame 对象转换为构建它所需的代码字符串？

Answer 1

您可以随时保存 DataFrame到 CSV 文件(和其他格式，如 pickle):

df.to_csv('my_data.csv')

当然重新加载它:

pd.DataFrame.from_csv('my_data.csv')

关于您的“测试数据”:

问题始终是您要测试什么功能。对我来说，这听起来好像您只是想测试特定结果等某些例程的实现，而不是您的 API。由于您不打算测试您的 API，因此只需将其废弃，并在不必要时不要使用它。

如果有的话，我会编写一个脚本/程序，从我的 API 获取数据，将其存储为我的“测试数据”并将其用于我的单元测试。除非您可以即时生成测试数据(在相当长的时间内)，否则您也可以这样做。