
    ?d                         d 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	 G d d	ej                        Z
ed
k(  r ej                          yy)zTests for yapf.format_token.    N)pytree)token)format_tokenc                       e Zd Zd Zd Zd Zy)"TabbedContinuationAlignPaddingTestc                     d}t        j                  d|d      }| j                  |d       t        j                  d|d      }| j                  |d       t        j                  d|d      }| j                  |d       y )NSPACEr       z     z     r   _TabbedContinuationAlignPaddingassertEqualselfalign_stylepads      =/usr/lib/python3/dist-packages/yapftests/format_token_test.py	testSpacez,TabbedContinuationAlignPaddingTest.testSpace   ss    K

6
6q+q
ICS"

6
6q+q
ICS'"

6
6q+q
ICS'"    c                     d}t        j                  d|d      }| j                  |d       t        j                  d|d      }| j                  |d       t        j                  d|d      }| j                  |d       y )	NFIXEDr      r   r
   	r   		r   r   s      r   	testFixedz,TabbedContinuationAlignPaddingTest.testFixed&   ss    K

6
6q+q
ICS"

6
6q+q
ICS$

6
6q+q
ICS(#r   c                 P   d}t        j                  d|d      }| j                  |d       t        j                  d|d      }| j                  |d       t        j                  d|d      }| j                  |d       t        j                  d|d      }| j                  |d       y )	NzVALIGN-RIGHTr   r   r   r
   r   r   r   r   r   s      r   testVAlignRightz2TabbedContinuationAlignPaddingTest.testVAlignRight2   s     K

6
6q+q
ICS"

6
6q+q
ICS$

6
6q+q
ICS$

6
6q+q
ICS(#r   N)__name__
__module____qualname__r   r   r    r   r   r   r      s    
#
$$r   r   c                       e Zd Zd Zd Zy)FormatTokenTestc                    t        j                  t        j                  t        j
                  d      d      }| j                  dt        |             | j                  |j                         t        j                  t        j                  t        j                  d      d      }| j                  dt        |             | j                  |j                         y )Nz'hello world'STRINGzTFormatToken(name=DOCSTRING, value='hello world', column=0, lineno=0, splitpenalty=0)z# A commentCOMMENTzPFormatToken(name=COMMENT, value=# A comment, column=0, lineno=0, splitpenalty=0))r   FormatTokenr   Leafr   r&   r   str
assertTrue	is_stringr'   
is_commentr   toks     r   
testSimplezFormatTokenTest.testSimpleD   s    

"
"ELL/2H>C	$%(X/ 	OOCMM"

"
"EMM=19>C	$%(X/ 	OOCNN#r   c                 P   t        j                  t        j                  t        j
                  d      d      }| j                  |j                         t        j                  t        j                  t        j
                  d      d      }| j                  |j                         y )Nz"""hello"""r&   zr"""hello""")r   r(   r   r)   r   r&   r+   is_multiline_stringr.   s     r   testIsMultilineStringz%FormatTokenTest.testIsMultilineStringS   sn    

"
"ELL-0(<COOC++,

"
"ELL.18=COOC++,r   N)r   r    r!   r0   r3   r"   r   r   r$   r$   B   s    $-r   r$   __main__)__doc__unittestlib2to3r   lib2to3.pgen2r   yapf.yapflibr   TestCaser   r$   r   mainr"   r   r   <module>r<      sU    #    %'$):): '$T-h'' -6 z(--/ r   