测试 python 函数

Test python function

早上好!我需要为 python 函数创建一个测试。要测试的结果是 json 格式的非常大的数据帧。我可以在 assert 中输入什么来测试 json 文件是否已正确创建?谢谢

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

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

方法一(功能性)

返回的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

检查所有值是否符合预期