Stop greedy choice after full input match

pyleri/choice.py

@@ -37,6 +37,8 @@ def _most_greedy_result(self, root, tree, rule, s, node):
             if is_valid and pos > mg_pos:
                 node.children = children
                 mg_is_valid, mg_pos = is_valid, pos
+                if pos == root._len_string:
+                    break
 
         if mg_is_valid:
             root._append_tree(tree, node, mg_pos)

test/test_choice.py

@@ -7,10 +7,12 @@
 from pyleri import (
     KeywordError,
     create_grammar,
+    Repeat,
     Sequence,
     Choice,
     Keyword,
     Regex,
+    Token,
 )  # nopep8
 
 
@@ -28,6 +30,20 @@ def test_choice_most_greedy(self):
         self.assertTrue(grammar.parse(' hi iris ').is_valid)
         self.assertFalse(grammar.parse(' hi sasha ').is_valid)
 
+    def test_choice_most_greedy_selects_full_match(self):
+        short = Token('a')
+        long = Token('aa')
+        grammar = create_grammar(Repeat(Choice(short, long)))
+
+        result = grammar.parse('aa')
+        repeat_node = result.tree.children[0]
+        choice_node = repeat_node.children[0]
+
+        self.assertTrue(result.is_valid)
+        self.assertEqual(len(repeat_node.children), 1)
+        self.assertEqual(choice_node.children[0].element, long)
+        self.assertEqual(choice_node.children[0].string, 'aa')
+
     def test_choice_first_match(self):
         k_hi = Keyword('hi')
         k_iris = Keyword('iris')