B
    ­»ˆdÙ  ã               @   sr   d Z ddlmZ ddlmZ ddlmZ ddlZddlZddlmZ G dd„ dej	ƒZ
d	d
„ Zedkrne ¡  dS )z7Tests for google3.third_party.py.pasta.base.test_utils.é    )Úabsolute_import)Údivision)Úprint_functionN)Ú
test_utilsc               @   s4   e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd	d
„ ZdS )ÚCheckAstEqualityTestc             C   s   d}t  |¡}|  ||¡ d S )NÚ )ÚastÚparseÚcheckAstsEqual)ÚselfÚsrcÚt© r   úL/var/www/html/venv/lib/python3.7/site-packages/pasta/base/test_utils_test.pyÚ
test_empty   s    
zCheckAstEqualityTest.test_emptyc             C   s   d}t  |¡}|  ||¡ d S )NzX = 1
)r   r	   r
   )r   r   r   r   r   r   Útest_one_global"   s    
z$CheckAstEqualityTest.test_one_globalc             C   s   d}t  |¡}|  ||¡ d S )NzX = 1
Y = 2
)r   r	   r
   )r   r   r   r   r   r   Útest_two_globals'   s    
z%CheckAstEqualityTest.test_two_globalsc          	   C   sF   d}|d }t  |¡}t  |¡}|  t¡ |  ||¡ W d Q R X d S )NzX = 1
def Foo():
  return None
zY = 2
)r   r	   ÚassertRaisesÚAssertionErrorr
   )r   Zsrc1Zsrc2Út1Út2r   r   r   Útest_different_number_of_nodes,   s    

z3CheckAstEqualityTest.test_different_number_of_nodesc             C   s   d}t  |¡}|  ||¡ d S )Nzdef foo(x):
  return x + 1
)r   r	   r
   )r   Úcoder   r   r   r   Útest_simple_function_def4   s    
z-CheckAstEqualityTest.test_simple_function_defN)Ú__name__Ú
__module__Ú__qualname__r   r   r   r   r   r   r   r   r   r      s
   r   c              C   s   t  ¡ } |  t  t¡¡ | S )N)ÚunittestZ	TestSuiteZaddTestsZ	makeSuiter   )Úresultr   r   r   Úsuite;   s    r   Ú__main__)Ú__doc__Ú
__future__r   r   r   r   r   Z
pasta.baser   ZTestCaser   r   r   Úmainr   r   r   r   Ú<module>   s    