Support lambda invocations

openrewrite · Oct 18, 2024 · 07da017 · 07da017
1 parent 90f7ed5
commit 07da017
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/rewrite/rewrite/python/_parser_visitor.py b/rewrite/rewrite/python/_parser_visitor.py
@@ -1156,7 +1156,7 @@ def visit_Call(self, node):
                 self.__map_type(node.func.value),
                 None
             )
-        elif isinstance(node.func, (ast.Call, ast.Subscript)):
+        elif isinstance(node.func, (ast.Call, ast.Lambda, ast.Subscript)):
             select = self.__pad_right(cast(Expression, self.__convert(node.func)), self.__whitespace())
             # printer handles empty name by not printing `.` before it
             name = self.__convert_name('')

diff --git a/rewrite/tests/python/all/method_invocation_test.py b/rewrite/tests/python/all/method_invocation_test.py
@@ -18,6 +18,11 @@ def test_invoke_function_receiver():
     rewrite_run(python("assert a(0)(1)"))
 
 
+def test_invoke_lambda_receiver():
+    # language=python
+    rewrite_run(python("assert (lambda x: x)(1)"))
+
+
 def test_invoke_array_access_receiver():
     # language=python
     rewrite_run(python("assert a[0](1)"))