mirror of https://github.com/python/cpython
23 lines
872 B
Python
23 lines
872 B
Python
|
import dis
|
||
|
|
import _opcode
|
||
|
|
from test.support import run_unittest
|
||
|
|
import unittest
|
||
|
|
|
||
|
|
class OpcodeTests(unittest.TestCase):
|
||
|
|
|
||
|
|
def test_stack_effect(self):
|
||
|
|
self.assertEqual(_opcode.stack_effect(dis.opmap['POP_TOP']), -1)
|
||
|
|
self.assertEqual(_opcode.stack_effect(dis.opmap['DUP_TOP_TWO']), 2)
|
||
|
|
self.assertEqual(_opcode.stack_effect(dis.opmap['BUILD_SLICE'], 0), -1)
|
||
|
|
self.assertEqual(_opcode.stack_effect(dis.opmap['BUILD_SLICE'], 1), -1)
|
||
|
|
self.assertEqual(_opcode.stack_effect(dis.opmap['BUILD_SLICE'], 3), -2)
|
||
|
|
self.assertRaises(ValueError, _opcode.stack_effect, 30000)
|
||
|
|
self.assertRaises(ValueError, _opcode.stack_effect, dis.opmap['BUILD_SLICE'])
|
||
|
|
self.assertRaises(ValueError, _opcode.stack_effect, dis.opmap['POP_TOP'], 0)
|
||
|
|
|
||
|
|
def test_main():
|
||
|
|
run_unittest(OpcodeTests)
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
test_main()
|