get arg unpacking to work for lambdas

src/core/cfg.cpp

@@ -2502,7 +2502,7 @@ CFG* computeCFG(SourceInfo* source, std::vector<AST_stmt*> body) {
         }
     }
 
-    if (source->ast->type == AST_TYPE::FunctionDef) {
+    if (source->ast->type == AST_TYPE::FunctionDef || source->ast->type == AST_TYPE::Lambda) {
         // Unpack tuple arguments
         // Tuple arguments get assigned names ".0", ".1" etc. So this
         // def f(a, (b,c), (d,e)):
@@ -2510,7 +2510,12 @@ CFG* computeCFG(SourceInfo* source, std::vector<AST_stmt*> body) {
         // def f(a, .1, .2):
         //     (b, c) = .1
         //     (d, e) = .2
-        AST_arguments* args = ast_cast<AST_FunctionDef>(source->ast)->args;
+        AST_arguments* args;
+        if (source->ast->type == AST_TYPE::FunctionDef) {
+            args = ast_cast<AST_FunctionDef>(source->ast)->args;
+        } else {
+            args = ast_cast<AST_Lambda>(source->ast)->args;
+        }
         int counter = 0;
         for (AST_expr* arg_expr : args->args) {
             if (arg_expr->type == AST_TYPE::Tuple) {

test/tests/arg_unpacking.py

@@ -23,3 +23,10 @@ g(1, (2, 3), 4, (5, 6), (7, (8, 9), 10))
 def h((a,)):
     print a
 h((3,))
+
+print 'testing lambda'
+f = lambda (a, b) : a + b
+print f((3, 4))
+
+f = lambda (a,) : a + 1
+print f((3,))