Test that non-square tables work

dstansby · dstansby · commit 4d412fda1f2c · 2016-11-27T17:23:18.000Z
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)
