Skip to content

Commit 59bf241

Browse files
[refactor to use match] AssertionRewriter.visit_Call()
1 parent 125b045 commit 59bf241

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

src/_pytest/assertion/rewrite.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1069,10 +1069,11 @@ def visit_Call(self, call: ast.Call) -> tuple[ast.Name, str]:
10691069
arg_expls.append(expl)
10701070
new_args.append(res)
10711071
for keyword in call.keywords:
1072-
if isinstance(
1073-
keyword.value, ast.Name
1074-
) and keyword.value.id in self.variables_overwrite.get(self.scope, {}):
1075-
keyword.value = self.variables_overwrite[self.scope][keyword.value.id] # type:ignore[assignment]
1072+
match keyword.value:
1073+
case ast.Name(id=id) if id in self.variables_overwrite.get(
1074+
self.scope, {}
1075+
):
1076+
keyword.value = self.variables_overwrite[self.scope][id] # type:ignore[assignment]
10761077
res, expl = self.visit(keyword.value)
10771078
new_kwargs.append(ast.keyword(keyword.arg, res))
10781079
if keyword.arg:

0 commit comments

Comments
 (0)