FIX: title as well

jklymak · jklymak · commit f9c1c63ddce5 · 2022-02-02T14:24:57.000+01:00
diff --git a/lib/matplotlib/tests/test_colorbar.py b/lib/matplotlib/tests/test_colorbar.py
@@ -982,6 +982,7 @@ def test_colorbar_set_formatter_locator():
 
 
 def test_offset_text_loc():
+    plt.style.use('mpl20')
     fig, ax = plt.subplots()
     np.random.seed(seed=19680808)
     pc = ax.pcolormesh(np.random.randn(10, 10)*1e6)
@@ -991,3 +992,17 @@ def test_offset_text_loc():
     # colorbar axes.  In this case the colorbar axes is the same
     # height as the parent, so use the parents bbox.
     assert cb.ax.yaxis.offsetText.get_position()[1] > ax.bbox.y1
+
+
+def test_title_text_loc():
+    plt.style.use('mpl20')
+    fig, ax = plt.subplots()
+    np.random.seed(seed=19680808)
+    pc = ax.pcolormesh(np.random.randn(10, 10))
+    cb = fig.colorbar(pc, location='right', extend='max')
+    cb.ax.set_title('Aardvark')
+    fig.draw_without_rendering()
+    # check that the title is in the proper place above the
+    # colorbar axes, including its extend triangles....
+    assert (cb.ax.title.get_window_extent(fig.canvas.get_renderer()).ymax >
+            cb.ax.spines['outline'].get_window_extent().ymax)
