OwlCyberSecurity - MANAGER
Edit File: test_constrainedlayout.cpython-38.pyc
U ��]}5 � @ sj d dl Zd dlZd dlmZ d dlmZ d dlm Z d dl m Z m Z d[dd�Z d\dd �Zed gd gd �d d� �Zedgd gd �dd� �Zedgd gd �dd� �Zedgd�dd� �Zedgdd gd�d d � �Zed gd gd �d d � �Zd!d"� Zed#gd gd �d$d%� �Zed&gd gd �d'd(� �Zed)gd gd �d*d+� �Zed,gd gd �d-d.� �Zed/gd gd �d0d1� �Zed2gd gd �d3d4� �Zed5gd6d gd�d7d8� �Z ed9gd gd �d:d;� �Z ed<gd gd �d=d>� �Z ed?gd gd �d@dA� �Z edBgd gd �dCdD� �Z dEdF� Z!dGdH� Z"dIdJ� Z#dKdL� Z$dMdN� Z%dOdP� Z&ej'�(dQ�ej'j)dRdS�edTgd gdUdVdW�dXdT� ���Z*dYdZ� Z+dS )]� N)�image_comparison)�ticker�rcParams� Fc C sb | � ddg� | jdd� |sJ| jd|d� | jd|d� | jd|d� n| �d � | �d � d S ) N� � � �Znbins�x-label��fontsize�y-label�Title� )�plotZlocator_params� set_xlabel� set_ylabel� set_titleZset_xticklabelsZset_yticklabels)�axr �nodec� r �I/usr/lib/python3/dist-packages/matplotlib/tests/test_constrainedlayout.py� example_plot s r c C s� d\}}t jtdd| |�tdd| |�f \}}d|d |d |d t �|d |d � }| j|||dd d d d �}| jd |d� | jd|d� | jd|d� |S )N)�333333�?r ���r r � @� r ZRdBu_rg �� �?T)ZcmapZvminZvmaxZ rasterizedr r r r )�npZmgrid�sliceZexp� pcolormeshr r r )r r ZdxZdy�y�x�z�pcmr r r �example_pcolor s � 4�r% Zconstrained_layout1Zpng)�baseline_images� extensionsc C s& t jdd�} | �d�}t|dd� dS )z,Test constrained_layout for a single subplotT��constrained_layout�o � r N)�plt�figure� add_subplotr )�figr r r r �test_constrained_layout1$ s r0 Zconstrained_layout2c C s0 t jdddd�\} }|jD ]}t|dd� qdS )z(Test constrained_layout for 2x2 subplotsr Tr( r+ r N)r, �subplots�flatr �r/ �axsr r r r �test_constrained_layout2- s r5 Zconstrained_layout3c C sZ t jdddd�\} }t|j�D ]6\}}t|dd�}|dkr@d}nd}| j|||d � q d S ) z3Test constrained_layout for colorbars with subplotsr Tr( r+ r r g{�G�z�?�{�G�z�?)r �padN)r, r1 � enumerater2 r% �colorbar)r/ r4 Znnr r$ r7 r r r �test_constrained_layout36 s r: Zconstrained_layout4)r& c C sB t jdddd�\} }|jD ]}t|dd�}q| j||ddd� d S ) z;Test constrained_layout for a single colorbar with subplotsr Tr( r+ r �{�G�z�?r �r r7 �shrinkN�r, r1 r2 r% r9 �r/ r4 r r$ r r r �test_constrained_layout4D s r@ Zconstrained_layout5g�������?)r& Ztolr' c C sF t jdddd�\} }|jD ]}t|dd�}q| j||dddd d � d S ) zZ Test constrained_layout for a single colorbar with subplots, colorbar bottom r Tr( r+ r Fr; r �bottom)r Z use_gridspecr7 r= �locationNr> r? r r r �test_constrained_layout5M s �rC Zconstrained_layout6c C s� t jdd�} | jdd| d�}|d �dd�}|d �dd�}g }|D ]$}| �|�}||g7 }t|dd� qD|�d � g }|D ]$}| �|�}||g7 }t|dd�}q|| j||d d d t j d d�d� dS )z,Test constrained_layout for nested gridspecsTr( r r �r- r r r zx-label MultiLiner; g�G�z��?rA r r )r r7 r= rB ZticksN) r, r- Z add_gridspecZ subgridspecr. r r r% r9 r Z MaxNLocator)r/ �gs�gsl�gsr�axslr Zaxsrr$ r r r �test_constrained_layout6\ s* �rI c C s� t jtdd��j tjdd�} t�dd�}t�dd|d �}t�dd|d �}g }|D ]}| �|�}qT| � | j � � � W 5 Q R X dS ) z2Test for proper warning if fig not set in GridSpeczPCalling figure.constrained_layout, but figure not setup to do constrained layout)�matchTr( r r r N) �pytestZwarns� UserWarningr, r- �gridspec�GridSpec�GridSpecFromSubplotSpecr. �draw�canvasZ get_renderer)r/ rE rF rG rH r r r r �test_constrained_layout7u s rR Zconstrained_layout8c C s� t jddd�} tjdd| d�}g }dD ]|}|dkr:d g}ndd g}|D ]Z}| �|||f �}||g7 }t|d d �}|dkr�|�d � |d k r�|�d � |�d � qFq&| �|ddd�f �}||g7 }t|d d �}| j ||ddd� dS )z/Test for gridspecs that are not completely full)� r T)�figsizer) r r rD �r r r r � � r r r Nr; r r< ) r, r- rM rN r. r% r r r r9 )r/ rE r4 �jZilist�ir r$ r r r �test_constrained_layout8� s( rZ Zconstrained_layout9c C sr t jdddddd�\} }|jD ]$}t|dd�}|�d� |�d� q |�d� | j||d d d � | jd d d� dS )z4Test for handling suptitle and for sharex and shareyr TF)r) ZsharexZshareyr+ r r r r; r r< z Test Suptitle� N) r, r1 r2 r% r r Z set_aspectr9 �suptitler? r r r �test_constrained_layout9� s � r] Zconstrained_layout10c C sF t jdddd�\} }|jD ]}|jt�d�dd� q|jddd � d S ) z%Test for handling legend outside axisr Tr( r zThis is a label)Zlabelz center left)g�������?� �?)ZlocZbbox_to_anchorN)r, r1 r2 r r ZarangeZlegendr3 r r r �test_constrained_layout10� s r_ Zconstrained_layout11c C s� t jddd�} tjdd| d�}t�dd|d �}t�dd|d �}| �|d �}t|dd � g }|D ]$}| �|�}||g7 }t|dd �}qd| j||d d d � | �|d �}t|dd � d S )z#Test for multiple nested gridspecs T�rS r �r) rT r r rD r rW r r � �Q@�r r= �aspectN� r, r- rM rN rO r. r r% r9 �r/ �gs0rF Zgsl0r r4 rE r$ r r r �test_constrained_layout11� s rh Zconstrained_layout11ratc C s� t jddd�} tjdd| ddgd�}t�dd|d �}tjdd|d d dgd �}| �|d �}t|d d � g }|D ]$}| �|�}||g7 }t|d d �}qr| j||ddd� | �|d �}t|d d � dS )z4Test for multiple nested gridspecs with width_ratiosTr` ra r r g @r )r- Z width_ratiosr r )Z height_ratiosrW r r rb rc Nre rf r r r � test_constrained_layout11rat� s � ri Zconstrained_layout12c C s� t jdd�} tjdd| d�}| �|dd�df �}| �|dd�df �}t|d d � t|d d � | �|d d�d f �}t|dd � | �|dd �d f �}t|dd � | �|d d�d f �}t|dd � |�d� dS )z/Test that very unbalanced labeling still works.Tr( � r rD Nr r r+ r r )r rV r )r, r- rM rN r. r r )r/ rg �ax1�ax2r r r r �test_constrained_layout12� s rm Zconstrained_layout13r6 c C sR t jdddd�\} }|jD ]$}t|dd�}| j||dddd � q| jd d d � d S ) �Test that padding works.r Tr( r r r � 4@r6 �r r= rd r7 gUUUUUU�?)�w_pad�h_padN�r, r1 r2 r% r9 Zset_constrained_layout_padsr? r r r �test_constrained_layout13� s rt Zconstrained_layout14c C sV t jdddd�\} }|jD ]$}t|dd�}| j||dddd � q| jd d d d d � d S )rn r Tr( r r r ro r6 rp gUUUUUU�?皙�����?)rq rr ZhspaceZwspaceNrs r? r r r �test_constrained_layout14 s �rv Zconstrained_layout15c C s4 dt d< t�dd�\} }|jD ]}t|dd� q dS )zTest that rcparams work.Tz figure.constrained_layout.user r r N)r r, r1 r2 r r3 r r r �test_constrained_layout15 s rw Zconstrained_layout16c C s2 t jdd�\} }t|dd� | �ddddg�}dS )zTest ax.set_position.Tr( r r ru 皙�����?N)r, r1 r �add_axes�r/ r rl r r r �test_constrained_layout16 s r{ Zconstrained_layout17c C s� t jdd�} tjdd| d�}| �|d �}| �|ddd�f �}| �|dd�dd �f �}| �|dd�d f �}t|� t|� t|� t|� dS ) zTest uneven gridspecsTr( r rD �r r r r Nr ���)r, r- rM rN r. r )r/ rE rk rl Zax3Zax4r r r �test_constrained_layout17$ s r~ c C sV t jdd�\} }|�� }t|� t|dd� | j�� t|�� j|�� jk�sRt �dS )z Test twinxTr( r+ r N) r, r1 Ztwinxr rQ rP �all� get_position�extents�AssertionErrorrz r r r �test_constrained_layout186 s r� c C sj t jdd�\} }|�� }t|� t|dd� |�d� |�d� | j�� t|�� j |�� j k�sft �dS )z Test twinyTr( r+ r r N) r, r1 Ztwinyr r rQ rP r r� r� r� rz r r r �test_constrained_layout19@ s r� c C s\ t �ddd�} t �| | dd�df �}t�� }|�ddddg�}|�| | |�}|�|� dS )z)Smoke test cl does not mess up added axes���r rV Nr r )r ZlinspaceZhypotr, r- ry r r9 )ZgxZimgr/ r Zmeshr r r �test_constrained_layout20L s r� c C sj t jdd�\} }| �d� | j�� t�|�� j�}| �d� | j�� t�|�� j�}tj � ||� dS )z>#11035: repeated calls to suptitle should not alter the layoutTr( Z Suptitle0Z Suptitle1N) r, r1 r\ rQ rP r �copyr� r� �testing�assert_allclose�r/ r Zextents0�extents1r r r �test_constrained_layout21W s r� c C sd t jdd�\} }| j�� t�|�� j�}| jddd� | j�� t�|�� j�}tj � ||� dS )zC#11035: suptitle should not be include in CL if manually positionedTr( ZSuptitler^ )r! N) r, r1 rQ rP r r� r� r� r\ r� r� r� r r r �test_constrained_layout22f s r� c C s6 t d�D ](} tjdddd�\}}|�d�| �� qdS )zu Comment in #11035: suptitle used to cause an exception when reusing a figure w/ CL with ``clear=True``. r Z123T)Znumr) �clearz Suptitle{}N)�ranger, r1 r\ �format)rY r/ r r r r �test_constrained_layout23t s r� � r )Zreruns�test_colorbar_locationTZmpl20)r& r' Z remove_textZstylec C s� t jdddd�\} }|jD ] }t|�}|�d� |�d� q| j||dd�df dd � | j||d dd �f d d d� | j||dd d�f d d d� | j||ddd�f d dd� | j||d d dd� | j||dd�d f d dd� dS )zY Test that colorbar handling is as expected for various complicated cases... rV r Tr( r Nr rx )r r= r} r r^ rA )r r= rB r ���r �topr| �left�right)r, r1 r2 r% r r r9 r? r r r r� � s c C s^ t jdddd�\} }|d �d� | j�� t�|d �� j�}tj j |ddd d gd d � d S ) Nr Tr( rU Fr| g��|�R�?g�A�F���?g�Բ��H�?g��9?�q�?g�h㈵��>)Zrtol) r, r1 Z set_visiblerQ rP r r� r� r� r� r� )r/ r4 r� r r r �test_hidden_axes� s �r� )r F)r ),Znumpyr rK Z matplotlib.testing.decoratorsr Zmatplotlib.pyplotZpyplotr, Zmatplotlib.gridspecrM Z matplotlibr r r r% r0 r5 r: r@ rC rI rR rZ r] r_ rh ri rm rt rv rw r{ r~ r� r� r� r� r� r� ZmarkZtimeoutZflakyr� r� r r r r �<module> s� � � � � � � � � � � � � � � � � �