Python matplotlib check_figures_equal用法及代码示例

本文简要介绍 python 语言中 matplotlib.testing.decorators.check_figures_equal 的用法。

用法

matplotlib.testing.decorators.check_figures_equal(*, extensions=('png', 'pdf', 'svg'), tol=0)

用于生成和比较两个图形的测试用例的装饰器。

装饰函数必须采用两个关键字参数 fig_test 和 fig_ref ，并在其上绘制测试和参考图像。函数返回后，保存数字并进行比较。

如果可能的话，这个装饰器应该优先于image_comparison，以防止测试套件的大小膨胀。

参数：

extensions 列表，默认：["png", "pdf", "svg"]

要测试的扩展。

tol 浮点数

高于该阈值的测试被视为失败的 RMS 阈值。

抛出：

RuntimeError

如果在测试函数中创建了任何新图形(并且随后没有关闭)。

例子

检查使用单个参数调用 Axes.plot 是否将其与 [0, 1, 2, ...] 相匹配：

@check_figures_equal()
def test_plot(fig_test, fig_ref):
    fig_test.subplots().plot([1, 3, 5])
    fig_ref.subplots().plot([0, 1, 2], [1, 3, 5])