p53209761
/
test_ok2
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fixup comments

This commit is contained in:
hauntsaninja 2021-04-12 11:33:40 -07:00
parent da66f00413
commit e3dc34ee41
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -685,8 +685,8 @@ class AssertionRewriter(ast.NodeVisitor):
# Nothing to do. # Nothing to do.
return return
# Insert some special imports at the top of the module but after any # We'll insert some special imports at the top of the module, but after any
# docstrings and __future__ imports. # docstrings and __future__ imports, so first figure out where that is.
doc = getattr(mod, "docstring", None) doc = getattr(mod, "docstring", None)
expect_docstring = doc is None expect_docstring = doc is None
if doc is not None and self.is_rewrite_disabled(doc): if doc is not None and self.is_rewrite_disabled(doc):
@ -718,6 +718,7 @@ class AssertionRewriter(ast.NodeVisitor):
lineno = item.decorator_list[0].lineno lineno = item.decorator_list[0].lineno
else: else:
lineno = item.lineno lineno = item.lineno
# Now actually insert the special imports.
if sys.version_info >= (3, 10): if sys.version_info >= (3, 10):
aliases = [ aliases = [
ast.alias("builtins", "@py_builtins", lineno=lineno, col_offset=0), ast.alias("builtins", "@py_builtins", lineno=lineno, col_offset=0),
@ -737,6 +738,7 @@ class AssertionRewriter(ast.NodeVisitor):
ast.Import([alias], lineno=lineno, col_offset=0) for alias in aliases ast.Import([alias], lineno=lineno, col_offset=0) for alias in aliases
] ]
mod.body[pos:pos] = imports mod.body[pos:pos] = imports
# Collect asserts. # Collect asserts.
nodes: List[ast.AST] = [mod] nodes: List[ast.AST] = [mod]
while nodes: while nodes: