
    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 ddl	m
Z
 ddl	mZ  G d	 d
ej                        Zd Zedk(  r ej                           yy)zTests for augment.rename.    )absolute_import)division)print_functionN)rename)scope)
test_utilsc                   H    e Zd Zd Zd Zd Zd Zd Zd Zd Z	d Z
d	 Zd
 Zy)
RenameTestc                 H   d}t        j                  |      }| j                  t        j                  |dd             | j                  |t        j                  d             t        j                  |      }| j                  t        j                  |dd             | j                  |t        j                  d             t        j                  |      }| j                  t        j                  |dd             | j                  |t        j                  |             y )Nz$import aaa.bbb.ccc
aaa.bbb.ccc.foo()aaa.bbbxxx.yyyz$import xxx.yyy.ccc
xxx.yyy.ccc.foo()aaa.bbb.ccczimport xxx.yyy
xxx.yyy.foo()bbbastparse
assertTruer   rename_externalcheckAstsEqualassertFalseselfsrcts      I/home/dcms/DCMS/lib/python3.12/site-packages/pasta/augment/rename_test.pytest_rename_external_in_importz)RenameTest.test_rename_external_in_import   s    
1C		#AOOF**1iCD399%LMN		#AOOF**1mYGH399%DEF		#AV++Aui@A399S>*    c                     d}t        j                  |      }| j                  t        j                  |dd             | j                  |t        j                  d             y )Nz#import aaa.bbb.ccc as ddd
ddd.foo()r   r   z#import xxx.yyy.ccc as ddd
ddd.foo()r   r   r   r   r   r   r   s      r   *test_rename_external_in_import_with_asnamez5RenameTest.test_rename_external_in_import_with_asname-   sI    
0C		#AOOF**1iCD399%KLMr   c                     d}t        j                  |      }| j                  t        j                  |dd             | j                  |t        j                  d             y )Nz import aaa, aaa.bbb, aaa.bbb.cccr   r   z import aaa, xxx.yyy, xxx.yyy.cccr   r   s      r   /test_rename_external_in_import_multiple_aliasesz:RenameTest.test_rename_external_in_import_multiple_aliases3   sI    
,C		#AOOF**1iCD399%GHIr   c                 H   d}t        j                  |      }| j                  t        j                  |dd             | j                  |t        j                  d             t        j                  |      }| j                  t        j                  |dd             | j                  |t        j                  d             t        j                  |      }| j                  t        j                  |dd             | j                  |t        j                  |             y )Nz%from aaa.bbb.ccc import ddd
ddd.foo()r   r   z%from xxx.yyy.ccc import ddd
ddd.foo()r   z!from xxx.yyy import ddd
ddd.foo()r   r   r   s      r   "test_rename_external_in_importfromz-RenameTest.test_rename_external_in_importfrom9   s    
2C		#AOOF**1iCD399%MNO		#AOOF**1mYGH399%IJK		#AV++Aui@A399S>*r   c                     d}t        j                  |      }| j                  t        j                  |dd             | j                  |t        j                  d             y )Nz!from aaa.bbb import ccc
ccc.foo()r   r   zfrom xxx import yyy
yyy.foo()r   r   s      r   (test_rename_external_in_importfrom_aliasz3RenameTest.test_rename_external_in_importfrom_aliasG   sI    
.C		#AOOF**1mYGH399%EFGr   c                     d}t        j                  |      }| j                  t        j                  |dd             | j                  |t        j                  d             y )Nz(from aaa.bbb import ccc as abc
abc.foo()r   r   z$from xxx import yyy as abc
abc.foo()r   r   s      r   4test_rename_external_in_importfrom_alias_with_asnamez?RenameTest.test_rename_external_in_importfrom_alias_with_asnameM   sI    
5C		#AOOF**1mYGH399%LMNr   c                     d}t        j                  |      }t        j                  |      }| j	                  t        j                  ||dd             | j                  |t        j                  d             y )N	aaa.bbb()aaaxxxz	xxx.bbb())r   r   r   analyzer   r   _rename_readsr   r   r   r   scs       r   test_rename_reads_namez!RenameTest.test_rename_reads_nameS   sX    
C		#A	q	BOOF((Qu=>399[12r   c                     d}t        j                  |      }t        j                  |      }t	        j
                  ||dd       | j                  |t        j                  d             y )Nr*   r+   r   zxxx.yyy.bbb()r   r   r   r-   r   r.   r   r/   s       r   #test_rename_reads_name_as_attributez.RenameTest.test_rename_reads_name_as_attributeZ   sO    
C		#A	q	B
Qy1399_56r   c                     d}t        j                  |      }t        j                  |      }t	        j
                  ||dd       | j                  |t        j                  d             y )Naaa.bbb.ccc()r   r   zxxx.yyy.ccc()r3   r/   s       r   test_rename_reads_attributez&RenameTest.test_rename_reads_attributea   sO    
C		#A	q	B
Q	95399_56r   c                    d}t        j                  |      }t        j                  |      }t	        j
                  ||dd       t	        j
                  ||dd       | j                  |t        j                  |             y )Nr6   zaaa.bbb.ccc.dddr   zbbb.aaar3   r/   s       r   test_rename_reads_noopz!RenameTest.test_rename_reads_nooph   sc    
C		#A	q	B
Q 19=
Q	95399S>*r   N)__name__
__module____qualname__r   r    r"   r$   r&   r(   r1   r4   r7   r9    r   r   r
   r
      s8    +NJ+HO377+r   r
   c                  ~    t        j                         } | j                  t        j                  t                     | S )N)unittest	TestSuiteaddTests	makeSuiter
   )results    r   suiterD   q   s,    &//($$Z01	-r   __main__)__doc__
__future__r   r   r   r   r?   pasta.augmentr   
pasta.baser   r   TestCaser
   rD   r:   mainr=   r   r   <module>rL      sV     '  % 
     !Q+$$ Q+h
 z(--/ r   