added tests

r3kste · r3kste · commit 827e6b17355f · 2024-08-25T01:40:05.000+05:30
diff --git a/lib/matplotlib/tests/test_collections.py b/lib/matplotlib/tests/test_collections.py
@@ -1331,3 +1331,81 @@ def test_striped_lines(fig_test, fig_ref, gapcolor):
     for x, gcol, ls in zip(x, itertools.cycle(gapcolor),
                            itertools.cycle(linestyles)):
         ax_ref.axvline(x, 0, 1, linestyle=ls, gapcolor=gcol, alpha=0.5)
+
+
+@check_figures_equal(extensions=['png', 'pdf', 'svg', 'eps'], tol=0.026)
+def test_collection_hatchcolor(fig_test, fig_ref):
+    from matplotlib.collections import PathCollection
+
+    ax_test = fig_test.subplots()
+    ax_ref = fig_ref.subplots()
+
+    rect = mpath.Path.unit_rectangle().deepcopy()
+    rect.vertices *= [0.2, 0.2]
+
+    # Test that setting hatchcolor works with how hatchcolor was previously controlled
+    r = rect.deepcopy()
+    r.vertices = rect.vertices + [0.05, 0.05]
+    with mpl.rc_context({"hatch.color": "red"}):
+        ax_ref.add_collection(PathCollection([r], hatch='/', edgecolor='blue'))
+    ax_test.add_collection(PathCollection([r], hatch='/', edgecolor='blue',
+                                          hatchcolor='red'))
+
+    color_list_1 = ['purple', 'red', 'green', 'yellow']
+    color_list_2 = ['yellow', 'green', 'red', 'purple']
+
+    # Test for when edgecolor and hatchcolor is set
+    # fig_ref uses a workaround to set the hatchcolor different from the edgecolor
+    r = [mpl.path.Path(rect.vertices + [0.3, 0.05 + 0.25 * i]) for i in range(4)]
+    ax_ref.add_collection(PathCollection(r, hatch='//', lw=0,
+                                         edgecolor=color_list_1))
+    ax_ref.add_collection(PathCollection(r, fc='none',
+                                         edgecolor=color_list_2))
+    ax_test.add_collection(PathCollection(r, hatch='//',
+                                          edgecolor=color_list_2,
+                                          hatchcolor=color_list_1))
+
+    # Test for explicitly setting edgecolor and then hatchcolor
+    r = [mpl.path.Path(e.vertices + [0.25, 0]) for e in r]
+    ax_ref.add_collection(PathCollection(r, hatch='//', lw=0,
+                                         edgecolor=color_list_1))
+    ax_ref.add_collection(PathCollection(r, fc='none',
+                                         edgecolor=color_list_2))
+    col = PathCollection(r, hatch='//')
+    col.set_edgecolor(color_list_2)
+    assert_array_equal(col.get_hatchcolor(), [mpl.colors.to_rgba(c)
+                                              for c in color_list_2])
+    col.set_hatchcolor(color_list_1)
+    assert_array_equal(col.get_hatchcolor(), [mpl.colors.to_rgba(c)
+                                              for c in color_list_1])
+    ax_test.add_collection(col)
+
+    # Test for explicitly setting hatchcolor and then edgecolor
+    r = [mpl.path.Path(e.vertices + [0.25, 0]) for e in r]
+    ax_ref.add_collection(PathCollection(r, hatch='//', lw=0,
+                                         edgecolor=color_list_1))
+    ax_ref.add_collection(PathCollection(r, fc='none',
+                                         edgecolor=color_list_2))
+    col = PathCollection(r, hatch='//')
+    col.set_hatchcolor(color_list_1)
+    assert_array_equal(col.get_hatchcolor(), [mpl.colors.to_rgba(c)
+                                              for c in color_list_1])
+    col.set_edgecolor(color_list_2)
+    assert_array_equal(col.get_hatchcolor(), [mpl.colors.to_rgba(c)
+                                              for c in color_list_1])
+    ax_test.add_collection(col)
+
+    # Test for default hatch color when edgecolor and hatchcolor is not set
+    col = PathCollection([rect], hatch='//')
+    assert_array_equal(col.get_hatchcolor(), [mpl.colors.to_rgba(
+        mpl.rcParams['patch.edgecolor'])])
+
+    # Test for hatch color when edgecolor is set to 'none' and hatchcolor is not set
+    col = PathCollection([rect], hatch='//', edgecolor='none')
+    assert_array_equal(col.get_hatchcolor(), [mpl.colors.to_rgba(
+        mpl.rcParams['patch.edgecolor'])])
+
+    # Test for hatch color when edgecolor is set to 'face' and hatchcolor is not set
+    col = PathCollection([rect], hatch='//', edgecolor='face')
+    assert_array_equal(col.get_hatchcolor(), [mpl.colors.to_rgba(
+        mpl.rcParams['patch.edgecolor'])])
diff --git a/lib/matplotlib/tests/test_patches.py b/lib/matplotlib/tests/test_patches.py
@@ -960,3 +960,46 @@ 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'], tol=0.021)
+def test_patch_hatchcolor(fig_test, fig_ref):
+    ax_ref = fig_ref.subplots()
+    ax_test = fig_test.subplots()
+
+    # Test that setting hatchcolor works with how hatchcolor was previously controlled
+    with mpl.rc_context({'hatch.color': 'red'}):
+        ax_ref.add_patch(Rectangle((0.05, 0.05), 0.4, 0.4, hatch='//'))
+    ax_test.add_patch(Rectangle((0.05, 0.05), 0.4, 0.4, hatch='//', hatchcolor='red'))
+
+    # Test for when edgecolor and hatchcolor is set
+    # fig_ref uses a workaround to set hatchcolor to blue and edgecolor to red
+    ax_ref.add_patch(Rectangle((0.05, 0.5), 0.4, 0.4, ec='yellow',
+                               lw=0, hatch='//'))
+    ax_ref.add_patch(Rectangle((0.05, 0.5), 0.4, 0.4, ec='red', fc='none'))
+    ax_test.add_patch(Rectangle((0.05, 0.5), 0.4, 0.4, hatch='//', ec='red',
+                                hatchcolor='yellow'))
+
+    # Test for explicitly setting edgecolor and then hatchcolor
+    ax_ref.add_patch(Rectangle((0.5, 0.05), 0.4, 0.4, ec='cyan', lw=0, hatch='//'))
+    ax_ref.add_patch(Rectangle((0.5, 0.05), 0.4, 0.4, ec='orange', fc='none'))
+    rect = Rectangle((0.5, 0.05), 0.4, 0.4, hatch='//')
+    rect.set_edgecolor('orange')
+    assert rect._hatch_color == mcolors.to_rgba('orange')
+    rect.set_hatchcolor('cyan')
+    assert rect._hatch_color == mcolors.to_rgba('cyan')
+    ax_test.add_patch(rect)
+
+    # Test for explicitly setting hatchcolor and then edgecolor
+    ax_ref.add_patch(Rectangle((0.5, 0.5), 0.4, 0.4, ec='purple', lw=0, hatch='//'))
+    ax_ref.add_patch(Rectangle((0.5, 0.5), 0.4, 0.4, ec='green', fc='none'))
+    rect = Rectangle((0.5, 0.5), 0.4, 0.4, hatch='//')
+    rect.set_hatchcolor('purple')
+    assert rect._hatch_color == mcolors.to_rgba('purple')
+    rect.set_edgecolor('green')
+    assert rect._hatch_color == mcolors.to_rgba('purple')
+    ax_test.add_patch(rect)
+
+    # Test for default hatch color when edgecolor and hatchcolor is not set
+    rect = Rectangle((0, 0), 1, 1, hatch='//')
+    assert rect._hatch_color == mcolors.to_rgba(mpl.rcParams['patch.edgecolor'])
