TST: add test that colorbar goes on bottom

jklymak · jklymak · commit 05f14ceaf835 · 2020-07-23T11:17:04.000-07:00
diff --git a/lib/mpl_toolkits/tests/test_mplot3d.py b/lib/mpl_toolkits/tests/test_mplot3d.py
@@ -1011,3 +1011,21 @@ def test_equal_box_aspect():
     ax.set_zlim3d(XYZlim)
     ax.axis('off')
     ax.set_box_aspect((1, 1, 1))
+
+
+def test_colorbar_pos():
+    num_plots = 2
+    fig = plt.figure(figsize=plt.figaspect(1 / num_plots),
+                     constrained_layout=True)
+    ax_l = []
+    for idx in range(num_plots):
+        ax = fig.add_subplot(1, num_plots, idx+1, projection='3d')
+        p_tri = ax.plot_trisurf(np.random.randn(5), np.random.randn(5), np.random.randn(5),
+                                cmap=plt.cm.coolwarm)
+        ax_l.append(ax)
+
+    cbar = plt.colorbar(p_tri, ax=ax_l, orientation='horizontal')
+
+    fig.canvas.draw()
+    # check that actually on the bottom
+    assert cbar.ax.get_position().extents[1] < 0.2
