Added test for legend.shadow argument types

tuncbkose · tuncbkose · commit 60417e3d0c9c · 2022-12-08T16:00:35.000+02:00
diff --git a/lib/matplotlib/tests/test_legend.py b/lib/matplotlib/tests/test_legend.py
@@ -532,6 +532,30 @@ def test_empty_bar_chart_with_legend():
     plt.legend()
 
 
+def test_shadow_argument_types():
+    # Test that different arguments for shadow work as expected
+    fig, ax = plt.subplots()
+    ax.plot([1, 2, 3], label='test')
+    shadows = [True, False, 'red', (0.1, 0.2, 0.5), 'tab:cyan', False]
+
+    legends = (ax.legend(loc='upper left', shadow=True),               # True
+               ax.legend(loc='upper right', shadow=False),             # False
+               ax.legend(loc='center left', shadow='red'),             # Color string
+               ax.legend(loc='center right', shadow=(0.1, 0.2, 0.5)),  # Color tuple
+               ax.legend(loc='lower left', shadow='tab:cyan')          # Color tab
+               )
+    for l in legends:
+        ax.add_artist(l)
+    ax.legend(loc='lower right')  # default
+
+    legends = [c for c in ax.get_children() if isinstance(c, mpl.legend.Legend)]
+    assert len(legends) == len(shadows)
+    for i in range(len(legends)):
+        assert legends[i].shadow == shadows[i]
+
+    with pytest.raises(ValueError, match="color or bool"):
+        ax.legend(loc="upper left", shadow="aardvark")  # Bad argument
+
 def test_shadow_framealpha():
     # Test if framealpha is activated when shadow is True
     # and framealpha is not explicitly passed'''
