blob: ab93d4a0ec37ae8c6150041a31cf30fd3fdf51e6 [file] [log] [blame]
from test.test_support import verify, verbose, TestFailed, run_unittest
import sys
import dis
import StringIO
# Minimal tests for dis module
import unittest
# placement is crucial!!! move the start of _f and you have to adjust the
# line numbers in dis_f
def _f(a):
print a
return 1
dis_f = """\
13 0 LOAD_FAST 0 (a)
3 PRINT_ITEM
4 PRINT_NEWLINE
14 5 LOAD_CONST 1 (1)
8 RETURN_VALUE
9 LOAD_CONST 0 (None)
12 RETURN_VALUE
"""
class DisTests(unittest.TestCase):
def test_opmap(self):
self.assertEqual(dis.opmap["STOP_CODE"], 0)
self.assertEqual(dis.opmap["LOAD_CONST"] in dis.hasconst, True)
self.assertEqual(dis.opmap["STORE_NAME"] in dis.hasname, True)
def test_opname(self):
self.assertEqual(dis.opname[dis.opmap["LOAD_FAST"]], "LOAD_FAST")
def test_boundaries(self):
self.assertEqual(dis.opmap["EXTENDED_ARG"], dis.EXTENDED_ARG)
self.assertEqual(dis.opmap["STORE_NAME"], dis.HAVE_ARGUMENT)
def test_dis(self):
s = StringIO.StringIO()
save_stdout = sys.stdout
sys.stdout = s
dis.dis(_f)
sys.stdout = save_stdout
got = s.getvalue()
# Trim trailing blanks (if any).
lines = got.split('\n')
lines = [line.rstrip() for line in lines]
got = '\n'.join(lines)
self.assertEqual(dis_f, got)
def test_main():
run_unittest(DisTests)
if __name__ == "__main__":
test_main()