handl 0 dimensions for tabular

Author: dbatten5

src/arraytex/api.py

@@ -82,9 +82,13 @@ def to_tabular(
         DimensionMismatchError: when there are mismatched column identifiers and
             dimensions
     """
-    if len(arr.shape) == 1:
+    n_dims = len(arr.shape)
+
+    if n_dims == 0:
+        n_cols = 1
+    elif n_dims == 1:
         n_cols = arr.shape[0]
-    elif len(arr.shape) == 2:
+    elif n_dims == 2:
         n_cols = arr.shape[1]
     else:
         raise TooManyDimensionsError

tests/test_api.py

@@ -259,5 +259,22 @@ def test_one_dimensional(self) -> None:
 \midrule
 1 & 2 & 3 \\
 \endrule
+\end{tabular}"""
+        )
+
+    def test_0_d(self) -> None:
+        """0 dimensional scalars are handled properly."""
+        mat = np.array(1)
+
+        out = to_tabular(mat)
+
+        assert (
+            out
+            == r"""\begin{tabular}{c}
+\toprule
+Col 1 \\
+\midrule
+1 \\
+\endrule
 \end{tabular}"""
         )