Add more alignment tests

Author: DenverCoder1

table2ascii/__init__.py

@@ -69,6 +69,12 @@ def __init__(
 
         self.__alignments = alignments or [Alignment.CENTER] * self.__columns
 
+        # check if alignments specified have a different number of columns
+        if alignments and len(alignments) != self.__columns:
+            raise ValueError(
+                "Length of `alignments` list must equal the number of columns"
+            )
+
         """
         ╔═════╦═══════════════════════╗   ABBBBBCBBBBBDBBBBBDBBBBBDBBBBBE
         ║  #  ║  G     H     R     S  ║   F     G     H     H     H     F

tests/test_alignments.py

@@ -1,5 +1,7 @@
 from table2ascii import table2ascii as t2a, Alignment
 
+import pytest
+
 
 def test_first_left_four_right():
     text = t2a(
@@ -20,3 +22,25 @@ def test_first_left_four_right():
         "╚═════╩═══════════════════════╝\n"
     )
     assert text == expected
+
+
+def test_wrong_number_alignments():
+    with pytest.raises(ValueError):
+        t2a(
+            header=["#", "G", "H", "R", "S"],
+            body=[["1", "30", "40", "35", "30"], ["2", "30", "40", "35", "30"]],
+            footer=["SUM", "130", "140", "135", "130"],
+            first_col_heading=True,
+            alignments=[Alignment.LEFT, Alignment.CENTER, Alignment.RIGHT],
+        )
+
+
+def test_invalid_alignments():
+    with pytest.raises(ValueError):
+        t2a(
+            header=["#", "G", "H", "R", "S"],
+            body=[["1", "30", "40", "35", "30"], ["2", "30", "40", "35", "30"]],
+            footer=["SUM", "130", "140", "135", "130"],
+            first_col_heading=True,
+            alignments=[9999, -1, Alignment.RIGHT],
+        )