TST: Add test for large subscript

jklymak · jklymak · commit cb636e653507 · 2018-08-18T21:18:38.000-07:00
diff --git a/lib/matplotlib/axes/_base.py b/lib/matplotlib/axes/_base.py
@@ -2510,7 +2510,7 @@ def _update_title_position(self, renderer):
         for title in titles:
             x, y0 = title.get_position()
             y = 1
-            # need to start agin in case of window resizing
+            # need to start again in case of window resizing
             title.set_position((x, 1.0))
             # need to check all our twins too...
             axs = self._twinned_axes.get_siblings(self)
@@ -2528,19 +2528,17 @@ def _update_title_position(self, renderer):
                     # this happens for an empty bb
                     y = 1
             bt = title.get_window_extent(renderer)
-            # need to do twice to get right for manual resizing
-            for nn in range(1):
+            dy = bt.ymin - top
+            if dy < 0:
+                y = self.transAxes.inverted().transform(
+                        (0., top))[1]
+                title.set_position((x, y))
+                bt = title.get_window_extent(renderer)
                 dy = bt.ymin - top
                 if dy < 0:
                     y = self.transAxes.inverted().transform(
-                            (0., top))[1]
-                    title.set_position((x, y))
-                    bt = title.get_window_extent(renderer)
-                    dy = bt.ymin - top
-                    if dy < 0:
-                        y = self.transAxes.inverted().transform(
-                            (0., top - dy))[1]
-                    title.set_position((x, y))
+                        (0., top - dy))[1]
+                title.set_position((x, y))
             ymax = np.max((y, ymax))
         for title in titles:
             # now line up all the titles at the highest baseline.
diff --git a/lib/matplotlib/tests/baseline_images/test_text/large_subscript_title.png b/lib/matplotlib/tests/baseline_images/test_text/large_subscript_title.png
diff --git a/lib/matplotlib/tests/test_text.py b/lib/matplotlib/tests/test_text.py
@@ -510,3 +510,20 @@ def test_annotation_update():
 
     assert not np.allclose(extent1.get_points(), extent2.get_points(),
                            rtol=1e-6)
+
+
+@image_comparison(baseline_images=['large_subscript_title'],
+                  extensions=['png'], style='mpl20')
+def test_large_subscript_title():
+    fig, axs = plt.subplots(1, 2, figsize=(9, 2.5), constrained_layout=True)
+    ax = axs[0]
+    ax.set_title('$\sum_{i} x_i$')
+    ax.set_title('New way', loc='left')
+    ax.set_xticklabels('')
+
+    ax = axs[1]
+    tt = ax.set_title('$\sum_{i} x_i$')
+    x, y = tt.get_position()
+    tt.set_position((x, 1.01))
+    ax.set_title('Old Way', loc='left')
+    ax.set_xticklabels('')
