language-servers/ruff
1
0
Fork 0
mirror of https://github.com/astral-sh/ruff.git synced 2025-09-27 20:42:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

Use Stmt::parse in lieu of Suite unwraps (#5002)

Browse source
This commit is contained in:
Charlie Marsh 2023-06-10 00:55:31 -04:00 • committed by GitHub
parent 42c8054268
commit 445e1723ab
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 25 additions and 35 deletions
Download patch file Download diff file Expand all files Collapse all files

12
crates/ruff_python_ast/src/source_code/generator.rs
View file

@ -1456,7 +1456,7 @@ impl<'a> Generator<'a> {
#[cfg(test)]
mod tests {
use rustpython_ast::Suite;
use rustpython_ast::Stmt;
use rustpython_parser::Parse;
use ruff_python_whitespace::LineEnding;
@ -1468,10 +1468,9 @@ mod tests {
let indentation = Indentation::default();
let quote = Quote::default();
let line_ending = LineEnding::default();
let program = Suite::parse(contents, "<filename>").unwrap();
let stmt = program.first().unwrap();
let stmt = Stmt::parse(contents, "<filename>").unwrap();
let mut generator = Generator::new(&indentation, quote, line_ending);
generator.unparse_stmt(stmt);
generator.unparse_stmt(&stmt);
generator.generate()
}
@ -1481,10 +1480,9 @@ mod tests {
line_ending: LineEnding,
contents: &str,
) -> String {
let program = Suite::parse(contents, "<filename>").unwrap();
let stmt = program.first().unwrap();
let stmt = Stmt::parse(contents, "<filename>").unwrap();
let mut generator = Generator::new(indentation, quote, line_ending);
generator.unparse_stmt(stmt);
generator.unparse_stmt(&stmt);
generator.generate()
}