
    ?d
                         d Z ddlZddlZddlZddlZddlZej                  d        Zej                  	 	 	 	 	 	 	 	 	 dd       Zej                  	 	 	 dd       Z	y)zUtilities for tests.    Nc              #      K   t         j                  }| t         _        	 d  |t         _        y # |t         _        w xY ww)N)sysstdout)stream
old_stdouts     1/usr/lib/python3/dist-packages/yapftests/utils.pystdout_redirectorr	      s,     zz*#*	CJCJs   ?/ ?<?c	              #     K   t         j                  dk  r|d}|d}t        j                  ||||      \  }	}
t	        j
                  |	| ||||      }||
f |j                          t        j                  |
       yw)z;Context manager creating a new temporary file in text mode.)      N tmp)suffixprefixdirtext)mode	bufferingencodingerrorsnewline)	r   version_infotempfilemkstempioopencloseosremove)r   r   r   r   r   r   r   dirnamer   fdfnamefs               r   NamedTempFiler$   ,   s      	~f~f  Fd<+2ugg! 	
5.'')))Es   A?Bc              #      K   t        | d|||      5 \  }}|j                  |       |j                          | d d d        y # 1 sw Y   y xY ww)Nw)r    r   r   r   r   )r$   writeflush)r    contentsr   r   r   r#   r"   s          r   TempFileContentsr*   J   sV       
 #EGGHGGI
K  s   A)A	AA
A)	zw+bNNNNNNF)zutf-8r   N)
__doc__
contextlibr   r   r   r   contextmanagerr	   r$   r*        r   <module>r1      s      	 	 
   &  :  & 	 r0   