o
    ØtBhb  ã                   @   s¨   d dl mZ d dlZd dlmZ d dlmZm	Z	 d dl
mZ dd„ Zedgdd	d
d„ ƒZedgƒdd„ ƒZedgdd	dd„ ƒZdd„ Zedgƒdd„ ƒZdd„ ZdS )é    N)Úimage_comparison)Ú
CustomCellÚTable)ÚPathc                  C   s   ddg} t j| d d S )NÚbÚr)ÚcellColours)ÚpltÚtable)Ú
cellcolors© r   úr/var/www/html/riverr-enterprise-integrations-main/venv/lib/python3.10/site-packages/matplotlib/tests/test_table.pyÚtest_non_square	   s   r   ztable_zorder.pngT)Úremove_textc                  C   sÀ   ddgddgg} d}dd„ dD ƒ}g }t  t|ƒ¡}t| ƒD ]}||7 }| d	d„ |D ƒ¡ qt  d
dt j d¡}tj|t  	|¡ddd tj
|||ddd tj
|||ddd t g ¡ d S )NiR iØ¨ ivã  iÓÐ )ÚFreezeÚWindc                 S   s   g | ]}d | ‘qS )z%d yearr   ©Ú.0Úxr   r   r   Ú
<listcomp>   s    ztest_zorder.<locals>.<listcomp>)éd   é2   c                 S   s   g | ]}d |d  ‘qS )z%1.1fg     @@r   r   r   r   r   r      s    r   é   r   é   )ÚlwÚzorderÚcenteréþÿÿÿ)ÚcellTextÚ	rowLabelsÚ	colLabelsÚlocr   zupper center)ÚnpÚzerosÚlenÚreversedÚappendÚlinspaceÚpir	   ÚplotÚcosr
   Úyticks)Údatar    r   r   ÚyoffÚrowÚtr   r   r   Útest_zorder   s2   ÿüür0   ztable_labels.pngc            	      C   sê   d} t  dd| ¡}tj |¡}dg|  g|  }t ¡ }| ddd¡}| d¡ |j||dd | ddd	¡}| d¡ |j||d
g|  dd | ddd¡}| d¡ |j||dd | ddd¡}| d¡ |j||d
g|  dd d S )Né   r   é   Ú1r   ÚoffÚbest)r   Ú
rowColoursr!   r   ÚHeader)r   r6   r   r!   )r   Ú
colColoursr!   )r   r8   r    r!   )	r"   r'   r	   ÚcmÚRdYlGnÚfigureÚadd_subplotÚaxisr
   )	ÚdimÚcÚcoloursr   ÚfigÚax1Úax2Úax3Úax4r   r   r   Útest_label_colours0   s>   
þ
ý
þ

ýrF   ztable_cell_manipulation.pngc                  C   s„   d} dgt | ƒ gd }dgt | ƒ }tjt | ƒdt | ƒd fd\}}t|| ƒD ]\}}|j||d|d	 | d
¡ q)t ¡  d S )N©Ú
horizontalÚverticalÚopenÚclosedÚTÚRÚBÚLr3   r   gš™™™™™¹?r   r2   )ÚnrowsÚfigsizer   )Ú	colWidthsr   r!   Úedgesr4   )r$   r	   ÚsubplotsÚzipr
   r=   Útight_layout)Úcellsr   rR   Ú_ÚaxsÚaxÚcellr   r   r   Útest_diff_cell_tableU   s   "ür\   c                  C   s  d} t jt jt jt jt jft jt jt jt jt jft jt jt jt jt jft jt jt jt jt jft jt jt jt jt jft jt jt jt jt jft jt jt jt jt jft jt jt jt jt jff}t| |ƒD ]\}}td|ddd}tdd„ | ¡  ¡ D ƒƒ}||ks€J ‚qad S )NrG   ©r   r   r2   )Úvisible_edgesÚwidthÚheightc                 s   s    | ]\}}|V  qd S )Nr   )r   rX   Úsr   r   r   Ú	<genexpr>v   s   € z"test_customcell.<locals>.<genexpr>)	r   ÚMOVETOÚLINETOÚ	CLOSEPOLYrU   r   ÚtupleÚget_pathÚiter_segments)ÚtypesÚcodesr/   r?   r[   Úcoder   r   r   Útest_customcellg   s   øýrl   ztable_auto_column.pngc            	      C   s”  t  ¡ } |  ddd¡}| d¡ |jddgddggddgd	d
gdd}| d¡ | d¡ | g d¢¡ |  ddd¡}| d¡ |jddgddggddgd	d
gdd}| d¡ | d¡ | d¡ |  ddd¡}| d¡ |jddgddggddgd	d
gdd}| d¡ | d¡ | d¡ | d¡ | d¡ |  ddd¡}| d¡ |jddgddggddgd	d
gdd}| d¡ | d¡ | d¡ d S )Nr   r2   r4   zFit Textr   z-very long long text, Longer text than defaultÚArN   ÚCol1ÚCol2r   )r   r   r    r!   Fé   )éÿÿÿÿr   r2   r1   rq   r   z-101)r	   r;   r<   r=   r
   Úauto_set_font_sizeÚset_fontsizeÚauto_set_column_width)	rA   rB   Útb1rC   Útb2rD   Útb3rE   Útb4r   r   r   Útest_auto_columnz   sn   
ÿû


ÿû



ÿû





ÿû

ry   c                  C   s€   t  ¡ \} }t|ƒ}| dddd¡}t|tƒsJ ‚||d u s!J ‚tdddd d}||d< |d |u s5J ‚| ¡  t  |¡ d S )Nr2   r   )r2   r   r]   )r^   )r   r2   )r	   rT   r   Úadd_cellÚ
isinstancer   Ú
propertiesÚsetp)rA   rZ   r
   r[   Úcell2r   r   r   Útest_table_cellsµ   s   r   )Úmatplotlib.pyplotÚpyplotr	   Únumpyr"   Úmatplotlib.testing.decoratorsr   Úmatplotlib.tabler   r   Úmatplotlib.pathr   r   r0   rF   r\   rl   ry   r   r   r   r   r   Ú<module>   s     
 
$

: