matplotlib
diff --git a/‎lib/matplotlib/tests/baseline_images/test_table/table_auto_column.png‎
26.7 KB b/‎lib/matplotlib/tests/baseline_images/test_table/table_auto_column.png‎
26.7 KB
diff --git a/‎lib/matplotlib/tests/test_table.py‎
Lines changed: 40 additions & 0 deletions b/‎lib/matplotlib/tests/test_table.py‎
Lines changed: 40 additions & 0 deletions
@@ -121,3 +121,43 @@ def test_customcell():
         cell = CustomCell((0, 0), visible_edges=t, width=1, height=1)
         code = tuple(s for _, s in cell.get_path().iter_segments())
         assert_equal(c, code)
+
+@image_comparison(baseline_images=['table_auto_column'],
+                  extensions=['png'])
+def test_auto_column():
+    fig = plt.figure()
+
+    #iteratble list input
+    ax1 = fig.add_subplot(3, 1, 1)
+    ax1.axis('off')
+    tb1 = ax1.table(cellText=[['Fit Text', 2],['Longer text than default', 1]],
+          rowLabels=["A","B"],
+          colLabels=["Col1","Col2"],
+          loc="center")
+    tb1.auto_set_font_size(False)
+    tb1.set_fontsize(12)
+    tb1.auto_set_column_width([-1,0,1])
+
+    #iteratble tuple input
+    ax2 = fig.add_subplot(3, 1, 2)
+    ax2.axis('off')
+    tb2 = ax2.table(cellText=[['Fit Text', 2],['Longer text than default', 1]],
+          rowLabels=["A","B"],
+          colLabels=["Col1","Col2"],
+          loc="center")
+    tb2.auto_set_font_size(False)
+    tb2.set_fontsize(12)
+    tb2.auto_set_column_width((-1,0,1))
+
+    #3 single input
+    ax3 = fig.add_subplot(3, 1, 3)
+    ax3.axis('off')
+    tb3 = ax3.table(cellText=[['Fit Text', 2],['Longer text than default', 1]],
+          rowLabels=["A","B"],
+          colLabels=["Col1","Col2"],
+          loc="center")
+    tb3.auto_set_font_size(False)
+    tb3.set_fontsize(12)
+    tb3.auto_set_column_width(-1)
+    tb3.auto_set_column_width(0)
+    tb3.auto_set_column_width(1)