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