使用pytest测试Bigquery返回的Dataframe时出现值错误该如何解决

错误原因 #

该报错不是pytest的bug，属于pandas的默认规则：直接对DataFrame、Index对象做布尔判定（你写的assert df本质是隐式执行bool(df)判断真假）时，pandas无法明确你要判断的是「对象是否非空」「是否所有值为真」还是其他逻辑，因此主动抛出该错误。
你在本地Notebook和普通.py文件中运行未报错，是因为本地环境的pandas版本较低，旧版本pandas兼容了bool(df)等价于len(df) > 0的逻辑，而pytest环境的pandas版本更高，已经移除了该兼容逻辑，所以触发报错。

解决方法 #

根据你实际需要校验的逻辑修改断言语句即可：

• 若要校验查询返回的DataFrame不为空表：将assert df改为assert not df.empty
• 若要校验DataFrame中至少存在一个真值：改为assert df.any(axis=None)
• 若要校验DataFrame中所有值都是真值：改为assert df.all(axis=None)

大部分场景下你是需要确认查询有返回结果，直接替换为assert not df.empty即可解决问题。

内容的提问来源于stack exchange，提问作者3Mcollab