OwlCyberSecurity - MANAGER
Edit File: compare.cpython-38.pyc
U ��]TH � @ sX d Z ddlZddlZddlZddlmZ ddlZddlZddlZddl Z ddl m Z ddl Z ddlZddlmZ ddlmZ ddd gZd d � Ze�d �d-d d��Zdd� Zd.dd�Zdd� Ze�d�jZ dd� Z G dd� de �Z G dd� de �Z!G d d � d e!�Z"G d d � d e!�Z#d d!� Z$i Z%e$� d"d � Z&d#d$� Z'd%d&� Z(d'd(� Z)d/d*d�Z*d+d,� Z+dS )0zD Provides a collection of utilities for comparing (image) results. � N)�Path)� TemporaryFile)�ImageComparisonFailure)�cbook� compare_float�compare_images�comparable_formatsc C s t j�| �\}}d|||f S )zU Make a new filename by inserting `purpose` before the file's extension. z%s-%s%s)�os�path�splitext)�fnameZpurpose�baseZext� r �</usr/lib/python3/dist-packages/matplotlib/testing/compare.py�make_test_filename s r z3.0c C s� |dkr|dkrt d��d}|dk r^t| | �}||k r^dddddg}|d�d d � |D ��7 }|dk r�t| | �}| r�|t| � }||k r�dddd d g}|d�d d � |D ��7 }|p�dS )z� Fail if the floating point values are not close enough, with the given message. You can specify a relative tolerance, absolute tolerance, or both. Nz}You haven't specified a 'relTol' relative tolerance or a 'absTol' absolute tolerance function argument. You must specify one.� zExpected: {expected}zActual: {actual}zAbs diff: {absDiff}zAbs tol: {absTol}� c S s g | ]}|j f t� ��qS r ��format�locals��.0�liner r r � <listcomp>9 s z!compare_float.<locals>.<listcomp>zRel diff: {relDiff}zRel tol: {relTol}c S s g | ]}|j f t� ��qS r r r r r r r I s )� ValueError�abs�join)�expected�actualZrelTolZabsTol�msgZabsDiff�templateZrelDiffr r r r " s4 � �c C sj t �� } | d krtd��tj�| d�}zt|�jddd� W n tk rR Y d S X t� |tj �sfd S |S )Nz1Could not find a suitable configuration directoryZ test_cacheT)�parents�exist_ok) �mplZ get_cachedir� RuntimeErrorr r r r �mkdir�IOError�access�W_OK)Zcachedir� cache_dirr r r � get_cache_dirN s r* � c C s� t �� }t| d��"}|�|�}|s$q0|�|� qW 5 Q R X | �d�rb|�tt�d�j �� d�� n&| �d�r�|�tt�d�j �� d�� |� � S )N�rbz.pdf�gszutf-8z.svg�inkscape) �hashlib�md5�open�read�update�endswith�strr# �_get_executable_info�version�encodeZ hexdigest)r Z block_sizer0 �fd�datar r r � get_file_hash\ s � �r; c s � fdd�}|S )Nc s~ � | |�}t j|dt jt jd�}|�� \}}|�� }tj�|�rD|rzdd�|� }|rb|d| 7 }|rr|d| 7 }t |��d S )NT)Zuniversal_newlines�stdout�stderrz Conversion command failed: %s � zStandard output: %s zStandard error: %s ) � subprocess�Popen�PIPEZ communicate�waitr r �existsr r& )�old�newZcmdline�piper< r= Zerrcoder ��cmdr r �convertp s � z1make_external_conversion_command.<locals>.convertr )rH rI r rG r � make_external_conversion_commando s rJ s [^a-zA-Z0-9_@%+=:,./-]c C s$ t | �d kr| S d| �dd� d S )N� 's '"'"')�_find_unsafe_bytes�replace)�br r r �_shlex_quote_bytes� s �rO c @ s e Zd ZdS )�_ConverterErrorN)�__name__� __module__� __qualname__r r r r rP � s rP c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� _Converterc C s d | _ t�| j� d S �N)�_proc�atexit�register�__del__)�selfr r r �__init__� s z_Converter.__init__c C sN | j rJ| j �� | j �� td | j j| j j| j jg�D ] }|�� q6d | _ d S rU )rV �killrB �filter�stdinr< r= �close)rZ �streamr r r rY � s � z_Converter.__del__c C sL t � }| jj�d�}|s t�|�|� |�|�rt|dt|� � �S qdS )z!Read until the prompt is reached.� N) � bytearrayrV r< r2 rP �extendr4 �bytes�len)rZ � terminatorZbuf�cr r r � _read_until� s z_Converter._read_untilN)rQ rR rS r[ rY rh r r r r rT � s rT c @ s e Zd Zdd� ZdS )� _GSConverterc C s | j sVtjt�d�jddgtjtjd�| _ z| �d� W n tk rT t d��Y nX dd� }| j j � d ||� d ||� d � | j j � � | �d �}| �d �}|s�t j�|�s�|r�t|dd � �nd}| j j � d| � t|d | d �t�� d���d S )Nr- z -dNOPAUSEz-sDEVICE=png16m)r^ r< s GSzFailed to start Ghostscriptc S s" t �| ��dd��dd��dd�S )N� \s \\� (s \(� )s \))r �fsencoderM )�namer r r �encode_and_escape� s � � �z0_GSConverter.__call__.<locals>.encode_and_escapes << /OutputFile (s ) >> setpagedevice (s ) run flush s GS� >ra r s pop rM )rV r? r@ r# r6 � executablerA rh rP �OSErrorr^ �write�flushr r rC �intr �decode�sys�getfilesystemencoding)rZ �orig�destro �err�stack� stack_sizer r r �__call__� sJ � ������ ��z_GSConverter.__call__N�rQ rR rS r~ r r r r ri � s ri c @ s e Zd Zdd� ZdS )� _SVGConverterc C s@ | j r| j �� d k r�tj�� }|�dd � tj|d< t� }tj dddgtj tj ||d�| _ || j _ z| � d� W n t k r� td��Y nX ttttj||g��\}}d |ks�d |kr�td d � �||�S | j j�|d | d � | j j�� z| � d� W n< t k �r: | j�d � t| j�� �t� � d���Y nX d S )NZDISPLAYZINKSCAPE_PROFILE_DIRr. z --without-guiz--shell)r^ r< r= �envs >z,Failed to start Inkscape in interactive mode� c S s dd| d|gS )Nr. z-zz --export-pngr )rD rE r r r �<lambda>� s �z(_SVGConverter.__call__.<locals>.<lambda>s --export-png=r rM ) rV Zpollr �environ�copy�pop�devnullr r? r@ rA r= rh rP rr �maprO rm rJ r^ rs rt Z_stderr�seekr r2 rv rw rx )rZ ry rz r� r= Zorig_bZdest_br r r r~ � sN � �� � ��z_SVGConverter.__call__Nr r r r r r� � s r� c C sh zt �d� W n tk r" Y nX t� td< td< zt �d� W n tk rX Y n X t� td< d S )Nr- �pdf�epsr. Zsvg)r# r6 �FileNotFoundErrorri � converterr� r r r r �_update_converter s r� c C s dt �S )z� Return the list of file formats that `.compare_images` can compare on this system. Returns ------- supported_formats : list of str E.g. ``['png', 'pdf', 'svg', 'eps']``. �png)r� )r� r r r r r s c C s2 | � dd�\}}|tkrZd| }ddlm} |� rFddl}|�|� nddlm} ||��|d | d }tj � | �s�t d | ��tj � |�r�t� |�j t� | �j k �r.|r�t� } nd} | dk �r t| �} tj �|�d } tj �| | | �} tj � | ��r t�| |� |S t| | |� | dk �r.t�|| � |S ) at Convert the named file to png; return the name of the created file. If *cache* is True, the result of the conversion is cached in `matplotlib.get_cachedir() + '/test_cache/'`. The caching is based on a hash of the exact contents of the input file. There is no limit on the size of the cache, so it may need to be manually cleared periodically. �.ra z)Don't know how to convert %s files to png)�is_called_from_pytestr N)�SkipTest�_z.pngz'%s' does not exist)�rsplitr� r r� �pytest�skipZnoser� r r rC r& �stat�st_mtimer* r; r r �shutilZcopyfile) �filename�cacher � extension�reasonr� r� r� Znewnamer) Z hash_valueZnew_extZ cached_filer r r rI . s: � rI c C s� | dd� dkr�|dd� dkr�|j \}}}|j \}}} |t|d |d �t|d |d ��t|d |d �t|d |d ��f }||fS )Ni�������r� r� � )�shaperu ) Z actual_path� actual_imageZ expected_path�expected_imageZawZahZadZewZehZedr r r � crop_to_same^ s � �r� c C s< | j |j kr td�| j |j ���t�| | �t�d �� �S )zHCalculate the per-pixel errors, then compute the root mean square error.�9Image sizes do not match expected size: {} actual size {}r� )r� r r �npZsqrt�astype�floatZmean)r� r� r r r � calculate_rmsi s ��r� Fc s� ddl m} tj�|�s$td| ��t�|�jdkr@td| ��| �d�d }tj�| �sft d| ��|dkr�t |d �}t | d �} |� | �}|� |�}|d d �d d �d d �f }|d d �d d �d d �f }t ||| |�\}}t |d �}|dkr�t�||�r�d S |�tj�}|�tj�}t||�} | |k�r.d S t| ||� t| t| �t|�t|�|d�� |�s�ddddddg} d�� fdd�| D ��� � S )a? Compare two "image" files checking differences within a tolerance. The two given filenames may point to files which are convertible to PNG via the `.converter` dictionary. The underlying RMS is calculated with the `.calculate_rms` function. Parameters ---------- expected : str The filename of the expected image. actual : str The filename of the actual image. tol : float The tolerance (a color value difference, where 255 is the maximal difference). The test fails if the average pixel difference is greater than this value. in_decorator : bool Determines the output format. If called from image_comparison decorator, this should be True. (default=False) Returns ------- comparison_result : None or dict or str Return *None* if the images are equal within the given tolerance. If the images differ, the return value depends on *in_decorator*. If *in_decorator* is true, a dict with the following entries is returned: - *rms*: The RMS of the image difference. - *expected*: The filename of the expected image. - *actual*: The filename of the actual image. - *diff_image*: The filename of the difference image. - *tol*: The comparison tolerance. Otherwise, a human-readable multi-line string representation of this information is returned. Examples -------- :: img1 = "./baseline/plot.png" img2 = "./output/plot.png" compare_images(img1, img2, 0.001) r ��_pngz Output image %s does not exist.z Output image file %s is empty.r� ���z!Baseline image %r does not exist.r� FTN� z failed-diff)�rmsr r Zdiff�tolz!Error: Image files did not match.zRMS Value: {rms}zExpected: {expected}zActual: {actual}zDifference: {diff}zTolerance: {tol}r c s g | ]}|j f � ��qS r )r r �Zresultsr r r � s z"compare_images.<locals>.<listcomp>)� matplotlibr� r r rC � Exceptionr� �st_size�splitr& rI Z read_png_intr� r r� Z array_equalr� Zint16r� �save_diff_image�dictr5 r ) r r r� Z in_decoratorr� r� r� r� Z diff_imager� r r r� r r s s\ 1 � ��c C s$ ddl m} |�| �ddd�f }|�|�ddd�f }t||| |�\}}t�|��t�}t�|��t�}|j|jkr�t d� |j|j���t� || �}|d9 }t� |dd��tj �}|j\}} } | dkr�tj|| d ftj d �} || dd�dd�dd�f< | }d|dd�dd�df< |�||� dS ) z� Parameters ---------- expected : str File path of expected image. actual : str File path of actual image. output : str File path to save difference image to. r r� .Nr� r� i� � � )Zdtype)r� r� Zread_pngr� r� Zarrayr� r� r� r r r ZclipZuint8�emptyZ write_png) r r �outputr� r� r� Zabs_diff_imageZ save_image_npZheight�widthZdepthZ with_alphar r r r� � s8 � �� r� )NN)r+ )F),�__doc__rW r/ r �pathlibr �rer� r? rw Ztempfiler Znumpyr� r� r# Z matplotlib.testing.exceptionsr r �__all__r Z deprecatedr r* r; rJ �compile�searchrL rO r� rP �objectrT ri r� r� r� r rI r� r� r r� r r r r �<module> sF + &60 m