MNT: catch more invalid input

tacaswell · tacaswell · commit 6606961c837a · 2020-04-27T01:00:20.000-04:00
diff --git a/lib/matplotlib/figure.py b/lib/matplotlib/figure.py
@@ -1702,6 +1702,8 @@ def _do_layout(gs, layout, unique_ids, nested):
                         gs[slc], **{'label': str(name), **subplot_kw})
 
             for (j, k), nested_layout in nested.items():
+                if not all(len(r) for r in nested_layout):
+                    raise ValueError("All of the rows must be the same length")
                 nested_layout = np.asarray(nested_layout, dtype=object)
                 rows, cols = nested_layout.shape
                 gs_n = gs[j, k].subgridspec(rows, cols, **gridspec_kw)
@@ -1714,7 +1716,8 @@ def _do_layout(gs, layout, unique_ids, nested):
                                      f"and the nested layout\n{nested_layout}")
                 output.update(nested_output)
             return output
-
+        if not all(len(r) for r in layout):
+            raise ValueError("All of the rows must be the same length")
         layout = np.asarray(layout, dtype=object)
         rows, cols = layout.shape
         gs = self.add_gridspec(rows, cols, **gridspec_kw)
