feat: add _get_max_column_widths

still doesn't work because _get_delimiter_locations doesn't support stripping spaces

src/md_to_py.py

@@ -2,7 +2,8 @@
 Convert a MarkDown table to a formatted Python list.
 """
 
-from itertools import repeat
+from itertools import pairwise, repeat, starmap
+from operator import sub
 from typing import Callable, Iterable, Iterator, Sequence, TypeVar
 
 T = TypeVar("T")
@@ -60,25 +61,40 @@ def _get_column_widths(
 
 
 def _get_delimiter_locations(rows: Sequence[str], delimiter: str = "|") -> Iterator[int]:
+    """
+    Yield an iterator of delimiter locations.
+
+    The length of the iterator should be equal to any row.count(delimiter)
+    """
+
+    if len(delimiter) != 1:
+        raise ValueError(
+            f"`delimiter` must be one character, is {len(delimiter)} characters",
+        )
+    if delimiter == " ":
+        raise ValueError("`delimiter` cannot be a space")
+
     remaining_rows = len(rows)
     location_number = [0 for _ in range(remaining_rows)]
     column = 1
+    location_number[1] = rows[0].count(delimiter)
     for pos in range(len(max(rows, key=len))):
         for index, row in enumerate(rows):
+            if index == 1 or pos > len(row):
+                continue
             if pos == len(row):
                 remaining_rows -= 1
                 continue
-            if pos > len(row):
-                continue
             if row[pos] == delimiter:
                 location_number[index] += 1
             if all(num >= column for num in location_number):
                 column += 1
                 yield pos
 
 
-def _get_max_column_widths():
-    raise NotImplementedError()
+def _get_max_column_widths(rows: Sequence[str], delimiter: str = "|") -> Iterator[int]:
+    for item in starmap(sub, pairwise(_get_delimiter_locations(rows, delimiter))):
+        yield -item - 1
 
 
 def _pad_columns(row: str, widths: tuple[int, ...] | int, delimiter: str = "|") -> str: