floraachy
/
pytest2
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

do not rewrite assertion if a message is present

This commit is contained in:
ed 2024-05-23 17:56:09 +02:00
parent cbf6bd9dd2
commit acbe63008d
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/_pytest/assertion/rewrite.py
View File

@ -752,10 +752,13 @@ class AssertionRewriter(ast.NodeVisitor):
new: List[ast.AST] = [] new: List[ast.AST] = []
for i, child in enumerate(field): for i, child in enumerate(field):
if isinstance(child, ast.Assert): if isinstance(child, ast.Assert):
# Transform assert. # Transform assert if no message
if child.msg is None:
new.extend(self.visit(child)) new.extend(self.visit(child))
else: else:
new.append(child) new.append(child)
else:
new.append(child)
if isinstance(child, ast.AST): if isinstance(child, ast.AST):
nodes.append(child) nodes.append(child)
setattr(node, name, new) setattr(node, name, new)