fix linter: previous effect state could brake lambda arguments usage

baverman · Jan 14, 2012 · 8bbcaa1 · 8bbcaa1
1 parent 4c1632f
commit 8bbcaa1
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/supplement/linter.py b/supplement/linter.py
@@ -339,9 +339,10 @@ def visit_Lambda(self, node):
         self.scope = Scope(self.scope)
         self.scope.lineno = node.lineno
         self.scope.offset = node.col_offset
-        with self.loop():
-            with self.indirect(False):
-                self.generic_visit(node)
+        with self.effect(None):
+            with self.loop():
+                with self.indirect(False):
+                    self.generic_visit(node)
 
         self.scope = self.scope.parent
 

diff --git a/tests/test_linter.py b/tests/test_linter.py
@@ -228,3 +228,10 @@ def test_lambda():
         def foo():
             return lambda boo, bar: boo + bar
     ''')
+
+def test_lambda_assigned_to_var():
+    assert_names('''
+        def foo():
+            $result$ = lambda boo, bar: boo + bar
+    ''')
+