使用pytest测试matplotlib 3D图

Question

我想知道py测试是否可以使用pytest断言3D绘图输出，例如，如果我有一个接受用户输入的参数函数：

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为它编写一个@ pytest.mark.parametrize（）测试。

此致