OwlCyberSecurity - MANAGER
Edit File: test_tightlayout.cpython-38.pyc
U ��]�( � @ st d dl Z d dlZd dlmZ d dlmZ d dlmZ d dl m Z m Z d dl mZ d@dd�Zed gd �d d � �Zed gd �dd� �Zedgd �dd� �Zedgdd�dd� �Zedgd �dd� �Zedgd �d d � �Zed gd �d d � �Zed!gd �d"d#� �Zed$gd �d%d&� �Zd'd(� ZdAd,d-�Zed.d/gd �d0d1� �Zd2d3� Z d4d5� Z d6d7� Z d8d9� Z d:d;� Z d<d=� Z!d>d?� Z"dS )B� N)�assert_array_equal)�image_comparison)�AnchoredOffsetbox� DrawingArea)� Rectangle� c C sH | � ddg� | jdd� | jd|d� | jd|d� | jd|d� d S ) N� � � �Znbins�x-label��fontsize�y-labelZTitle)�plot�locator_params� set_xlabel� set_ylabel� set_title)�axr � r �C/usr/lib/python3/dist-packages/matplotlib/tests/test_tightlayout.py� example_plot s r Z tight_layout1)�baseline_imagesc C s$ t �� \} }t|dd� t �� dS )z&Test tight_layout for a single subplot� r N��plt�subplotsr � tight_layout��figr r r r �test_tight_layout1 s r! Z tight_layout2c C sJ t jddd�\} \\}}\}}t|� t|� t|� t|� t �� dS )�'Test tight_layout for multiple subplotsr )ZnrowsZncolsNr �r �ax1�ax2�ax3Zax4r r r �test_tight_layout2 s r' Z tight_layout3c C sJ t �� } t �d�}t �d�}t �d�}t|� t|� t|� t �� dS )r" �� �� �z N)r �figure�subplotr r )r r$ r% r& r r r �test_tight_layout3' s r- Z tight_layout4)z2.5.5z2.6.1)r Zfreetype_versionc C sr t �� } t �dd�}t jdddd�}t jddddd�}t jdddd �}t|� t|� t|� t|� t �� d S ) z"Test tight_layout for subplot2grid)r r �r r )r r r )�colspan)r r �r/ �rowspan)r r )r1 N)r r+ � subplot2gridr r r# r r r �test_tight_layout48 s r3 Z tight_layout5c C s< t �� } t �d�}t�d��d�}|j|dd� t �� dS )zTest tight_layout for image�o �d )� r6 Znone)Z interpolationN)r r+ r, �npZarangeZreshapeZimshowr )r r Zarrr r r �test_tight_layout5L s r8 Z tight_layout6c C s\ t �� ��H t �dt� t�� } ddlm} |�dd�}| � |d �}| � |d �}t |� t |� |j | ddddgd� |�dd�}|D ]*}| � |�}t |� |� d � |� d � q�|j d d d � |j | ddddgd d� t|j|j�}t|j|j�} |j | dd| |j dd|j| gd� |j | dd| |j dd|j| gd d� W 5 Q R X dS )z Test tight_layout for gridspec�ignorer Nr r � �?��rectr � r r r g�������?)r< Zh_pad)�warnings�catch_warnings� simplefilter� UserWarningr r+ Zmatplotlib.gridspec�gridspecZGridSpec� add_subplotr r r r �min�top�max�bottom) r rB Zgs1r$ r% Zgs2Zssr rE rG r r r �test_tight_layout6Y s< � ��rH Z tight_layout7c C sr d} t �� \}}|�ddg� |jdd� |jd| d� |jd| d� |jd d | d � |jd d | d � t �� d S )Nr r r r r r r r z Left Title�left)�locr z Right Title�right)r r r r r r r r )r r r r r r �test_tight_layout7� s rL Z tight_layout8c C s0 t �� } | �ddi� | �d�}t|dd� dS )�"Test automatic use of tight_layout�pad皙�����?r4 r r N)r r+ Zset_tight_layoutrC r r r r r �test_tight_layout8� s rP Z tight_layout9c C s. t �dd�\} }|d d �d� t �� d S )Nr r F)r r � set_visibler )�fZaxarrr r r �test_tight_layout9� s rS c C s� t �� } | �d�}|jjdddd� |jjdddd� |jjdddddd � |jjdddddd � |jjd gdd � |jjd gdd � | �d �}|jjd ddd� |jjd ddd� | �d�}|jjdddd� |jjdddd� | �d�}|jjdddd� |jjdddd� t �� ddgddggddgddggddgddggddgddggg}t| j �D ](\}}t t � |� � �� d�|| � �q^dS )rM r( �out� r )�tickdir�length�width� T�minor)rV rW rX Ztick1OnZwhichr )rZ �� �inr) Zinout�� g����K�?g��C�l�?g��ʡE��?gB`��"��?g�I +��?g/�$��?g�Q� ��?gm����?N)r r+ rC ZxaxisZset_tick_paramsZyaxisZ set_ticksr � enumerateZaxesr r7 �round� get_positionZ get_points)r r ZansZnnr r r �test_outward_ticks� sN � � ��ra r6 rO �blackc C s� |d| }}| | f| df| |f|dfd|f||fd| f|| fd| fg }|D ]P}t ||�}td|||dddd�} |�| � td|d d|| jd d �} | �| � qZ| S ) z& Surround ax with OffsetBoxes r r: r. �Noner F)rX ZheightZ facecolorZ edgecolorZ linewidthZ antialiased�centerg )rJ �childrN ZframeonZbbox_to_anchorZbbox_transformZ borderpad)r r Z add_artistr Z transAxes) r �sizeZmargin�color�mZmpZ anchor_pointsZpointZdaZ backgroundZ anchored_boxr r r �add_offsetboxes� s< � � � ri Ztight_layout_offsetboxes1Ztight_layout_offsetboxes2c s� d ��ddddg� ddg ��� ����fdd �} | � }t �� | � }|�d d �� D ]&}|�� D ]}t|t�rf|�d � qfqZt �� d S ) Nr ZredZblueZgreenZyellowr r c sL t ����\} }|j}t|� �D ]&\}}|j��|d� t|d|d� q |S )N)rg � )r r Zflat�zipr ri )�_�axsr rg �ZcolorsZcolsZrows�x�yr r � _subplots� s z0test_tight_layout_offsetboxes.<locals>._subplotsF)r r Z get_children� isinstancer rQ )rq rm r re r rn r � test_tight_layout_offsetboxes� s rs c C s t �� } | �� dS )zOTests that tight layout doesn't cause an error when there are no axes. N)r Zgcfr )r r r r �test_empty_layout s rt c C sN t jdd�\} }|�d� tjdd�� }| �� t|�dks@t�W 5 Q R X dS ��-Test that warning emitted when xlabel too big�r r �ZfigsizeZdaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaT��recordr N�r r r r> r? r �len�AssertionError�r r �wr r r �"test_verybig_decorators_horizontal s r� c C sN t jdd�\} }|�d� tjdd�� }| �� t|�dks@t�W 5 Q R X dS ru �r r r r> r? r r| r} r~ r r r � test_verybig_decorators_vertical s r� c C sd t jdddd�\} }|d �d� |d �d� tjdd �� }| �� t|�dksVt�W 5 Q R X d S ) rv r r rw rx r Z aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaZ bbbbbbbbbbbbbbbbbbbbbbbbbbbbbbTry Nr{ �r rm r r r r � test_big_decorators_horizontal( s r� c C sd t jdddd�\} }|d �d� |d �d� tjdd �� }| �� t|�dksVt�W 5 Q R X d S ) rv r r rw rx r ZaaaaaaaaaaaaaaaaaaaaZbbbbbbbbbbbbbbbbbbbbTry Nr� r� r r r � test_big_decorators_vertical2 s r� c C sR t �dd�} t jddddd�}tjdd �� }t �� t|�d ksDt�W 5 Q R X d S ) N)� � r. )r� r� )r r r r� r0 Try r )r r2 r> r? r r| r} )r$ Zax5r r r r �test_badsubplotgrid< s r� c C s� t jdd�\} }|�ddg� |�ddg� |jdddd� |�� }tjdd ��6}t �� |�� }|j |j ksrt �t |�dks�t �W 5 Q R X t jddd d gd � d S ) NT)r r r zBIG LONG STRING)g �?r )g %@g �?)ZxyZxytextry g�������?r; ) r r Zset_xlimZset_ylimZannotater` r> r? r rX r} r| )r r Zp1r Zp2r r r �test_collapsedG s r� )r )r6 rO rb )#r> Znumpyr7 Z numpy.testingr Z matplotlib.testing.decoratorsr Zmatplotlib.pyplotZpyplotr Zmatplotlib.offsetboxr r Zmatplotlib.patchesr r r! r' r- r3 r8 rH rL rP rS ra ri rs rt r� r� r� r� r� r� r r r r �<module> sP � + ! � '