OwlCyberSecurity - MANAGER
Edit File: test_png.cpython-38.pyc
U ��]� � @ s� d dl mZ d dlZd dlZd dlZd dlZd dlmZ d dl m Z d dl m Z d dlZejdkZedgdgdd �d d � �Zd d � Zdd� Zdd� ZdS )� )�BytesION)�image_comparison)�pyplotZwin32�pngsuiteZpngg� ��Q��?)�baseline_images� extensionsZtolc C s� t j�t j�t�dd�} tt�t j�| d���}tj t |�dfd�}t |�D ]B\}}t� |�}d }|j dkrrtj}tj|||d ddg|d� qLt�� j�d � t�� �dt |�� d S ) Nr r z basn*.png� )Zfigsize� r )Zextent�cmapz#ddffff)�os�path�join�dirname�__file__�sorted�globZiglob�pltZfigure�len� enumerate�imread�ndim�cmZgrayZimshowZgcaZpatchZ set_facecolorZset_xlim)r �filesZfig�i�fname�datar � r �;/usr/lib/python3/dist-packages/matplotlib/tests/test_png.py� test_pngsuite s � r c C sR ddl m} | �tj�tj�t�d��}|jt j ks8t �t � |� � �dksNt �d S )Nr )�_pngz#baseline_images/test_png/uint16.pngi ��)� matplotlibr Z read_png_intr r r r r Zdtype�npZuint16�AssertionError�sumZflatten)r Zimgr r r �test_imread_png_uint16% s �r$ c C s� | � d�}t|�d��}t|�d��}t�|� t|d��}|�� }W 5 Q R X t|d��}|�|d d� � W 5 Q R X t� t �� t� |� W 5 Q R X d S )NZtestztest.pngztest_truncated.png�rb�wb� ) �mkdir�strr r �savefig�open�read�write�pytest�raises� Exceptionr )Ztmpdir�dr Zfname_tZfinZbufZfoutr r r �test_truncated_file. s r2 c C sV t � } t�| � | �d� t | �d��}|�d� t�t�� t�|� W 5 Q R X d S )Nr r' ) r r r* �seekr, r. r/ r0 r )�bZb2r r r �test_truncated_buffer< s r5 )�ior r r Znumpyr! r. Z matplotlib.testing.decoratorsr r r r Z matplotlib.cmr �sys�platformZon_winr r$ r2 r5 r r r r �<module> s �