Skip to content

Commit 7471ede

Browse files
[refactor to use match] AssertionRewriter.visit_Call()
1 parent a1eb153 commit 7471ede

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
@@ -1072,10 +1072,11 @@ def visit_Call(self, call: ast.Call) -> tuple[ast.Name, str]:
10721072
arg_expls.append(expl)
10731073
new_args.append(res)
10741074
for keyword in call.keywords:
1075-
if isinstance(
1076-
keyword.value, ast.Name
1077-
) and keyword.value.id in self.variables_overwrite.get(self.scope, {}):
1078-
keyword.value = self.variables_overwrite[self.scope][keyword.value.id] # type:ignore[assignment]
1075+
match keyword.value:
1076+
case ast.Name(id=id) if id in self.variables_overwrite.get(
1077+
self.scope, {}
1078+
):
1079+
keyword.value = self.variables_overwrite[self.scope][id] # type:ignore[assignment]
10791080
res, expl = self.visit(keyword.value)
10801081
new_kwargs.append(ast.keyword(keyword.arg, res))
10811082
if keyword.arg:

0 commit comments

Comments
 (0)