FIX: do not allow single element list of str in subplot_mosaic

tacaswell · tacaswell · commit a7ab36a8dd1b · 2021-03-03T22:53:31.000-05:00
Also check that the first element of the a list layout is not a string. closes #19631
diff --git a/lib/matplotlib/figure.py b/lib/matplotlib/figure.py
@@ -1692,6 +1692,8 @@ def _make_array(inp):
 
             """
             r0, *rest = inp
+            if isinstance(r0, str):
+                raise ValueError('List layout specification must be 2D')
             for j, r in enumerate(rest, start=1):
                 if isinstance(r, str):
                     raise ValueError('List layout specification must be 2D')
diff --git a/lib/matplotlib/tests/test_figure.py b/lib/matplotlib/tests/test_figure.py
@@ -760,6 +760,8 @@ def test_empty(self, fig_test, fig_ref, x, empty_sentinel):
     def test_fail_list_of_str(self):
         with pytest.raises(ValueError, match='must be 2D'):
             plt.subplot_mosaic(['foo', 'bar'])
+        with pytest.raises(ValueError, match='must be 2D'):
+            plt.subplot_mosaic(['foo'])
 
     @check_figures_equal(extensions=["png"])
     @pytest.mark.parametrize("subplot_kw", [{}, {"projection": "polar"}, None])
