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

Question

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

我希望采用here所描述的方法来使用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')

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