Include argument parentheses in range (#5)

Author: zanieb

parser/src/function.rs

@@ -136,7 +136,6 @@ mod tests {
     use super::*;
     use crate::{ast, parser::ParseErrorType, Parse};
 
-    #[cfg(feature = "all-nodes-with-ranges")]
     macro_rules! function_and_lambda {
         ($($name:ident: $code:expr,)*) => {
             $(
@@ -149,6 +148,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(feature = "all-nodes-with-ranges")]
     function_and_lambda! {
         test_function_no_args: "def f(): pass",

parser/src/parser.rs

@@ -1107,7 +1107,7 @@ def args_to_tuple(*args: *Ts) -> Tuple[*Ts]: ...
 @my_decorator
 def test():
     pass
-    
+
 @class_decorator
 class Abcd:
     pass

parser/src/python.lalrpop

@@ -981,8 +981,14 @@ FuncDef: ast::Stmt = {
 Parameters: ast::Arguments = {
     <location:@L> "(" <a: (ParameterList<TypedParameter, StarTypedParameter, DoubleStarTypedParameter>)?> ")" <end_location:@R> =>? {
         a.as_ref().map(validate_arguments).transpose()?;
+
+        let range = optional_range(location, end_location);
         let args = a
-            .unwrap_or_else(|| ast::Arguments::empty(optional_range(location, end_location)));
+            .map(|mut arguments| {
+                arguments.range = range;
+                arguments
+            })
+            .unwrap_or_else(|| ast::Arguments::empty(range));
 
         Ok(args)
     }