OwlCyberSecurity - MANAGER
Edit File: test_subplots.cpython-38.pyc
U ��]� � @ s� d dl Z d dlZd dlZd dlmZ d dlmZ d dlZdd� Z dd� Z dd� Z d d � Z d d � Z ed gdd�dd� �Zdd� Zdd� ZdS )� N)�image_comparisonc C s� t �t| �t| �ttd||g���D ]p\\}}\}}\}\}} ||krHq$t| d d�|���||�| ||f ks$td||| ||f r�dnd|f ��q$dS )z� x_shared and y_shared are n x n boolean matrices; entry (i, j) indicates whether the x (or y) axes of subplots i and j should be shared. Zxyr z_shared_{}_axesz,axes %i and %i incorrectly %ssharing %s axisznot � N)� itertools�product� enumerate�zip�getattr�formatZjoined�AssertionError) �axsZx_sharedZy_sharedZi1Zax1Zi2Zax2Zi3�nameZshared� r �@/usr/lib/python3/dist-packages/matplotlib/tests/test_subplots.py� check_shared s$ � �� ��r c C s� dd� }t | ||�D ]v\}}}|�� |�� jg D ] }|�� |ks2td||� ��q2|�� |�� jg D ] }|�� |kshtd||� ��qhqd S )Nc S s | rdS dS )NZ invisible�visibler )�vr r r �tostr s z check_visible.<locals>.tostrzX axis was incorrectly %szY axis was incorrectly %s)r Zget_xticklabelsZ get_xaxisZ offsetTextZ get_visibler Zget_yticklabelsZ get_yaxis)r Z x_visibleZ y_visibler �axZvxZvy�lr r r � check_visible s � �r c C s\ d} t j| d d� td�t j| d d� td�t �ddddgddddgddddgddddgg�t �ddddgddddgddddgddddgg�d�}ddddgddddgdgd dgd dgd ddddgd�ddddgdgd ddddgdgd dgd ddddgd�d �}|d |d< |d |d< t�dd�\}\\}}\}}||||g}t||d |d � t�|� ddd d d d g} | D ]z} | D ]n} tjdd| | d�\}\\}}\}}||||g}t||| || � t ||d | |d | � t�|� �q��q~tjddddd�\}\\}}\}}||||g}|D ]} | � � �q,t |ddddgddddg� d S )N)� r � r )ZdtypeFT)�all�none�row�colr )r r r r FT)�x�yr r r r ��sharex�shareyr r ) �numpyZones�boolZzerosZarray�plt�subplotsr �closer Z label_outer) ZrdimZsharer �f�a1�a2Za3Za4r ZopsZxoZyor r r r � test_shared* sd � �� � �� " " r) c C s� t jdddd�\} \}}t|gdgdg� |j�� t|gdgdg� t jdddd�\} \}}t|gdgdg� |j�� t|gdgdg� d S )N� r T�r F�r )r# r$ r ZyaxisZ tick_leftZxaxisZ tick_bottom)r& r' r( r r r �test_shared_and_movedg s r- c C s� t �t�� tjdddd� W 5 Q R X t �t�� tjdddd� W 5 Q R X t�� �� tjddtd� t �t�� t�ddd� W 5 Q R X t �t�� t�ddd � W 5 Q R X t �t�� t�ddd � W 5 Q R X W 5 Q R X d S ) Nr Zblahr, r+ �ignorez .*sharex argument to subplots)�message�category���r � ) �pytestZraises� ValueErrorr# r$ �warnings�catch_warnings�filterwarnings� UserWarningr r r r �test_exceptionsu s � r9 Zsubplots_offset_textF)Zbaseline_imagesZ remove_textc C sz t �ddd�} t �ddd�d }tjdddd d �\}}|d �| | � |d �| | � |d �|| � |d �|| � d S )Nr g _�Bg e��A�d � g ��@r r r r )r r )r* r )r r* )r* r* )r! Zaranger# r$ Zplot)r r �figZaxesr r r �test_subplots_offsettext� s r= c C s( t �� \} }|�� �� |�� ks$t�d S )N)r# r$ Zget_subplotspecZ get_gridspecr )r< r r r r �test_get_gridspec� s r> c C sR ddi} dddgi}t jdd| |d�\}}| ddiks:t�|dddgiksNt�d S )Nr r Z width_ratiosr* r )� subplot_kw� gridspec_kw)r# r$ r )r? r@ r< r r r r �test_dont_mutate_kwargs� s � rA )r r5 r! Zmatplotlib.pyplotZpyplotr# Z matplotlib.testing.decoratorsr r3 r r r) r- r9 r= r> rA r r r r �<module> s =