Move formatting variables under the "has impls" if

Small optimization, move the generation of the intermediate
formatting variables inside the 'if _check_if_assertion_pass_impl():'
block.
This commit is contained in:
Bruno Oliveira 2019-06-26 19:26:12 -03:00
parent 8edf68f3c0
commit 629eb3ec6a
1 changed files with 4 additions and 4 deletions

View File

@ -761,11 +761,11 @@ class AssertionRewriter(ast.NodeVisitor):
)
# If any hooks implement assert_pass hook
hook_impl_test = ast.If(
self.helper("_check_if_assertion_pass_impl"), [hook_call_pass], []
self.helper("_check_if_assertion_pass_impl"),
self.expl_stmts + [hook_call_pass],
[],
)
statements_pass = []
statements_pass.extend(self.expl_stmts)
statements_pass.append(hook_impl_test)
statements_pass = [hook_impl_test]
# Test for assertion condition
main_test = ast.If(negation, statements_fail, statements_pass)