Parse sequences of words into one literal

dictation-toolbox · Nov 10, 2019 · 320dc67 · 320dc67
1 parent eba046b
commit 320dc67
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/dragonfly/parsing/parse.py b/dragonfly/parsing/parse.py
@@ -26,7 +26,7 @@ def optional(self, args):
         return Optional(args[0])
 
     def literal(self, args):
-        return Literal(args[0])
+        return Literal(" ".join(args))
 
     def sequence(self, args):
         return Sequence(args)

diff --git a/dragonfly/test/test_lark_parser.py b/dragonfly/test/test_lark_parser.py
@@ -21,6 +21,9 @@ class TestLarkParser(unittest.TestCase):
     def test_literal(self):
         check_parse_tree("test   ", Literal(u"test"))
 
+    def test_multiple_literals(self):
+        check_parse_tree("test  hello world ", Literal(u"test hello world"))
+
     def test_parens(self):
         check_parse_tree("(test )   ", Literal(u"test"))