1 import compiler

     2 import dis

     3 import types

     4 

     5 def extract_code_objects(co):

     6     l = [co]

     7     for const in co.co_consts:

     8         if type(const) == types.CodeType:

     9             l.append(const)

    10     return l

    11 

    12 def compare(a, b):

    13     if not (a.co_name == "?" or a.co_name.startswith('<lambda')):

    14         assert a.co_name == b.co_name, (a, b)

    15     if a.co_stacksize != b.co_stacksize:

    16         print "stack mismatch %s: %d vs. %d" % (a.co_name,

    17                                                 a.co_stacksize,

    18                                                 b.co_stacksize)

    19         if a.co_stacksize > b.co_stacksize:

    20             print "good code"

    21             dis.dis(a)

    22             print "bad code"

    23             dis.dis(b)

    24             assert 0

    25 

    26 def main(files):

    27     for file in files:

    28         print file

    29         buf = open(file).read()

    30         try:

    31             co1 = compile(buf, file, "exec")

    32         except SyntaxError:

    33             print "skipped"

    34             continue

    35         co2 = compiler.compile(buf, file, "exec")

    36         co1l = extract_code_objects(co1)

    37         co2l = extract_code_objects(co2)

    38         for a, b in zip(co1l, co2l):

    39             compare(a, b)

    40 

    41 if __name__ == "__main__":

    42     import sys

    43     main(sys.argv[1:])