测试python函数

Question

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

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

Answer 1

Hi取决于您正在搜索的测试。

方法I(功能性(

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

方法II(非回归(

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