jinx/python-uncompyle6
1
Fork 0
mirror of https://github.com/rocky/python-uncompyle6.git synced 2025-08-04 01:09:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

More stringent transform test for assert

Browse Source
This commit is contained in:
rocky
2019-12-14 11:31:13 -05:00
parent 56bf3e3125
commit 41b6e91286
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
uncompyle6/semantics/transform.py
View File

@@ -90,7 +90,11 @@ class TreeTransform(GenericASTTraversal, object):
if stmts in ("c_stmts",) and len(stmts) == 1: if stmts in ("c_stmts",) and len(stmts) == 1:
stmt = stmts[0] stmt = stmts[0]
raise_stmt = stmt[0] raise_stmt = stmt[0]
if raise_stmt == "raise_stmt1" and len(testexpr[0]) == 2: if (
raise_stmt == "raise_stmt1"
and len(testexpr[0]) == 2
and raise_stmt.first_child().pattr == "AssertionError"
):
assert_expr = testexpr[0][0] assert_expr = testexpr[0][0]
assert_expr.kind = "assert_expr" assert_expr.kind = "assert_expr"
jump_cond = testexpr[0][1] jump_cond = testexpr[0][1]