Fix end location of nodes containing body

2022-12-11 12:35:28 +09:00
parent 3abdc87076
commit 1b7a272b77
3 changed files with 9 additions and 6 deletions
--- a/parser/src/parser.rs
+++ b/parser/src/parser.rs
@@ -173,7 +173,8 @@ class Foo(A, B):
 def __init__(self):
  pass
 def method_with_default(self, arg='default'):
-  pass";
+  pass
+";
        insta::assert_debug_snapshot!(parse_program(source, "<test>").unwrap());
    }

--- a/parser/src/snapshots/rustpython_parserparsertests__parse_class.snap
+++ b/parser/src/snapshots/rustpython_parserparsertests__parse_class.snap
@@ -62,8 +62,8 @@ expression: "parse_program(source, \"<test>\").unwrap()"
                    },
                    end_location: Some(
                        Location {
-                            row: 4,
-                            column: 1,
+                            row: 3,
+                            column: 6,
                        },
                    ),
                    custom: (),