a
    Ų
žd-  ć                   @   s~   d Z ddlZddlZddlmZ ddlmZ ddlmZ ddlmZ ddl	m
Z
 G dd	 d	e
jZd
d Zedkrze ”  dS )z%Tests for yapf.format_decision_state.é    N)Śpytree_utils)Śformat_decision_state)Ślogical_line)Śstyle)Śyapf_test_helperc                   @   s(   e Zd Zedd Zdd Zdd ZdS )ŚFormatDecisionStateTestc                 C   s   t  t  ” ” d S )N)r   ZSetGlobalStyleZCreateYapfStyle)Ścls© r	   śm/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/yapftests/format_decision_state_test.pyŚ
setUpClass   s    z"FormatDecisionStateTest.setUpClassc                 C   są  t  d”}t |”}t dt|d ”}| ”  t 	|d”}| 
”  |  d|jj” |  | d”” | dd” |  d|jj” |  | d”” |  | ” ” | dd” |  d|jj” |  | d”” |  | ” ” | dd” |  d|jj” |  | d”” |  | ” ” | dd” |  d	|jj” |  | d”” |  | ” ” | dd” |  d
|jj” |  | d”” |  | ” ” | dd” |  d|jj” |  | d”” |  | ” ” | ” }|  t|t|” d S ©Nz'
      def f(a, b):
        pass
      r   ŚfFTś(Śaś,Śbś)ś:)ŚtextwrapŚdedentr   ŚParseAndUnwrapr   ŚLogicalLineŚ_FilterLineŚCalculateFormattingInformationr   ŚFormatDecisionStateŚMoveStateToNextTokenŚassertEqualŚ
next_tokenŚvalueŚassertFalseŚCanSplitŚAddTokenToStateZ	MustSplitŚ
assertTrueŚCloneŚrepr©ŚselfŚcodeZllinesŚllineŚstateŚcloner	   r	   r
   Ś$testSimpleFunctionDefWithNoSplitting!   sD    

z<FormatDecisionStateTest.testSimpleFunctionDefWithNoSplittingc                 C   s  t  d”}t |”}t dt|d ”}| ”  t 	|d”}| 
”  |  d|jj” |  | d”” | dd” |  d|jj” |  | d”” | dd” |  d|jj” |  | d”” | dd” |  d|jj” |  | d”” | dd” |  d	|jj” |  | d”” | dd” |  d
|jj” |  | d”” | dd” |  d|jj” |  | d”” | ” }|  t|t|” d S r   )r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r    r!   r"   r#   r$   r%   r	   r	   r
   Ś"testSimpleFunctionDefWithSplittingW   s8    

z:FormatDecisionStateTest.testSimpleFunctionDefWithSplittingN)Ś__name__Ś
__module__Ś__qualname__Śclassmethodr   r+   r,   r	   r	   r	   r
   r      s   
6r   c                 C   s   dd | j D S )z4Filter out nonsemantic tokens from the LogicalLines.c                 S   s   g | ]}|j tjvr|qS r	   )Śnamer   ZNONSEMANTIC_TOKENS)Ś.0Śftr	   r	   r
   Ś
<listcomp>   s   ’z_FilterLine.<locals>.<listcomp>)Śtokens)r(   r	   r	   r
   r      s    ’r   Ś__main__)Ś__doc__r   ZunittestZyapf.pytreer   Zyapf.yapflibr   r   r   Z	yapftestsr   ZYAPFTestr   r   r-   Śmainr	   r	   r	   r
   Ś<module>   s   m