o
    2g                     @  sz   d dl mZ d dlZd dlmZmZmZmZmZm	Z	m
Z
 d dlmZ d dlmZ dd
dZdddZG dd dejZdS )    )annotationsN)Assign
BasicBlockIntegerIntOpOpRegisterUnreachablegenerate_names_for_ir)int_rprimitivenamestrreturnr   c                 C  s   t tdddS )NfooT)is_arg)r   r   )r    r   N/home/garg/my-data/venv/lib/python3.10/site-packages/mypyc/test/test_pprint.pyregister
   s   r   opslist[Op]r   c                 C  s   t  }|j|  |S N)r   r   extend)r   blockr   r   r   
make_block   s   r   c                   @  s4   e Zd ZdddZdddZdddZdd	d
ZdS )TestGenerateNamesr   Nonec                 C  s   t g g i ks	J d S r   r
   )selfr   r   r   
test_empty   s   zTestGenerateNames.test_emptyc                 C  s$   t d}t|gg |diksJ d S )Nr   )r   r   )r   regr   r   r   test_arg   s   zTestGenerateNames.test_argc                 C  s`   t d}t d}tt||tj}tt||tj}t||t g}tg |g|d|diks.J d S )N      r0r1)r   r   r   ADDr   r	   r   )r   n1n2op1op2r   r   r   r   test_int_op   s    zTestGenerateNames.test_int_opc                 C  sN   t d}td}t||}t||}t||g}t|g|g|diks%J d S )Nr   r!   )r   r   r   r   r   )r   r   nr(   r)   r   r   r   r   test_assign$   s   

zTestGenerateNames.test_assignN)r   r   )__name__
__module____qualname__r   r    r*   r,   r   r   r   r   r      s
    


r   )r   r   r   r   )r   r   r   r   )
__future__r   unittestmypyc.ir.opsr   r   r   r   r   r   r	   mypyc.ir.pprintr   mypyc.ir.rtypesr   r   r   TestCaser   r   r   r   r   <module>   s    $

