测试python函数答案

【问题标题】：Test python function测试python函数
【发布时间】：2021-12-31 20:34:07
【问题描述】：

早上好！我需要为 python 函数创建一个测试。要测试的结果是一个非常大的 json 格式的数据框。我可以在断言中添加什么来测试 json 文件是否已正确创建？谢谢

def function (param):
   ...
   ...
assert ()

【问题讨论】：

大型数据集的缩短版本？测试的想法是确保功能正常工作并且能够在遇到错误时处理它们。因此，请尝试向函数提供真实数据的几个微小变体。
您可以编写一个验证结构的函数（称为 validator）并适当地返回 True 或 False。然后： assert validator(df) 其中 df 是对要验证的结构的引用

【解决方案1】：

您好，取决于您要搜索的测试。

返回的json是有结构的，所以可以验证列数是否正确，行数是否正确等等：

jdf = DataFrame({'A':[1,2], 'B':[3,4]}).to_json()
assert len(json.loads(jdf)) ==2
assert set(json.loads(jdf).keys()) == {'A','B'}

如果您之前在其他地方存储了经过验证的 df，您可以使用 assert_frame_equal 检查所有值是否符合预期

【讨论】：