a
    ضab                     @   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)ZcellColours)plttable)Z
cellcolors r
   k/Users/vegardjervell/Documents/master/model/venv/lib/python3.9/site-packages/matplotlib/tests/test_table.pytest_non_square	   s    r   ztable_zorder.pngT)Z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 )ZFreezeZWindc                 S   s   g | ]}d | qS )z%d yearr
   .0xr
   r
   r   
<listcomp>       ztest_zorder.<locals>.<listcomp>)d   2   c                 S   s   g | ]}d |d  qS )z%1.1fg     @@r
   r   r
   r
   r   r      r   r      r      )Zlwzordercenter)cellText	rowLabels	colLabelslocr   zupper center)npZzeroslenreversedappendlinspacepir   Zplotcosr	   Zyticks)datar   r   r   Zyoffrowtr
   r
   r   test_zorder   s2    r'   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   r-   r   r   )r   
colColoursr   )r   r/   r   r   )	r   r!   r   cmZRdYlGnfigureadd_subplotaxisr	   )	ZdimcZcoloursr   figax1ax2ax3ax4r
   r
   r   test_label_colours0   s>    



r:   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
 qRt  d S )NZ
horizontalZverticalopenclosedTRBLr*   r   g?r   r)   )ZnrowsZfigsizer   )	colWidthsr   r   edgesr+   )r   r   subplotszipr	   r3   Ztight_layout)cellsr   rB   _Zaxsaxcellr
   r
   r   test_diff_cell_tableU   s    "rJ   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sJ qd S )Nr;   r   r   r)   )visible_edgeswidthheightc                 s   s   | ]\}}|V  qd S )Nr
   )r   rG   sr
   r
   r   	<genexpr>v   r   z"test_customcell.<locals>.<genexpr>)	r   ZMOVETOZLINETOZ	CLOSEPOLYrE   r   tupleget_pathZiter_segments)typescodesr&   r4   rI   coder
   r
   r   test_customcellg   s    rV   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   r)   r+   zFit Textr   z-very long long text, Longer text than defaultAr@   ZCol1ZCol2r   )r   r   r   r   F   )r   r)   r(   rY   r   z-101)r   r1   r2   r3   r	   Zauto_set_font_sizeZset_fontsizeZauto_set_column_width)	r5   r6   Ztb1r7   Ztb2r8   Ztb3r9   Ztb4r
   r
   r   test_auto_columnz   sn    















rZ   c                  C   s   t  \} }t|}|dddd}t|ts2J ||d u sBJ tdddd d}||d< |d |u sjJ |  t | d S )Nr)   r   )r)   r   rK   )rL   )r   r)   )r   rD   r   Zadd_cell
isinstancer   Z
propertiesZsetp)r5   rH   r	   rI   Zcell2r
   r
   r   test_table_cells   s    r\   )Zmatplotlib.pyplotZpyplotr   Znumpyr   Zmatplotlib.testing.decoratorsr   Zmatplotlib.tabler   r   Zmatplotlib.pathr   r   r'   r:   rJ   rV   rZ   r\   r
   r
   r
   r   <module>   s   
 
$

: