Add test

astral-sh · May 19, 2023 · 41a0ef8 · 41a0ef8
1 parent 25cc1da
commit 41a0ef8
Show file tree

Hide file tree

Showing 2 changed files with 92 additions and 0 deletions.
diff --git a/parser/src/parser.rs b/parser/src/parser.rs
@@ -853,4 +853,24 @@ def args_to_tuple(*args: *Ts) -> Tuple[*Ts]: ...
         .unwrap();
         insta::assert_debug_snapshot!(parse_ast);
     }
+
+    #[test]
+    #[cfg(not(feature = "all-nodes-with-ranges"))]
+    fn decorator_ranges() {
+        let parse_ast = parse_program(
+            r#"
+@my_decorator
+def test():
+    pass
+    
+@class_decorator
+class Abcd:
+    pass
+"#
+            .trim(),
+            "<test>",
+        )
+        .unwrap();
+        insta::assert_debug_snapshot!(parse_ast);
+    }
 }
diff --git a/parser/src/snapshots/rustpython_parser__parser__tests__decorator_ranges.snap b/parser/src/snapshots/rustpython_parser__parser__tests__decorator_ranges.snap