B u9a@@s8dZddlZddlZddlZeZGdddejZdS)zGbytecode_helper - support tools for testing correct bytecode generationNc@s0eZdZdZddZefddZefddZdS) BytecodeTestCasez1Custom assertion methods for inspecting bytecode.cCst}tj||d|S)N)file)ioStringIOdisgetvalue)selfcosr %/usr/lib/python3.7/bytecode_helper.pyget_disassembly_as_string sz*BytecodeTestCase.get_disassembly_as_stringcCstx2t|D]$}|j|kr |tks,|j|kr |Sq W||}|tkrTd||f}nd}||||f}||dS)z=Returns instr if op is found, otherwise throws AssertionErrorz%s not found in bytecode: %sz!(%s,%r) not found in bytecode: %sN)rget_instructionsopname _UNSPECIFIEDargvalr fail)rxrrinstr disassemblymsgr r r assertInBytecodes  z!BytecodeTestCase.assertInBytecodecCshxbt|D]T}|j|kr ||}|tkr:d||f}n|j|krVd}||||f}||q WdS)z$Throws AssertionError if op is foundz%s occurs in bytecode: %sz(%s,%r) occurs in bytecode: %sN)rrrr rrr)rrrrrrrr r r assertNotInBytecodes   z$BytecodeTestCase.assertNotInBytecodeN)__name__ __module__ __qualname____doc__r rrrr r r r r s r)rZunittestrrobjectrZTestCaserr r r r s