commit 270caac2a7bb08838c53cbbcbe7b3aaa4d3e8a87
author Javi Merino <javi.merino@arm.com> Tue Aug 12 10:41:38 2014 +0100
committer Javi Merino <javi.merino@arm.com> Wed Aug 12 14:52:41 2015 +0100
tree 6e8fa6cd710691e063dc2cd589829abcae8f006f
parent 8aa119c411ca4e872e4202c445c5dcd127e0f8ff

plot_utils: teach pre_plot_setup() setup multirow plots

tests/test_plot_utils.py

diff --git a/tests/test_plot_utils.py b/tests/test_plot_utils.py
index b44c276..6b2ac8e 100644
--- a/tests/test_plot_utils.py
+++ b/tests/test_plot_utils.py
@@ -69,6 +69,11 @@
         axis = plot_utils.pre_plot_setup(ncols=2)
         self.assertEquals(len(axis), 2)
 
+        axis = plot_utils.pre_plot_setup(nrows=2, ncols=3)
+        self.assertEquals(len(axis), 2)
+        self.assertEquals(len(axis[0]), 3)
+        self.assertEquals(len(axis[1]), 3)
+
     def test_post_plot_setup(self):
         """Test that post_plot_setup() doesn't bomb"""