fix(comment): #28701 use decorator @timhoffm #28702 (comment)

cmp0xff · cmp0xff · commit d4e05ced5ca9 · 2024-09-04T21:07:33.000+02:00
diff --git a/lib/matplotlib/tests/test_collections.py b/lib/matplotlib/tests/test_collections.py
@@ -831,28 +831,21 @@ def test_collection_set_verts_array():
         assert np.array_equal(ap._codes, atp._codes)
 
 
-def _fig_to_bytes(fig):
-    with io.BytesIO() as bio:
-        fig.savefig(bio)
-        return bio.getvalue()
-
-
 @pytest.mark.parametrize("fname", ["fill_between", "fill_betweenx"])
-def test_fbp_collection_set_data(fname):
+@check_figures_equal(extensions=["png"])
+def test_fbp_collection_set_data(fname, fig_test, fig_ref):
     t = np.linspace(0, 16)
     f1 = np.sin(t)
     f2 = f1 + 0.2
     f3 = f2.copy()
     f3[10], f3[20] = f3[2], f3[20]
 
-    fig1, ax1 = plt.subplots()
-    getattr(ax1, fname)(t, f1, f2)
-    fig1b = _fig_to_bytes(fig1)
+    ax_ref = fig_ref.subplots()
+    getattr(ax_ref, fname)(t, f1, f2)
 
-    fig2, ax2 = plt.subplots()
+    ax2 = fig_test.subplots()
     coll2 = getattr(ax2, fname)(t, f1, f3)
     coll2.set_data(t, f1, f2)
-    assert fig1b == _fig_to_bytes(fig2)
 
 
 @pytest.mark.parametrize(("t_direction", "shape", "where", "msg"), [
