Skip to content

Commit 658ce8a

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent fc4d1e5 commit 658ce8a

File tree

1 file changed

+15
-8
lines changed

1 file changed

+15
-8
lines changed

src/_pytest/assertion/rewrite.py

Lines changed: 15 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -661,7 +661,9 @@ def __init__(
661661
self.enable_assertion_pass_hook = False
662662
self.source = source
663663
self.scope: tuple[ast.AST, ...] = ()
664-
self.variables_overwrite: defaultdict[tuple[ast.AST, ...], Dict[str, str]] = defaultdict(dict)
664+
self.variables_overwrite: defaultdict[
665+
tuple[ast.AST, ...], Dict[str, str]
666+
] = defaultdict(dict)
665667

666668
def run(self, mod: ast.Module) -> None:
667669
"""Find all assert statements in *mod* and rewrite them."""
@@ -1049,16 +1051,19 @@ def visit_Call(self, call: ast.Call) -> Tuple[ast.Name, str]:
10491051
new_args = []
10501052
new_kwargs = []
10511053
for arg in call.args:
1052-
if isinstance(arg, ast.Name) and arg.id in self.variables_overwrite.get(self.scope, {}):
1053-
arg = self.variables_overwrite[self.scope][arg.id] # type:ignore[assignment]
1054+
if isinstance(arg, ast.Name) and arg.id in self.variables_overwrite.get(
1055+
self.scope, {}
1056+
):
1057+
arg = self.variables_overwrite[self.scope][
1058+
arg.id
1059+
] # type:ignore[assignment]
10541060
res, expl = self.visit(arg)
10551061
arg_expls.append(expl)
10561062
new_args.append(res)
10571063
for keyword in call.keywords:
1058-
if (
1059-
isinstance(keyword.value, ast.Name)
1060-
and keyword.value.id in self.variables_overwrite.get(self.scope, {})
1061-
):
1064+
if isinstance(
1065+
keyword.value, ast.Name
1066+
) and keyword.value.id in self.variables_overwrite.get(self.scope, {}):
10621067
keyword.value = self.variables_overwrite[self.scope][
10631068
keyword.value.id
10641069
] # type:ignore[assignment]
@@ -1095,7 +1100,9 @@ def visit_Attribute(self, attr: ast.Attribute) -> Tuple[ast.Name, str]:
10951100
def visit_Compare(self, comp: ast.Compare) -> Tuple[ast.expr, str]:
10961101
self.push_format_context()
10971102
# We first check if we have overwritten a variable in the previous assert
1098-
if isinstance(comp.left, ast.Name) and comp.left.id in self.variables_overwrite.get(self.scope, {}):
1103+
if isinstance(
1104+
comp.left, ast.Name
1105+
) and comp.left.id in self.variables_overwrite.get(self.scope, {}):
10991106
comp.left = self.variables_overwrite[self.scope][
11001107
comp.left.id
11011108
] # type:ignore[assignment]

0 commit comments

Comments
 (0)