Gecko [ v99win.today ]

Name	Size	Permission
__pycache__	[ DIR ]	drwxr-xr-x
__init__.py	69.69 KB	-rw-r--r--
bytecode_helper.py	1.61 KB	-rw-r--r--
hashlib_helper.py	1.86 KB	-rw-r--r--
import_helper.py	7.55 KB	-rw-r--r--
interpreters.py	5.69 KB	-rw-r--r--
logging_helper.py	916 B	-rw-r--r--
os_helper.py	19.5 KB	-rw-r--r--
script_helper.py	11.14 KB	-rw-r--r--
socket_helper.py	11.21 KB	-rw-r--r--
testresult.py	5.86 KB	-rw-r--r--
threading_helper.py	6.49 KB	-rw-r--r--
warnings_helper.py	6.44 KB	-rw-r--r--

Code Editor : bytecode_helper.py

"""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 opname 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 opname is found"""
        for instr in dis.get_instructions(x):
            if instr.opname == opname:
                disassembly = self.get_disassembly_as_string(x)
                if argval is _UNSPECIFIED:
                    msg = '%s occurs in bytecode:\n%s' % (opname, disassembly)
                    self.fail(msg)
                elif instr.argval == argval:
                    msg = '(%s,%r) occurs in bytecode:\n%s'
                    msg = msg % (opname, argval, disassembly)
                    self.fail(msg)

${this.title}

Code Editor : bytecode_helper.py