Merge pull request #7520 from dstansby/table-bug

NelleV · NelleV · commit 55465523622f · 2016-11-27T11:09:12.000-08:00
Fix table.py bug
diff --git a/lib/matplotlib/table.py b/lib/matplotlib/table.py
@@ -558,7 +558,7 @@ def table(ax,
         # assume just colours are needed
         rows = len(cellColours)
         cols = len(cellColours[0])
-        cellText = [[''] * rows] * cols
+        cellText = [[''] * cols] * rows
 
     rows = len(cellText)
     cols = len(cellText[0])
diff --git a/lib/matplotlib/tests/test_table.py b/lib/matplotlib/tests/test_table.py
@@ -5,13 +5,20 @@
 
 import matplotlib.pyplot as plt
 import numpy as np
-from matplotlib.testing.decorators import image_comparison
+from matplotlib.testing.decorators import image_comparison, cleanup
 
 from matplotlib.table import CustomCell
 from matplotlib.path import Path
 from nose.tools import assert_equal
 
 
+@cleanup
+def test_non_square():
+    # Check that creating a non-square table works
+    cellcolors = ['b', 'r']
+    plt.table(cellColours=cellcolors)
+
+
 @image_comparison(baseline_images=['table_zorder'],
                   extensions=['png'],
                   remove_text=True)
