Added test

oscargus · oscargus · commit 5e1b5fe206bd · 2022-05-29T17:56:29.000+02:00
diff --git a/examples/text_labels_and_annotations/legend.py b/examples/text_labels_and_annotations/legend.py
@@ -21,9 +21,9 @@
 ax.plot(a, d, 'k:', label='Data length')
 ax.plot(a, c + d, 'k', label='Total message length')
 
-#Create an arrow with pre-defined label.
+# Create an arrow with pre-defined label.
 ax.annotate("", xy=(1.5, 4.5), xytext=(1.5, 9.0),
-            arrowprops={'arrowstyle':'<->', 'color':'C7'}, label='Distance')
+            arrowprops={'arrowstyle': '<->', 'color': 'C7'}, label='Distance')
 
 legend = ax.legend(loc='upper center', shadow=True, fontsize='large')
 
diff --git a/lib/matplotlib/legend_handler.py b/lib/matplotlib/legend_handler.py
@@ -812,7 +812,8 @@ class HandlerFancyArrowPatch(HandlerPatch):
     """
     Handler for `~.FancyArrowPatch` instances.
     """
-    def _create_patch(self, legend, orig_handle, xdescent, ydescent, width, height, fontsize):
+    def _create_patch(self, legend, orig_handle, xdescent, ydescent, width,
+                      height, fontsize):
         arrow = FancyArrowPatch([-xdescent,
                                  -ydescent + height / 2],
                                 [-xdescent + width,
@@ -849,9 +850,7 @@ def create_artists(self, legend, orig_handle, xdescent, ydescent, width,
                        height, fontsize, trans):
 
         if orig_handle.arrow_patch is not None:
-
             # Arrow without text
-
             handler = HandlerFancyArrowPatch()
             handle = orig_handle.arrow_patch
 
diff --git a/lib/matplotlib/tests/test_legend.py b/lib/matplotlib/tests/test_legend.py
@@ -927,3 +927,22 @@ def test_legend_markers_from_line2d():
 
     assert markers == new_markers == _markers
     assert labels == new_labels
+
+
+def test_annotation_legend():
+    fig, ax = plt.subplots()
+    ax.annotate("", xy=(0.5, 0.5), xytext=(0.5, 0.7),
+                arrowprops={'arrowstyle': '<->'}, label="Bar")
+    legend = ax.legend()
+    assert len(legend.get_texts()) == 1
+    ax.annotate("Foo", xy=(0.3, 0.3))
+    legend = ax.legend()
+    assert len(legend.get_texts()) == 1
+    ax.annotate("FooBar", xy=(0.7, 0.7), xytext=(0.7, 0.9),
+                arrowprops={'arrowstyle': '->'})
+    legend = ax.legend()
+    assert len(legend.get_texts()) == 1
+    ax.annotate("Foo", xy=(0.1, 0.1), xytext=(0.1, 0.7),
+                arrowprops={'arrowstyle': '<-'}, label="Foo")
+    legend = ax.legend()
+    assert len(legend.get_texts()) == 2
