Skip to content

Commit 7308a72

Browse files
[refactor to use match] AssertionRewriter.visit_Call()
1 parent b4e8769 commit 7308a72

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

0 commit comments

Comments
 (0)