Added tests

kdpenner · kdpenner · commit 7d43704830e9 · 2021-08-06T12:14:02.000-04:00
diff --git a/lib/matplotlib/tests/test_legend.py b/lib/matplotlib/tests/test_legend.py
@@ -14,6 +14,7 @@
 import matplotlib.legend as mlegend
 from matplotlib import rc_context
 from matplotlib.font_manager import FontProperties
+from matplotlib.patches import Ellipse
 
 
 def test_legend_ordereddict():
@@ -492,6 +493,25 @@ def test_handler_numpoints():
     ax.legend(numpoints=0.5)
 
 
+def test_ellipse_nohandler_warning():
+    """Test that Ellipse artists with labels raise a warning"""
+    fig, ax = plt.subplots()
+    ellip = Ellipse(xy=(0, 0), width=1, height=1, label="ellipse")
+    ax.add_patch(ellip)
+    with pytest.warns(UserWarning) as record:
+        ax.legend()
+    assert len(record) == 1
+
+
+def test_text_nohandler_warning():
+    """Test that Text artists with labels raise a warning"""
+    fig, ax = plt.subplots()
+    ax.text(x=0, y=0, s="text", label="label")
+    with pytest.warns(UserWarning) as record:
+        ax.legend()
+    assert len(record) == 1
+
+
 def test_empty_bar_chart_with_legend():
     """Test legend when bar chart is empty with a label."""
     # related to issue #13003. Calling plt.legend() should not
