Skip to content

Commit 72afed6

Browse files
[refactor to use match] AssertionRewriter.visit_Call()
1 parent 5046151 commit 72afed6

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
@@ -1067,10 +1067,11 @@ def visit_Call(self, call: ast.Call) -> tuple[ast.Name, str]:
10671067
arg_expls.append(expl)
10681068
new_args.append(res)
10691069
for keyword in call.keywords:
1070-
if isinstance(
1071-
keyword.value, ast.Name
1072-
) and keyword.value.id in self.variables_overwrite.get(self.scope, {}):
1073-
keyword.value = self.variables_overwrite[self.scope][keyword.value.id] # type:ignore[assignment]
1070+
match keyword.value:
1071+
case ast.Name(id=id) if id in self.variables_overwrite.get(
1072+
self.scope, {}
1073+
):
1074+
keyword.value = self.variables_overwrite[self.scope][id] # type:ignore[assignment]
10741075
res, expl = self.visit(keyword.value)
10751076
new_kwargs.append(ast.keyword(keyword.arg, res))
10761077
if keyword.arg:

0 commit comments

Comments
 (0)