File: //opt/cppython/lib/python3.8/test/__pycache__/bytecode_helper.cpython-38.pyc
U
-D�g@ � @ s8 d Z ddlZddlZddlZe� ZG dd� dej�ZdS )zGbytecode_helper - support tools for testing correct bytecode generation� Nc @ s0 e Zd ZdZdd� Zefdd�Zefdd�ZdS ) �BytecodeTestCasez1Custom assertion methods for inspecting bytecode.c C s t �� }tj||d� |�� S )N)�file)�io�StringIO�dis�getvalue)�self�co�s� r �3/opt/cppython/lib/python3.8/test/bytecode_helper.py�get_disassembly_as_string s z*BytecodeTestCase.get_disassembly_as_stringc C st t �|�D ](}|j|kr
|tks*|j|kr
| S q
| �|�}|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)r �get_instructions�opname� _UNSPECIFIED�argvalr
�fail�r �xr r ZinstrZ disassembly�msgr r r �assertInBytecode s
z!BytecodeTestCase.assertInBytecodec C sd t �|�D ]T}|j|kr
| �|�}|tkr8d||f }n |j|krTd}||||f }| �|� q
dS )z$Throws AssertionError if op is foundz%s occurs in bytecode:
%sz (%s,%r) occurs in bytecode:
%sN)r r r r
r r r r r r r �assertNotInBytecode s
z$BytecodeTestCase.assertNotInBytecodeN)�__name__�
__module__� __qualname__�__doc__r
r r r r r r r r s r )r Zunittestr r �objectr ZTestCaser r r r r �<module> s