mirrors/ruff
1
0
Fork 0
mirror of https://github.com/astral-sh/ruff.git synced 2025-07-21 20:15:11 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

Format assert statement (#5168)

Browse source
This commit is contained in:
Chris Pryer 2023-07-14 03:01:33 -04:00 • committed by GitHub
parent 5a4516b812
commit a961f75e13
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
14 changed files with 442 additions and 370 deletions
Download patch file Download diff file Expand all files Collapse all files

33
crates/ruff_python_formatter/src/statement/stmt_assert.rs
View file

@ -1,4 +1,7 @@
use crate::{not_yet_implemented, FormatNodeRule, PyFormatter};
use crate::expression::maybe_parenthesize_expression;
use crate::expression::parentheses::Parenthesize;
use crate::{FormatNodeRule, PyFormatter};
use ruff_formatter::prelude::{space, text};
use ruff_formatter::{write, Buffer, FormatResult};
use rustpython_parser::ast::StmtAssert;
@ -7,6 +10,32 @@ pub struct FormatStmtAssert;
impl FormatNodeRule<StmtAssert> for FormatStmtAssert {
fn fmt_fields(&self, item: &StmtAssert, f: &mut PyFormatter) -> FormatResult<()> {
write!(f, [not_yet_implemented(item)])
let StmtAssert {
range: _,
test,
msg,
} = item;
write!(
f,
[
text("assert"),
space(),
maybe_parenthesize_expression(test, item, Parenthesize::IfBreaks)
]
)?;
if let Some(msg) = msg {
write!(
f,
[
text(","),
space(),
maybe_parenthesize_expression(msg, item, Parenthesize::IfBreaks),
]
)?;
}
Ok(())
}
}