Include argument parentheses in range (#5)

Author: MichaReiser

parser/src/function.rs

@@ -157,7 +157,6 @@ mod tests {
     use super::*;
     use crate::{ast, parser::ParseErrorType, Parse};
 
-    #[cfg(not(feature = "all-nodes-with-ranges"))]
     macro_rules! function_and_lambda {
         ($($name:ident: $code:expr,)*) => {
             $(
@@ -170,6 +169,12 @@ mod tests {
         }
     }
 
+    #[cfg(feature = "all-nodes-with-ranges")]
+    function_and_lambda! {
+        test_function_no_args_with_ranges: "def f(): pass",
+        test_function_pos_args_with_ranges: "def f(a, b, c): pass",
+    }
+
     #[cfg(not(feature = "all-nodes-with-ranges"))]
     function_and_lambda! {
         test_function_no_args: "def f(): pass",

parser/src/python.lalrpop

@@ -981,15 +981,20 @@ FuncDef: ast::Stmt = {
 
 Parameters: ast::Arguments = {
     <location:@L> "(" <a: (ParameterList<TypedParameter, StarTypedParameter>)?> ")" <end_location:@R> =>? {
-        let args = a.map(validate_arguments).transpose()?.unwrap_or_else(|| ast::Arguments {
+        let range = optional_range(location, end_location);
+
+        let args = a.map(|mut arguments| {
+            arguments.range = range;
+            validate_arguments(arguments)
+        }).transpose()?.unwrap_or_else(|| ast::Arguments {
             posonlyargs: vec![],
             args: vec![],
             vararg: None,
             kwonlyargs: vec![],
             kw_defaults: vec![],
             kwarg: None,
             defaults: vec![],
-            range: optional_range(location, end_location)
+            range
         });
 
         Ok(args)