cpython/Lib/test/test_dis.py

59 lines
1.5 KiB
Python
Raw Normal View History

2003-02-27 17:27:07 -04:00
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
2003-02-27 17:27:07 -04:00
dis_f = """\
13 0 LOAD_FAST 0 (a)
3 PRINT_ITEM
4 PRINT_NEWLINE
2003-02-27 17:27:07 -04:00
14 5 LOAD_CONST 1 (1)
8 RETURN_VALUE
2003-02-27 17:27:07 -04:00
9 LOAD_CONST 0 (None)
12 RETURN_VALUE
2003-02-27 17:27:07 -04:00
"""
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)
2003-02-27 17:27:07 -04:00
def test_main():
run_unittest(DisTests)
if __name__ == "__main__":
test_main()