Fixes issue when specifying PS2 or PS3 (#106)

sillsdev · Apr 16, 2024 · 2a80929 · 2a80929
1 parent 5d05e6d
commit 2a80929
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/machine/scripture/parse.py b/machine/scripture/parse.py
@@ -44,7 +44,7 @@ def parse_selection(selection: str, versification: Versification) -> Dict[int, L
     selection = selection.strip()
     chapters = {}
 
-    if selection[-1].isdigit():  # Specific chapters from one book
+    if selection[-1].isdigit() and len(selection) > 3:  # Specific chapters from one book
         book = book_id_to_number(selection[:3])
         if book == 0:
             raise ValueError(f"{selection[:3]} is an invalid book ID.")

diff --git a/tests/scripture/test_parse.py b/tests/scripture/test_parse.py
@@ -36,6 +36,7 @@ def test_get_books() -> None:
 def test_get_chapters() -> None:
     assert get_chapters([]) == {}
     assert get_chapters("MAL") == {39: []}
+    assert get_chapters("PS2") == {84: []}
     assert get_chapters("GEN,EXO") == {1: [], 2: []}
     assert get_chapters("1JN,2JN") == {62: [], 63: []}
     assert get_chapters("OT") == {i: [] for i in range(1, 40)}