B

"��g�!�@s�dZd
dl
Z
d
dlZd
dlZd
dlZd
dlZd
dlmZ
d
dlm	Z	
d
dl
mZmZ
e�
Gdd�dej
�Zedkr|e��
dS)	z4Tests for the pindent script in the Tools directory.�N)
�support)
�assert_python_ok)�
scriptsdir�skip_if_missingc@sreZ
dZej�ed
�Zdd�Zdd�Z	dd�Z
dd	�Zd
d�Zdd
�Z
dd�Zdd�Zdd�Zdd�Zdd�ZdS)�PindentTestsz
pindent.pyc
Cs@t|
�
�.}t|�
�}|�
|��|���
WdQRXWdQRXdS)
N)�open�assertEqual�	readlines)�selfZfn1Zfn2�f1�f2�r
�8/usr/local/lib/python3.7/test/test_tools/test_pindent.py�assertFileEquals

zPindentTests.assertFileEqualc	GsJtj
tj|jf|tjtjd
d��}|�|
�
\}}WdQRX|�|�

|S)NT)�stdin�stdoutZuniversal_newlines)�
subprocess�Popen�sys�
executable�script�PIPEZcommunicateZassertIsNone)r
�source�args�proc�out�errr
r
r�pindents








zPindentTests.pindentcCsd
�dd�|
�
�D�
�
d
S)N�

c
ss|]}
|
��V
qdS)
N)
�lstrip)�.0�liner
r
r�	<genexpr>!sz+PindentTests.lstriplines.<locals>.<genexpr>)�join�
splitlines)r
�datar
r
r�lstriplines s
zPindentTests.lstriplinesc

CsPd|_t
����6}
tj�|
d
�}t|j�
�}|��}WdQRXt|d��}|�	|�

WdQRXt
|jd|�\}}}|�|d�
|�|d�
|d}|�tj�
|�
�

t|�
�}|�|��|�
WdQRXt|�
�}|��}	WdQRXt|	d
d�
|�|�|	d�|�
|�|�|d�|	�
t
|jd|�\}}}|�|d�
|�|d�
t|�
�}|�|��|	�
WdQRXt|�
�}|�|��|�
WdQRX|�|�
}
t|d��}|�	|
�

WdQRXt
|jd|�\}}}|�|d�
|�|d�
t|�
�}|�|��|
�
WdQRXt|�
�}|��}WdQRXt|d
d�
|�|�|
d�|�
WdQRXdS)	Nz_test.py�
wz-d��
~�execz-cz-r)ZmaxDiffrZtemp_dir�os�pathr#rr�read�writerrZ
assertTrue�exists�compilerr&)r
Z	directoryZ	data_path�
f�closedZrcrrZbackup�clean�brokenZindentedr
r
r�
test_selftest#sL









































zPindentTests.test_selftestc	CsP|�|�
|
d
�|�
|�|�
|d�|
�
|�|�
}|�|�
|dddd�|�
dS)Nz-cz-dz-rz-ez-s�
4)rrr&)r
r3r2r4r
r
r�pindent_testNs




zPindentTests.pindent_testc
Cs$t�
d
�
}
t�
d�
}|�|
|�
dS)Na:            if a:
                pass

            if a:
                pass
            else:
                pass

            if a:
                pass
            elif:
                pass
            else:
                pass

            while a:
                break

            while a:
                break
            else:
                pass

            for i in a:
                break

            for i in a:
                break
            else:
                pass

            try:
                pass
            finally:
                pass

            try:
                pass
            except TypeError:
                pass
            except ValueError:
                pass
            else:
                pass

            try:
                pass
            except TypeError:
                pass
            except ValueError:
                pass
            finally:
                pass

            with a:
                pass

            class A:
                pass

            def f():
                pass
            a`            if a:
                pass
            # end if

            if a:
                pass
            else:
                pass
            # end if

            if a:
                pass
            elif:
                pass
            else:
                pass
            # end if

            while a:
                break
            # end while

            while a:
                break
            else:
                pass
            # end while

            for i in a:
                break
            # end for

            for i in a:
                break
            else:
                pass
            # end for

            try:
                pass
            finally:
                pass
            # end try

            try:
                pass
            except TypeError:
                pass
            except ValueError:
                pass
            else:
                pass
            # end try

            try:
                pass
            except TypeError:
                pass
            except ValueError:
                pass
            finally:
                pass
            # end try

            with a:
                pass
            # end with

            class A:
                pass
            # end class A

            def f():
                pass
            # end def f
            )�textwrap�dedentr7)r
r3r2r
r
r�test_statementsTs

?L
zPindentTests.test_statementsc
Cs$t�
d
�
}
t�
d�
}|�|
|�
dS)Nz�            def foobar(a, b):
                if a == b:
                    a = a+1
                elif a < b:
                    b = b-1
                    if b > a: a = a-1
                else:
                    print 'oops!'
            aJ
            def foobar(a, b):
                if a == b:
                    a = a+1
                elif a < b:
                    b = b-1
                    if b > a: a = a-1
                    # end if
                else:
                    print 'oops!'
                # end if
            # end def foobar
            )r8r9r7)r
r3r2r
r
r�test_multilevel�s

	

zPindentTests.test_multilevelc
	Cs�t�
d
�
}
t�
d�
}|�|�|
d�|�
|�|�|d�|
�
|�|�
}|�|�|dddd�|�
t�
d	�
}
t�
d
�
}|�|�|
d�|�
|�|�|d�|
�
|�|�
}|�|�|d�|�
dS)Nz\            if a:
                     if b:
                              pass
            z�            if a:
                     if b:
                              pass
                     # end if
            # end if
            z-cz-dz-rz-ez-s�
9zD            if a:
            	if b:
            		pass
            zo            if a:
            	if b:
            		pass
            	# end if
            # end if
            )r8r9rrr&)r
r3r2r4r
r
r�test_preserve_indents�s 













z"PindentTests.test_preserve_indentsc
Cs@t�
d
�
}
t�
d�
}|�|�|
d�|�
|�|�|d�|
�
dS)Nz�            class\
            \
             A:
               def            \
            f:
                  pass
            z�            class\
            \
             A:
               def            \
            f:
                  pass
               # end def f
            # end class A
            z-cz-d)r8r9rr)r
r3r2r
r
r�test_escaped_newline!
s




z!PindentTests.test_escaped_newlinec
Cs$t�
d
�
}
t�
d�
}|�|
|�
dS)Nz4            if a:

                pass
            zI            if a:

                pass
            # end if
            )r8r9r7)r
r3r2r
r
r�test_empty_line9
s



zPindentTests.test_empty_linec
Cs$t�
d
�
}
t�
d�
}|�|
|�
dS)Nz#            if a: pass
            z8            if a: pass
            # end if
            )r8r9r7)r
r3r2r
r
r�test_onelineG
s



zPindentTests.test_onelineN)�__name__�
__module__�__qualname__r+r,r#rrrrr&r5r7r:r;r=r>r?r@r
r
r
rrs
	+"r�__main__)�__doc__r+rZunittestrr8ZtestrZtest.support.script_helperrZtest.test_toolsrrZTestCaserrA�mainr
r
r
r�<module>
s





D