a
    ,=ic                     @   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   k/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/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   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   test_two_globals'   s    
z%CheckAstEqualityTest.test_two_globalsc                 C   sZ   d}|d }t |}t |}| t | || W d    n1 sL0    Y  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   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    