python - 如何测试 PyVista 是否成功绘制了图形？

Question

我在 PyVista 中生成 3D 网格，我想更新我的集成测试套件以确保它成功显示我的绘图。

我希望调整这里描述的方法，以使用 PyVista。不幸的是，我在 PyVista 中找不到任何与 plt.gcf() 等效的函数的任何结果。

有谁知道解决方法？

Answer 1

有几种方法可以做到这一点。首先，在成功的情节上pyvista返回一个实例。pyvista.plotting.renderer.CameraPosition例如：

>>> import pyvista
>>> sphere = pyvista.Sphere()
>>> cpos = sphere.plot(off_screen=True)
>>> print(type(cpos))
<class 'pyvista.plotting.renderer.CameraPosition'>

由于需要设置绘图和渲染器以正确显示绘图，因此获得返回相机位置意味着您的绘图成功。

或者，您可以保存屏幕截图并检查文件是否存在：

import os
import pyvista
sphere = pyvista.Sphere()
cpos = sphere.plot(off_screen=True, screenshot='tmp.png')
assert os.path.isfile('tmp.png')

您还可以检查保存图像的内容（或可能的文件大小）