FilesExpand file tree

python-stdlib

/

bytecode_helper.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

41 lines (35 loc) · 1.57 KB

python-stdlib

/

bytecode_helper.py

Copy path

Top

File metadata and controls

• Code
• Blame

41 lines (35 loc) · 1.57 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

"""bytecode_helper - support tools for testing correct bytecode generation"""

import unittest

import dis

import io

_UNSPECIFIED = object()

class BytecodeTestCase(unittest.TestCase):

"""Custom assertion methods for inspecting bytecode."""

def get_disassembly_as_string(self, co):

s = io.StringIO()

dis.dis(co, file=s)

return s.getvalue()

def assertInBytecode(self, x, opname, argval=_UNSPECIFIED):

"""Returns instr if op is found, otherwise throws AssertionError"""

for instr in dis.get_instructions(x):

if instr.opname == opname:

if argval is _UNSPECIFIED or instr.argval == argval:

return instr

disassembly = self.get_disassembly_as_string(x)

if argval is _UNSPECIFIED:

msg = '%s not found in bytecode:\n%s' % (opname, disassembly)

else:

msg = '(%s,%r) not found in bytecode:\n%s'

msg = msg % (opname, argval, disassembly)

self.fail(msg)

def assertNotInBytecode(self, x, opname, argval=_UNSPECIFIED):

"""Throws AssertionError if op is found"""

for instr in dis.get_instructions(x):

if instr.opname == opname:

disassembly = self.get_disassembly_as_string(co)

if opargval is _UNSPECIFIED:

msg = '%s occurs in bytecode:\n%s' % (opname, disassembly)

elif instr.argval == argval:

msg = '(%s,%r) occurs in bytecode:\n%s'

msg = msg % (opname, argval, disassembly)

self.fail(msg)