Как мне проверить, что PyVista успешно построила фигуру?

Question

Я создаю трехмерные сетки в PyVista, и я хотел бы обновить свой набор интеграционных тестов, чтобы убедиться, что он успешно отображает мои графики.

Я надеюсь адаптировать описанную здесь методологию для работы с PyVista. К сожалению, я не могу найти никаких результатов для какой-либо функции, эквивалентной plt.gcf() в PyVista.

Кто-нибудь знает обходной путь?

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')

Вы также можете проверить содержимое сохраненного изображения (или, возможно, размер файла)