a
    
d
                  
   @   s^   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d Zejddd	ZejdddZ	dS )zUtilities for tests.    Nc                 c   s*   t j}| t _zd V  W |t _n|t _0 d S )N)sysstdout)streamZ
old_stdout r   X/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/yapftests/utils.pystdout_redirector   s
    r   w+bFc	                 c   sL   t j||||d\}	}
tj|	| ||||d}||
fV  |  t|
 dS )z;Context manager creating a new temporary file in text mode.)suffixprefixdirtext)mode	bufferingencodingerrorsnewlineN)tempfilemkstempioopencloseosremove)r   r   r   r   r   r
   r   dirnamer   fdfnamefr   r   r   NamedTempFile,   s    

r   utf-8 c                 c   sR   t | d|||d,\}}|| |  |V  W d    n1 sD0    Y  d S )Nw)r   r   r   r   r
   )r   writeflush)r   contentsr   r   r
   r   r   r   r   r   TempFileContentsE   s    
r%   )	r   r	   NNNNNNF)r   r    N)
__doc__
contextlibr   r   r   r   contextmanagerr   r   r%   r   r   r   r   <module>   s.   
            