add a way to disable assertion rewriting for a module

This commit is contained in:
Benjamin Peterson
2011-05-24 17:30:35 -05:00
parent 993efe927b
commit 76cede83c0
2 changed files with 13 additions and 1 deletions

View File

@@ -76,6 +76,14 @@ class TestAssertionRewrite:
assert imp.col_offset == 0
assert isinstance(m.body[5], ast.Expr)
def test_dont_rewrite(self):
s = """'PYTEST_DONT_REWRITE'\nassert 14"""
m = rewrite(s)
assert len(m.body) == 2
assert isinstance(m.body[0].value, ast.Str)
assert isinstance(m.body[1], ast.Assert)
assert m.body[1].msg is None
def test_name(self):
def f():
assert False