Raise A002 for lambda

Author: cielavenir

flake8_builtins.py

@@ -58,7 +58,7 @@ def run(self):
             for child in ast.iter_child_nodes(statement):
                 child.__flake8_builtins_parent = statement
 
-        function_nodes = [ast.FunctionDef]
+        function_nodes = [ast.FunctionDef, ast.Lambda]
         if getattr(ast, 'AsyncFunctionDef', None):
             function_nodes.append(ast.AsyncFunctionDef)
         function_nodes = tuple(function_nodes)
@@ -136,7 +136,7 @@ def check_assignment(self, statement):
                     stack.extend(list(item.value.elts))
 
     def check_function_definition(self, statement):
-        if statement.name in self.names:
+        if not isinstance(statement, ast.Lambda) and statement.name in self.names:
             msg = self.assign_msg
             if type(statement.__flake8_builtins_parent) is ast.ClassDef:
                 msg = self.class_attribute_msg

run_tests.py

@@ -140,6 +140,11 @@ def bla(list):
     check_code(source, 'A002')
 
 
+def test_lambda_argument_message():
+    source = 'takefirst = lambda list: list[0]'
+    check_code(source, 'A002')
+
+
 def test_keyword_argument_message():
     source = """
     def bla(dict=3):