All unit tests with new fork syntax

Author: travis-bauer

src/talkpipe/chatterlang/parsers.py

@@ -290,13 +290,21 @@ def fork_section():
     yield lexeme(')')
     return ForkNode(branches=branches, params=bracket_content)
 
-transforms_section = (
-    # First transform can be without a leading pipe
-    (fork_section | segment | variable).map(lambda x: [x]) |
-    # Subsequent transforms require a leading pipe
-    (lexeme('|') >> (fork_section | segment | variable)).many()
-)
-"""A parser for the transforms section.  Transforms are separated by the '|' character."""
+@generate
+def transforms_section():
+    """A parser for the transforms section.  Transforms are separated by the '|' character."""
+    # First transform may or may not have a leading pipe (optional to allow empty transforms)
+    first_transform = yield (lexeme('|').optional() >> (fork_section | segment | variable)).optional()
+    if first_transform is None:
+        return []
+    
+    transforms = [first_transform]
+    
+    # Parse remaining segments (with leading pipes)
+    remaining = yield (lexeme('|') >> (fork_section | segment | variable)).many()
+    transforms.extend(remaining)
+    
+    return transforms
 
 # Parser for arrow fork target: -> identifier
 arrow_fork_target = (lexeme('->') >> identifier).map(lambda x: x.name)

tests/talkpipe/chatterlang/test_compiler.py

@@ -169,13 +169,13 @@ def test_fork_arrow_syntax_multiple_forks():
     script = """
              INPUT FROM range[lower=0, upper=3] -> fork1;
              INPUT FROM range[lower=10, upper=13] -> fork2;
-             fork1 -> lambda[expression="item * 2"] -> fork3;
+             fork1 -> lambda[expression="item * 2"] | lambda[expression="item + 1"] -> fork3;
              fork2 -> lambda[expression="item + 100"] -> fork3;
              fork3 -> toList
              """
     f = compiler.compile(script).as_function(single_in=True,single_out=True)
     ans = f()
-    assert sorted(ans) == sorted([0, 2, 4, 110, 111, 112])
+    assert sorted(ans) == sorted([1, 3, 5, 110, 111, 112])
 
 def test_fork_arrow_syntax_metadata():
 
@@ -187,10 +187,10 @@ def test_fork_arrow_syntax_metadata():
 
     script = """
              INPUT FROM range[lower=0, upper=3] | flushN[n=1] -> fork1;
-             fork1 -> collectMetadata -> toList
+             fork1 -> collectMetadata | toList
              """
-    f = compiler.compile(script)#.as_function(single_in=True,single_out=True)
-    ans2 = list(f())
+    f = compiler.compile(script).as_function(single_in=True,single_out=True)
+    ans2 = f()
     assert ans == ans2
 
 def test_variables_as_parameters():