тестирование 3D-графиков matplotlib с помощью pytest

Question

Мне было интересно, возможно ли для теста py утверждать вывод 3D-графика с использованием pytest, например, если у меня есть параметрическая функция, принимающая пользовательский ввод:

def 3D_plot(a=4,b=100,c=2,d=1):

    import matplotlib as mpl
    from mpl_toolkits.mplot3d import Axes3D
    import numpy as np
    import matplotlib.pyplot as plt

    pl.rcParams['legend.fontsize'] = 10

    fig = plt.figure()
    ax = fig.gca(projection='3d')
    theta = np.linspace(-a * np.pi, a * np.pi, b)
    z = np.linspace(-c, c, b)
    r = z**c + d
    x = r * np.sin(theta)
    y = r * np.cos(theta)
    ax.plot(x, y, z, label='parametric curve')
    ax.legend()

    plt.show()

Я хочу написать @pytest.Тест mark.parametrize () для него, используя pytest.

С уважением