Add test for patches setter and getter

Impaler343 · Impaler343 · commit 03fb397d1f31 · 2024-04-10T16:04:36.000+05:30
diff --git a/lib/matplotlib/collections.pyi b/lib/matplotlib/collections.pyi
@@ -48,7 +48,7 @@ class Collection(artist.Artist, cm.ScalarMappable):
     def get_urls(self) -> Sequence[str | None]: ...
     def set_hatch(self, hatch: str) -> None: ...
     def get_hatch(self) -> str: ...
-    def set_hatch_linewidth(self, hatch_linewidth: float) -> None: ...
+    def set_hatch_linewidth(self, lw: float) -> None: ...
     def get_hatch_linewidth(self) -> float: ...
     def set_offsets(self, offsets: ArrayLike) -> None: ...
     def get_offsets(self) -> ArrayLike: ...
diff --git a/lib/matplotlib/tests/test_patches.py b/lib/matplotlib/tests/test_patches.py
@@ -960,3 +960,20 @@ def test_arrow_set_data():
     )
     arrow.set_data(x=.5, dx=3, dy=8, width=1.2)
     assert np.allclose(expected2, np.round(arrow.get_verts(), 2))
+
+
+@check_figures_equal(extensions=["png", "pdf", "svg", "eps"])
+def test_set_and_get_hatch_linewidth(fig_test, fig_ref):
+    ax_test = fig_test.add_subplot()
+    ax_ref = fig_ref.add_subplot()
+
+    lw = 2.0
+
+    with plt.rc_context({"hatch.linewidth": lw}):
+        ax_ref.add_patch(mpatches.Rectangle((0, 0), 1, 1, hatch="x"))
+
+    ax_test.add_patch(mpatches.Rectangle((0, 0), 1, 1, hatch="x"))
+    ax_test.patches[0].set_hatch_linewidth(lw)
+
+    assert (ax_ref.patches[0].get_hatch_linewidth() == lw)
+    assert (ax_test.patches[0].get_hatch_linewidth() == lw)
