Allow type declarations on rest parameters

marijnh · marijnh · commit ac7539eef06b · 2025-03-05T15:15:12.000+01:00
FIX: Allow type declarations on rest parameters. Closes codemirror/dev#1536
diff --git a/src/python.grammar b/src/python.grammar
@@ -74,7 +74,7 @@ MappingPattern { "{" commaSep<"**" pattern | (VariableName | LiteralPattern) ":"
 
 ClassDefinition { kw<"class"> VariableName TypeParamList? ArgList? Body }
 
-param { VariableName TypeDef? (AssignOp{"="} test)? | "*" VariableName? | "**" VariableName | "/" }
+param { VariableName TypeDef? (AssignOp{"="} test)? | "*" (VariableName TypeDef?)? | "**" VariableName TypeDef? | "/" }
 
 TypeDef { ":" test }
 
diff --git a/test/statement.txt b/test/statement.txt
@@ -422,3 +422,15 @@ type Point = tuple[float, float]
 ==>
 
 Script(TypeDefinition(type,VariableName,MemberExpression(VariableName,VariableName,",",VariableName)))
+
+# Rest arg with type
+
+def f(*args: tuple[int]) -> int:
+  return 1
+
+==>
+
+Script(FunctionDefinition(def,VariableName,
+  ParamList(VariableName,TypeDef(MemberExpression(VariableName,VariableName))),
+  TypeDef(VariableName),
+  Body(ReturnStatement(return,Number))))
