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

write an explicit raise if the assertion fails

This commit is contained in:
Benjamin Peterson 2011-05-26 21:15:40 -05:00
parent e22d3e03fe
commit e56838cb6c
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
_pytest/assertion/rewrite.py
View File

@ -3,6 +3,7 @@
import ast import ast
import collections import collections
import itertools import itertools
import sys
import py import py
from _pytest.assertion import util from _pytest.assertion import util
@ -212,7 +213,13 @@ class AssertionRewriter(ast.NodeVisitor):
template = ast.Str(explanation) template = ast.Str(explanation)
msg = self.pop_format_context(template) msg = self.pop_format_context(template)
fmt = self.helper("format_explanation", msg) fmt = self.helper("format_explanation", msg)
body.append(ast.Assert(top_condition, fmt)) err_name = ast.Name("AssertionError", ast.Load())
exc = ast.Call(err_name, [fmt], [], None, None)
if sys.version_info[0] >= 3:
raise_ = ast.Raise(exc, None)
else:
raise_ = ast.Raise(exc, None, None)
body.append(raise_)
# Delete temporary variables. # Delete temporary variables.
names = [ast.Name(name, ast.Del()) for name in self.variables] names = [ast.Name(name, ast.Del()) for name in self.variables]
if names: if names: