Wrap return-bool-condition-directly fixes in bool() (#2240)

This commit is contained in:
Charlie Marsh 2023-01-26 18:22:34 -05:00 committed by GitHub
parent 615e62ae24
commit 093f9156e1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 12 additions and 5 deletions

View File

@ -178,7 +178,14 @@ pub fn return_bool_condition_directly(checker: &mut Checker, stmt: &Stmt) {
&& !has_comments_in(Range::from_located(stmt), checker.locator)
{
let return_stmt = create_stmt(StmtKind::Return {
value: Some(test.clone()),
value: Some(Box::new(create_expr(ExprKind::Call {
func: Box::new(create_expr(ExprKind::Name {
id: "bool".to_string(),
ctx: ExprContext::Load,
})),
args: vec![(**test).clone()],
keywords: vec![],
}))),
});
diagnostic.amend(Fix::replacement(
unparse_stmt(&return_stmt, checker.stylist),

View File

@ -40,7 +40,7 @@ pub fn explicit_true_false_in_ifexpr(
id: "bool".to_string(),
ctx: ExprContext::Load,
})),
args: vec![create_expr(test.node.clone())],
args: vec![test.clone()],
keywords: vec![],
}),
checker.stylist,

View File

@ -12,7 +12,7 @@ expression: diagnostics
column: 20
fix:
content:
- return a
- return bool(a)
location:
row: 3
column: 4
@ -30,7 +30,7 @@ expression: diagnostics
column: 20
fix:
content:
- return b
- return bool(b)
location:
row: 13
column: 4
@ -48,7 +48,7 @@ expression: diagnostics
column: 24
fix:
content:
- return b
- return bool(b)
location:
row: 24
column: 8