
    G8cb                         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 Z edgd      d	        Z ed
g      d        Z edgd      d        Zd Z edg      d        Zd Zy)    N)image_comparison)
CustomCellTable)Pathc                  8    ddg} t        j                  |        y )Nbr)cellColours)plttable)
cellcolorss    =/usr/lib/python3/dist-packages/matplotlib/tests/test_table.pytest_non_squarer   	   s    sJII*%    ztable_zorder.pngT)remove_textc            	      >   ddgddgg} d}dD cg c]  }d|z  	 }}g }t        j                  t        |            }t        |       D ],  }||z  }|j	                  |D cg c]
  }d|d	z  z   c}       . t        j
                  d
dt         j                  z  d      }t        j                  |t        j                  |      dd       t        j                  |||dd       t        j                  |||dd       t        j                  g        y c c}w c c}w )NiR iب iv  i )FreezeWind)d   2   z%d yearz%1.1fg     @@r      r      )lwzordercenter)cellText	rowLabels	colLabelslocr   zupper center)npzeroslenreversedappendlinspacepir   plotcosr   yticks)datar   xr   r   yoffrowts           r   test_zorderr0      s   FOFOD #I(121Q2I2H88C	N#D~ >t<!AfH-<=> 	Aqw$AHHQq	a*IIx!!	 IIx!! 	 JJrN1 3 =s   DD
ztable_labels.pngc                     d} t        j                  dd|       }t        j                  j	                  |      }dg| z  g| z  }t        j
                         }|j                  ddd      }|j                  d       |j                  ||d       |j                  ddd	      }|j                  d       |j                  ||d
g| z  d       |j                  ddd      }|j                  d       |j                  ||d       |j                  ddd      }|j                  d       |j                  ||d
g| z  d       y )N   r      1r   offbest)r   
rowColoursr    r   Header)r   r7   r   r    )r   
colColoursr    )r   r9   r   r    )	r!   r&   r   cmRdYlGnfigureadd_subplotaxisr   )	dimccoloursr   figax1ax2ax3ax4s	            r   test_label_coloursrG   0   s:   
C
Aq#AffmmAG}s"H
**,C
//!Q
"CHHUOIIx    //!Q
"CHHUOIIx !
S(  
 //!Q
"CHHUOIIx    //!Q
"CHHUOIIx !
S(  r   ztable_cell_manipulation.pngc                  J   d} dgt        |       z  gdz  }dgt        |       z  }t        j                  t        |       dt        |       dz   f      \  }}t        ||       D ]+  \  }}|j	                  ||d|	       |j                  d
       - t        j                          y )N
horizontalverticalopenclosedTRBLr4   r   g?r   r3   )nrowsfigsizer   )	colWidthsr   r    edgesr5   )r#   r   subplotszipr   r>   tight_layout)cellsr   rT   _axsaxcells          r   test_diff_cell_tabler^   U   s    LEE
"#a'HE
"I\\E
QE
14EFFAsUO D
#!	 	 	 	 r   c                  |   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 ]I  \  }}t        d|dd      }t        d |j                         j                         D              }||k(  rIJ  y )NrI   r   r   r3   )visible_edgeswidthheightc              3   &   K   | ]	  \  }}|  y w)N ).0rZ   ss      r   	<genexpr>z"test_customcell.<locals>.<genexpr>v   s     C41aQCs   )	r   MOVETOLINETO	CLOSEPOLYrW   r   tupleget_pathiter_segments)typescodesr/   r@   r]   codes         r   test_customcellrr   g   su   LE	dkk4;;T[[I	dkk4;;T[[I	dkk4;;T[[I	dkk4;;T^^L	dkk4;;T[[I	dkk4;;T[[I	dkk4;;T[[I	dkk4;;T[[I	
E E5! 1&1EC4==?#@#@#BCCDyyr   ztable_auto_column.pngc                  $   t        j                         } | j                  ddd      }|j                  d       |j	                  ddgddggddgd	d
gd      }|j                  d       |j                  d       |j                  g d       | j                  ddd      }|j                  d       |j	                  ddgddggddgd	d
gd      }|j                  d       |j                  d       |j                  d       | j                  ddd      }|j                  d       |j	                  ddgddggddgd	d
gd      }|j                  d       |j                  d       |j                  d       |j                  d       |j                  d       | j                  ddd      }|j                  d       |j	                  ddgddggddgd	d
gd      }|j                  d       |j                  d       |j                  d       y )Nr   r3   r5   zFit Textr   z-very long long text, Longer text than defaultArP   Col1Col2r   )r   r   r   r    F   )r   r3   r2   rx   r   z-101)r   r<   r=   r>   r   auto_set_font_sizeset_fontsizeauto_set_column_width)	rB   rC   tb1rD   tb2rE   tb3rF   tb4s	            r   test_auto_columnr   z   s=   
**,C //!Q
"CHHUO
))q/BAFH*6"  C 5!Rj) //!Q
"CHHUO
))q/BAFH*6"  C 5!Rj) //!Q
"CHHUO
))q/BAFH*6"  C 5!Rb!a a  //!Q
"CHHUO
))q/BAFH*6"  C 5!Rf%r   c                  *   t        j                         \  } }t        |      }|j                  dddd      }t	        |t
              sJ ||d   u sJ t        dddd       }||d<   |d   |u sJ |j                          t        j                  |       y )Nr3   r   )r3   r   r`   )ra   )r   r3   )r   rV   r   add_cell
isinstancer   
propertiessetp)rB   r\   r   r]   cell2s        r   test_table_cellsr      s    llnGC"IE>>!Q1%DdJ'''5;vq!48EE$K;% 
HHUOr   )matplotlib.pyplotpyplotr   numpyr!   matplotlib.testing.decoratorsr   matplotlib.tabler   r   matplotlib.pathr   r   r0   rG   r^   rr   r   r   re   r   r   <module>r      s      : .  & %&D9 :@ %&'! (!H 01tD E"& *+,7& -7&tr   