Tools/compiler/stacktest.py

   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:])