
    1Vh                         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ej                        Z
d	 Zed
k(  r ej                          yy)z7Tests for google3.third_party.py.pasta.base.test_utils.    )absolute_import)division)print_functionN)
test_utilsc                   *    e Zd Zd Zd Zd Zd Zd Zy)CheckAstEqualityTestc                 V    d}t        j                  |      }| j                  ||       y )N astparsecheckAstsEqualselfsrcts      J/home/dcms/DCMS/lib/python3.12/site-packages/pasta/base/test_utils_test.py
test_emptyzCheckAstEqualityTest.test_empty   s$    
C		#A1    c                 V    d}t        j                  |      }| j                  ||       y )NzX = 1
r   r   s      r   test_one_globalz$CheckAstEqualityTest.test_one_global"   s$    
C		#A1r   c                 V    d}t        j                  |      }| j                  ||       y )NzX = 1
Y = 2
r   r   s      r   test_two_globalsz%CheckAstEqualityTest.test_two_globals'   s$    
C		#A1r   c                     d}|dz   }t        j                  |      }t        j                  |      }| j                  t              5  | j	                  ||       d d d        y # 1 sw Y   y xY w)NzX = 1
def Foo():
  return None
zY = 2
)r   r   assertRaisesAssertionErrorr   )r   src1src2t1t2s        r   test_different_number_of_nodesz3CheckAstEqualityTest.test_different_number_of_nodes,   s^    /D)D	4B	4B			>	* "
"b!" " "s   A##A,c                 V    d}t        j                  |      }| j                  ||       y )Nzdef foo(x):
  return x + 1
r   )r   coder   s      r   test_simple_function_defz-CheckAstEqualityTest.test_simple_function_def4   s&    D		$A1r   N)__name__
__module____qualname__r   r   r   r!   r$    r   r   r   r      s    


"r   r   c                  ~    t        j                         } | j                  t        j                  t                     | S )N)unittest	TestSuiteaddTests	makeSuiter   )results    r   suiter/   ;   s-    &//($$%9:;	-r   __main__)__doc__
__future__r   r   r   r   r*   
pasta.baser   TestCaser   r/   r%   mainr(   r   r   <module>r6      sN    = '  % 
  !:.. @ z(--/ r   