File: //opt/alt/python312/lib64/python3.12/test/support/__pycache__/bytecode_helper.cpython-312.opt-2.pyc
�
�h�e� � �� � d dl Z d dlZd dlZd dlmZmZmZ e� Z G d� de j � Z
G d� de j � Z G d� de � Z G d � d
e � Z
G d � d e � Zy)
� N)�compiler_codegen� optimize_cfg�assemble_code_objectc �( � e Zd Z d� Zefd�Zefd�Zy)�BytecodeTestCasec �x � t j � }t j ||�� |j � S )N)�file)�io�StringIO�dis�getvalue)�self�co�ss �C/opt/alt/python312/lib64/python3.12/test/support/bytecode_helper.py�get_disassembly_as_stringz*BytecodeTestCase.get_disassembly_as_string
s' � � �K�K�M�� �������z�z�|�� c �H � | j |t j � t j |� D ]- }|j |k( s�|t
u s|j |k( s�+|c S | j |� }|t
u r|�d|��}n
d}||||fz }| j |� y )Nz not found in bytecode:
z!(%s,%r) not found in bytecode:
%s) �assertInr �opmap�get_instructions�opname� _UNSPECIFIED�argvalr �fail�r �xr r �instr� disassembly�msgs r �assertInBytecodez!BytecodeTestCase.assertInBytecode s� � �O� �
�
�f�c�i�i�(��)�)�!�,� !�E��|�|�v�%��\�)�U�\�\�V�-C� �L� !� �4�4�Q�7� � �\� !�5;�[�I�C�6�C�����5�5�C� � � �#�r c �T � | j |t j � t j |� D ]o }|j |k( s�| j |� }|t u r|�d|��}| j |� �E|j |k( s�Ud}||||fz }| j |� �q y )Nz occurs in bytecode:
z (%s,%r) occurs in bytecode:
%s) r r r r r r r r r r s r �assertNotInBytecodez$BytecodeTestCase.assertNotInBytecode! s� � �6� �
�
�f�c�i�i�(��)�)�!�,� #�E��|�|�v�%� "� <� <�Q� ?� ��\�)�:@�+�N�C��I�I�c�N��\�\�V�+�;�C� ���� =�=�C��I�I�c�N� #r N)�__name__�
__module__� __qualname__r r r! r# � r r r r
s � �;� �
2>�
� 5A� #r r c �� � e Zd Z eej
� Z eej ej z ej z � Z
ej e
� Z G d� d� Z
d� Zd� Zd� Zd� Zy)�CompilationStepTestCasec � � e Zd Zy)� CompilationStepTestCase.LabelN)r$ r% r& r' r r �Labelr+ 5 s � � r r, c � � | j |t � | j |t � | j |� }| j |� }| j t |� t |� � t ||� D ]� \ }}t
|t � r| j ||� �)| j |t � | j |t � t |� t |� kD r|d t |� }| j ||� �� y �N) �assertIsInstance�list�normalize_insts� assertEqual�len�zip�
isinstance�int�tuple)r �actual_� expected_�actual�expected�act�exps r �assertInstructionsMatchz/CompilationStepTestCase.assertInstructionsMatch8 s� � �
� � �g�t�,� � � �i��.��%�%�g�.���'�'� �2�� ����V��c�(�m�4� �F�H�-� '�H�C���#�s�#�� � ��c�*�� � !� !�#�u� -� � !� !�#�u� -��3�x�#�c�(�"��)�3�s�8�n�� � � �S�#� &� 'r c � � d}g }|D ]6 }t || j � r||_ �!|dz
}|j |� �8 |S )Nr � )r5 r, �value�append)r �insts�idx�res�items r �resolveAndRemoveLabelsz.CompilationStepTestCase.resolveAndRemoveLabelsP sL � ������ !�D��$��
�
�+� ��
��q����
�
�4� �
!� �
r c �B � | j |� }g }|D ]� }|^}}}t j j ||� }t || j
� r
|j }n|| j v r|nd }t j | }|j ||g|��� �� |S r. )
rG r r �getr5 r, rA �HAS_ARGr rB )r rC rE rF �opcode�oparg�loc�args r r1 z'CompilationStepTestCase.normalize_insts] s� � � � �+�+�E�2����� ,�D�"&� �F�E�C��Y�Y�]�]�6�6�2�F��%����,��k�k�� %�� � � 5�e�4���Z�Z��'�F� �J�J���*�c�*� +� ,� �
r c � � g }|D ]V }t |� }t j |d }|d }|dd dgdt |� z
z z }|j ||g|��� �X |S )Nr r@ � ���� )r0 r r r3 rB )r rC rE rF �instrK rL rM s r �complete_insts_infoz+CompilationStepTestCase.complete_insts_infoo sv � ���� .�D���:�D��Y�Y�t�A�w�'�F���G�E��q�r�(�b�T�Q��T��]� 3�3�C� �J�J���,��,� -�
.� �
r N)r$ r% r& �setr �hasargrJ �hasjrel�hasjabs�hasexc�
HAS_TARGET�union�HAS_ARG_OR_TARGETr, r> rG r1 rT r' r r r) r) / s\ � ��#�*�*�o�G��S�[�[�3�;�;�.����;�<�J� �
�
�j�1��
�
�'�0 ��$
r r) c � � e Zd Zd� Zy)�CodegenTestCasec �& � t |dd� \ }}|S )Nz
my_file.pyr )r )r �astrC �_s r �
generate_codez CodegenTestCase.generate_code~ s � �#�C� �q�9���q�� r N)r$ r% r& rb r' r r r^ r^ | s � �r r^ c � � e Zd Zdd�Zy)�CfgOptimizationTestCasec �h � | j |� }| j |� }t |||� }||fS r. )r1 rT r )r rC �consts�nlocalss r �
get_optimizedz%CfgOptimizationTestCase.get_optimized� s: � ��$�$�U�+���(�(��/�� �U�F�G�4���f�}� r N)r )r$ r% r& rh r' r r rd rd � s � � r rd c � � e Zd Zd� Zy)�AssemblerTestCasec � � t |||� }|S r. )r )r �filenamerC �metadatar s r �get_code_objectz!AssemblerTestCase.get_code_object� s � �
!�(�E�8�
<��� r N)r$ r% r& rn r' r r rj rj � s � �r rj )�unittestr r
�_testinternalcapir r r �objectr �TestCaser r) r^ rd rj r' r r �<module>rs sm �� M� �
� � R� R��x� �##�x�(�(� ##�JJ�h� /� /� J�Z�-� � � 5� ��/� r