Successfully parsing lambdef statement

dlorch · May 5, 2016 · 11a6e4e · 11a6e4e
1 parent 732b95f
commit 11a6e4e
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/pycep/parser.py b/pycep/parser.py
@@ -1456,7 +1456,17 @@ def _lambdef(tokens):
 
         lambdef: 'lambda' [varargslist] ':' test
     """
-    raise NotImplementedError
+    result = [symbol.lambdef]
+
+    result.append(tokens.accept(token.NAME, "lambda"))
+
+    if not tokens.check(token.OP, ":"):
+        result.append(_varargslist(tokens))
+
+    result.append(tokens.accept(token.OP, ":", result_token=token.COLON))
+    result.append(_test(tokens))
+
+    return result
 
 def _trailer(tokens):
     """Parse a trailer.

diff --git a/pycep/tests/snippets/pos/astrein.py b/pycep/tests/snippets/pos/astrein.py
@@ -107,6 +107,6 @@ def divide(x, y):
 
 4 ** 16
 
-#g = lambda x: x**2
+g = lambda x: x**2
 
 5 if foo == 30 else 19