You've already forked python-uncompyle6
mirror of
https://github.com/rocky/python-uncompyle6.git
synced 2025-08-03 00:45:53 +08:00
47 lines
1.9 KiB
Python
47 lines
1.9 KiB
Python
import pytest, re
|
|
from uncompyle6 import PYTHON_VERSION, PYTHON3, IS_PYPY # , PYTHON_VERSION
|
|
from uncompyle6.parser import get_python_parser
|
|
from uncompyle6.scanner import get_scanner
|
|
|
|
def test_grammar():
|
|
|
|
def check_tokens(tokens, opcode_set):
|
|
remain_tokens = set(tokens) - opcode_set
|
|
remain_tokens = set([re.sub('_\d+$','', t) for t in remain_tokens])
|
|
remain_tokens = set([re.sub('_CONT$','', t) for t in remain_tokens])
|
|
remain_tokens = set(remain_tokens) - opcode_set
|
|
assert remain_tokens == set([]), \
|
|
"Remaining tokens %s\n====\n%s" % (remain_tokens, p.dumpGrammar())
|
|
|
|
p = get_python_parser(PYTHON_VERSION, is_pypy=IS_PYPY)
|
|
lhs, rhs, tokens, right_recursive = p.checkSets()
|
|
expect_lhs = set(['expr1024', 'pos_arg'])
|
|
unused_rhs = set(['build_list', 'call_function', 'mkfunc', 'mklambda',
|
|
'unpack', 'unpack_list'])
|
|
expect_right_recursive = [['designList', ('designator', 'DUP_TOP', 'designList')]]
|
|
if PYTHON3:
|
|
expect_lhs.add('load_genexpr')
|
|
unused_rhs = unused_rhs.union(set("""
|
|
except_pop_except genexpr classdefdeco2 listcomp
|
|
""".split()))
|
|
else:
|
|
expect_lhs.add('kwarg')
|
|
assert expect_lhs == set(lhs)
|
|
assert unused_rhs == set(rhs)
|
|
assert expect_right_recursive == right_recursive
|
|
s = get_scanner(PYTHON_VERSION, IS_PYPY)
|
|
ignore_set = set(
|
|
"""
|
|
JUMP_BACK CONTINUE RETURN_END_IF
|
|
COME_FROM COME_FROM_EXCEPT COME_FROM_LOOP COME_FROM_WITH
|
|
LOAD_GENEXPR LOAD_ASSERT LOAD_SETCOMP LOAD_DICTCOMP
|
|
LAMBDA_MARKER RETURN_LAST
|
|
""".split())
|
|
if 2.6 <= PYTHON_VERSION <= 2.7:
|
|
opcode_set = set(s.opc.opname).union(ignore_set)
|
|
check_tokens(tokens, opcode_set)
|
|
elif PYTHON_VERSION == 3.4:
|
|
ignore_set.add('LOAD_CLASSNAME')
|
|
opcode_set = set(s.opc.opname).union(ignore_set)
|
|
check_tokens(tokens, opcode_set)
|