Merge pull request #20458 from dstansby/leg-text

tacaswell · web-flow · commit 9ee6ff5d4a8e · 2021-06-17T13:44:17.000-04:00
Set the axes of legend text
diff --git a/lib/matplotlib/legend.py b/lib/matplotlib/legend.py
@@ -828,6 +828,7 @@ def _init_legend_box(self, handles, labels, markerfirst=True):
                                    children=[self._legend_title_box,
                                              self._legend_handle_box])
         self._legend_box.set_figure(self.figure)
+        self._legend_box.axes = self.axes
         self.texts = text_list
         self.legendHandles = handle_list
 
diff --git a/lib/matplotlib/tests/test_legend.py b/lib/matplotlib/tests/test_legend.py
@@ -852,3 +852,12 @@ def test_legend_face_edgecolor():
     ax.fill_between([0, 1, 2], [1, 2, 3], [2, 3, 4],
                     facecolor='r', edgecolor='face', label='Fill')
     ax.legend()
+
+
+def test_legend_text_axes():
+    fig, ax = plt.subplots()
+    ax.plot([1, 2], [3, 4], label='line')
+    leg = ax.legend()
+
+    assert leg.axes is ax
+    assert leg.get_texts()[0].axes is ax
