Set the axes of legend text

Author: dstansby

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
 

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