OwlCyberSecurity - MANAGER
Edit File: test_table.cpython-38.pyc
U ��] � @ s� d dl mZ d dlZd dlmZ d dlmZm Z d dl m Z dd� Z edgdgd d �d d � �Z ed gdgd�dd� �Zedgdgd d �dd� �Zdd� Zedgdgd�dd� �Zdd� ZdS )� N)�image_comparison)� CustomCell�Table)�Pathc C s ddg} t j| d� d S )N�b�r)Z cellColours)�plt�table)Z cellcolors� r �=/usr/lib/python3/dist-packages/matplotlib/tests/test_table.py�test_non_square s r Z table_zorderZpngT)�baseline_images� extensionsZ remove_textc C s� ddgddgg} d}dd� dD �}g }t �t|��}t| �D ] }||7 }|�d d� |D �� q<t �d d t j d �}tj|t � |�d d d� tj |||ddd� tj |||dd d� t� g � d S )NiR iب iv� i�� )ZFreezeZWindc S s g | ] }d | �qS )z%d yearr ��.0�xr r r � <listcomp> s z test_zorder.<locals>.<listcomp>)�d �2 c S s g | ]}d |d �qS )z%1.1fg @�@r r r r r r s r � r � )Zlw�zorder�center���)�cellText� rowLabels� colLabels�locr z upper center) �npZzeros�len�reversed�append�linspaceZpir ZplotZcosr Zyticks)�datar r r Zyoff�row�tr r r � test_zorder s2 � ��r& Z table_labels)r r c C s� d} t �dd| �}tj�|�}dg| g| }t�� }|�ddd�}|�d� |j||dd� |�ddd �}|�d� |j||d g| dd � |�ddd�}|�d� |j||dd � |�ddd�}|�d� |j||d g| dd � d S )N� r � �1r �offZbest)r � rowColoursr r ZHeader)r r+ r r )r � colColoursr )r r, r r ) r r" r �cmZRdYlGn�figure� add_subplot�axisr ) Zdim�cZcoloursr �fig�ax1�ax2�ax3�ax4r r r �test_label_colours2 s> � � � �r7 Ztable_cell_manipulationc C s� d} dgt | � gd }dgt | � }tjt | �dt | �d fd�\}}t|| �D ]$\}}|j||d|d � |�d � qRt�� d S ) N�Z horizontalZvertical�open�closed�T�R�B�Lr) r g�������?r r( )ZnrowsZfigsizer )� colWidthsr r Zedgesr* )r r �subplots�zipr r0 Z tight_layout)Zcellsr r? �_Zaxes�ax�cellr r r �test_diff_cell_tableX s "� rE c C s d} t jt jt jt jt jft jt jt jt jt jft jt jt jt jt jft jt jt jt jt jft jt jt jt jt jft jt jt jt jt jft jt jt jt jt jft jt jt jt jt jff}t| |�D ]>\}}td|ddd�}tdd� |�� �� D ��}||ks�t �q�d S )Nr8 �r r r( )� visible_edges�widthZheightc s s | ]\}}|V qd S )Nr )r rB �sr r r � <genexpr>z s z"test_customcell.<locals>.<genexpr>) r ZMOVETOZLINETOZ CLOSEPOLYrA r �tupleZget_pathZ iter_segments�AssertionError)�typesZcodesr% r1 rD �coder r r �test_customcellk s � rO Ztable_auto_columnc C s� t �� } | �ddd�}|�d� |jddgddggddgd d gd d �}|�d � |�d� |�dddg� | �ddd�}|�d� |jddgddggddgd d gd d �}|�d � |�d� |�d� | �ddd�}|�d� |jddgddggddgd d gd d �}|�d � |�d� |�d� |�d� |�d� | �ddd�}|�d� |jddgddggddgd d gd d �}|�d � |�d� |�d� d S )Nr r( r* zFit Textr z-very long long text, Longer text than default�Ar= ZCol1ZCol2r )r r r r F� ���r )rR r r( r' z-101)r r. r/ r0 r Zauto_set_font_sizeZ set_fontsizeZauto_set_column_width) r2 r3 Ztb1r4 Ztb2r5 Ztb3r6 Ztb4r r r �test_auto_column~ sf �� �� �� �� rS c C s� t �� \} }t|�}|�dddd�}t|t�s2t�||d ksBt�tdddd d�}||d< |d |ksjt�|�� t �|� d S )Nr( r )r( r rF )rG )r r( ) r r@ r Zadd_cell� isinstancer rL Z propertiesZsetp)r2 rC r rD Zcell2r r r �test_table_cells� s rU )Zmatplotlib.pyplotZpyplotr Znumpyr Z matplotlib.testing.decoratorsr Zmatplotlib.tabler r Zmatplotlib.pathr r r& r7 rE rO rS rU r r r r �<module> s2 � � $ � � 6