Python/ruff
1
0
Fork 0
mirror of https://github.com/astral-sh/ruff synced 2026-01-23 06:20:55 -05:00
Code Issues Packages Projects Releases Wiki Activity

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
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()
}