Avoid ax.texts in rotation demo

QuLogic · QuLogic · commit 59c170b91521 · 2022-08-12T01:02:10.000-04:00
diff --git a/examples/text_labels_and_annotations/demo_text_rotation_mode.py b/examples/text_labels_and_annotations/demo_text_rotation_mode.py
@@ -50,6 +50,8 @@ def test_rotation_mode(fig, mode, subplot_location):
         {"bbox": dict(boxstyle="square,pad=0.", ec="none", fc="C1", alpha=0.3)}
     )
 
+    texts = {}
+
     # use a different text alignment in each axes
     for i, va in enumerate(va_list):
         for j, ha in enumerate(ha_list):
@@ -64,12 +66,12 @@ def test_rotation_mode(fig, mode, subplot_location):
                          size="x-large", rotation=40,
                          horizontalalignment=ha, verticalalignment=va,
                          rotation_mode=mode, **kw)
+            texts[ax] = tx
 
     if mode == "default":
         # highlight bbox
         fig.canvas.draw()
-        for ax in axs.flat:
-            text, = ax.texts
+        for ax, text in texts.items():
             bb = text.get_window_extent().transformed(ax.transData.inverted())
             rect = plt.Rectangle((bb.x0, bb.y0), bb.width, bb.height,
                                  facecolor="C1", alpha=0.3, zorder=2)
