�
�(�g� � �� � d Z ddlZddlZddlZddlmZmZmZ e� Z G d� dej � Z G d� dej � Z G d� d e � Z
G d
� d e � Z G d � d
e � Zy)zGbytecode_helper - support tools for testing correct bytecode generation� N)�compiler_codegen� optimize_cfg�assemble_code_objectc �* � e Zd ZdZd� Zefd�Zefd�Zy)�BytecodeTestCasez1Custom assertion methods for inspecting bytecode.c �x � t j � }t j ||�� |j � S )N)�file)�io�StringIO�dis�getvalue)�self�co�ss �3/usr/lib/python3.12/test/support/bytecode_helper.py�get_disassembly_as_stringz*BytecodeTestCase.get_disassembly_as_string
s' � � �K�K�M�� �������z�z�|�� c �F � | 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)zAReturns instr if opname is found, otherwise throws AssertionErrorz not found in bytecode:
z!(%s,%r) not found in bytecode:
%sN) �assertInr �opmap�get_instructions�opname� _UNSPECIFIED�argvalr �fail�r �xr r �instr� disassembly�msgs r �assertInBytecodez!BytecodeTestCase.assertInBytecode s� � � �
�
�f�c�i�i�(��)�)�!�,� !�E��|�|�v�%��\�)�U�\�\�V�-C� �L� !� �4�4�Q�7� � �\� !�5;�[�I�C�6�C�����5�5�C� � � �#�r c �R � | 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)z(Throws AssertionError if opname is foundz occurs in bytecode:
z (%s,%r) occurs in bytecode:
%sN) r r r r r r r r r r s r �assertNotInBytecodez$BytecodeTestCase.assertNotInBytecode! s� � � �
�
�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__�__doc__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 ]T }t || j t f� sJ �t || j � r||_ �?|dz
}|j |� �V |S )Nr � )r6 r- r8 �value�append)r �insts�idx�res�items r �resolveAndRemoveLabelsz.CompilationStepTestCase.resolveAndRemoveLabelsP se � ������ !�D� �d�T�Z�Z��$7�8� 8�8��$��
�
�+� ��
��q����
�
�4� �
!� �
r c �d � | j |� }g }|D ]� }t |t � sJ �|^}}}t j j ||� }t || j � r
|j }n|| j v r|nd}t j | }|j ||g|��� �� |S )zN Map labels to instruction index.
Map opcodes to opnames.
N) rH r6 r8 r r �getr- rB �HAS_ARGr rC )r rD rF rG �opcode�oparg�loc�args r r2 z'CompilationStepTestCase.normalize_insts] s� � � �+�+�E�2����� ,�D� �d�E�*� *�*�"&� �F�E�C��Y�Y�]�]�6�6�2�F��%����,��k�k�� %�� � � 5�e�4���Z�Z��'�F� �J�J���*�c�*� +� ,� �
r c �� � g }|D ]h }t |t � sJ �t |� }t j |d }|d }|dd dgdt |� z
z z }|j
||g|��� �j |S )Nr rA � ���� )r6 r8 r1 r r r4 rC )r rD rF rG �instrL rM rN s r �complete_insts_infoz+CompilationStepTestCase.complete_insts_infoo s� � ���� .�D� �d�E�*� *�*���:�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 �hasargrK �hasjrel�hasjabs�hasexc�
HAS_TARGET�union�HAS_ARG_OR_TARGETr- r? rH r2 rU 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 �astrD �_s r �
generate_codez CodegenTestCase.generate_code~ s � �#�C� �q�9���q�� r N)r$ r% r& rc r( r r r_ r_ | s � �r r_ c � � e Zd Zdd�Zy)�CfgOptimizationTestCasec �h � | j |� }| j |� }t |||� }||fS r/ )r2 rU r )r rD �consts�nlocalss r �
get_optimizedz%CfgOptimizationTestCase.get_optimized� s: � ��$�$�U�+���(�(��/�� �U�F�G�4���f�}� r N)r )r$ r% r& ri r( r r re re � s � � r re c � � e Zd Zd� Zy)�AssemblerTestCasec � � t |||� }|S r/ )r )r �filenamerD �metadatar s r �get_code_objectz!AssemblerTestCase.get_code_object� s � �
!�(�E�8�
<��� r N)r$ r% r& ro r( r r rk rk � s � �r rk )r' �unittestr r
�_testinternalcapir r r �objectr �TestCaser r* r_ re rk r( r r �<module>rt sm �� M� �
� � R� R��x� �##�x�(�(� ##�JJ�h� /� /� J�Z�-� � � 5� ��/� r |