o
    ä¸Gf³!  ã                   @   s¦  d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZ	d dlm
Z
 d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d	d
„ Zdd„ Ze	j  !d¡edgddd„ ƒƒZ"dd„ Z#e	j  !d¡edgddd„ ƒƒZ$dd„ Z%dd„ Z&dd„ Z'dd„ Z(dd„ Z)G d d!„ d!ƒZ*d"d#„ Z+d$d%„ Z,d&d'„ Z-d(d)„ Z.ej/ 0d*e
j1 2¡ ¡d+d,„ ƒZ3d-d.„ Z4d/d0„ Z5d1d2„ Z6d3d4„ Z7d5d6„ Z8d7d8„ Z9dS )9é    )ÚBytesION)Úcm)Úsubprocess_run_helper)Úcheck_figures_equal)Úrrulewrapper)ÚVertexSelector)Úparasite_axesc                  C   sè   t  ¡ } t | tƒ tj¡ t  d¡}t |tƒ tj¡ t jdd}t jt	 
d¡dd t  ¡  t |tƒ tj¡ t  ¡  t jt	 
d¡t	 
d¡d t t  ¡ tƒ tj¡ t  ¡ } t  ¡ }t  t	 
d¡¡ | d¡ t | tƒ tj¡ d S )	Néy   Úpolar)Z
projectioné
   Zfoobar©Zlabel)ÚxZheightÚlog)ÚpltÚfigureÚpickleÚdumpr   ÚHIGHEST_PROTOCOLÚsubplotÚaxesÚplotÚnpÚarangeÚlegendZbarZgcaZ
set_yscale©ÚfigÚax© r   ú<lib/python3.10/site-packages/matplotlib/tests/test_pickle.pyÚtest_simple   s    

r   c                 C   sÚ  |   d¡ t | ¡ t d¡ t d¡t d¡}}t ddd¡ dd¡ } }}t |d ¡}t 	ddd	¡ t 
ttdƒƒ¡ t d
¡ t 	ddd¡ tj|ddgd t ¡  t 	ddd¡ t |¡ t 	ddd¡ t |¡ t d¡ t 	ddd¡ t |¡ t 	ddd¡}| dd¡ | dd¡ t ||||¡ t 	ddd¡}| dd¡ | dd¡ t ||||¡ t 	ddd¡ tj||d dd tjdd t 	ddd¡ tj||d ddd tjdd |  ¡  d S )N)r   é   z!Can you fit any more in a figure?é   r   r   éP   g333333ã¿é   é   Zhelloé   z//Zooo)Zhatchesé   zhello
world!é   r    é   é	   z$x^2$r   z
upper left)Úlocg      à¿gš™™™™™É?gš™™™™™Ù?)ZxerrZyerrT)Z	draggable)Úset_size_inchesr   r   Zsuptitler   r   ZlinspaceÚreshapeZsinr   r   ÚlistÚrangeZylabelZcontourfZcolorbarZ
pcolormeshÚimshowZpcolorÚset_xlimZset_ylimZ
streamplotZquiverZscatterr   ZerrorbarZalign_ylabels)Úfig_refr   ÚyÚdataÚuÚvr   r   r   r   Ú_generate_complete_test_figure0   sD   







r6   ÚdefaultZpng)Ú
extensionsc                 C   st   t |ƒ t |tj¡}ddd„ t |¡D ƒvsJ ‚t |¡}|j ¡  |  	| 
¡ ¡ |  |jj ¡ ¡ t |¡ d S )NZFigureCanvasAggc                 S   s   g | ]\}}}|‘qS r   r   )Ú.0ÚopÚargÚposr   r   r   Ú
<listcomp>m   s    z!test_complete.<locals>.<listcomp>)r6   r   Údumpsr   ÚpickletoolsZgenopsÚloadsÚcanvasÚdrawr+   Úget_size_inchesÚfigimageÚrendererÚbuffer_rgbar   Úclose)Úfig_testr1   ZpklZloadedr   r   r   Útest_completec   s   

rI   c                  C   sd   dd l } dd l}| jd }t|dƒ}| |¡}W d   ƒ n1 s"w   Y  tt| |¡ƒƒ d S )Nr   ÚPICKLE_FILE_PATHÚrb)Úosr   ÚenvironÚopenÚloadÚprintÚstrr>   )rL   r   ÚpathZblobr   r   r   r   Ú_pickle_load_subprocessw   s   
ÿrS   c                 C   sÄ   t |ƒ |d }| ¡ rJ ‚| d¡}t ||tj¡ W d   ƒ n1 s&w   Y  | ¡ s1J ‚ttdt|ƒddœd}t 	t
 |j¡¡}|j ¡  |  | ¡ ¡ |  |jj ¡ ¡ t |¡ d S )Nzsinus.pickleÚwbé<   ZAgg)rJ   Z
MPLBACKEND)ZtimeoutZ	extra_env)r6   ÚexistsrN   r   r   r   r   rS   rQ   r@   ÚastZliteral_evalÚstdoutrA   rB   r+   rC   rD   rE   rF   r   rG   )rH   r1   Ztmp_pathÚfpÚfileÚprocZ
loaded_figr   r   r   Ú test_pickle_load_from_subprocessƒ   s"   ÿý
r\   c                  C   sp   t  d¡} tƒ }t | |tj¡ t  d¡ t jjj	i ksJ ‚t 
| ¡ ¡} t jjj	i ks.J ‚|  ¡ dks6J ‚d S )Nza labelÚall)r   r   r   r   r   r   rG   Z_pylab_helpersZGcfZfigsr@   Ú	getbufferZ	get_label)r   Úbufr   r   r   Útest_gcfŸ   s   

r`   c                  C   sT   ddl m}  t ¡ }| |ƒ}| ddd¡}| g d¢g d¢¡ t |tƒ tj	¡ d S )Nr   )ÚFigureCanvasPdfr$   )r$   r%   r#   )
Zmatplotlib.backends.backend_pdfra   ÚmfigureÚFigureÚadd_subplotr   r   r   r   r   )ra   r   Ú_r   r   r   r   Útest_no_pyplotª   s   rf   c                  C   s*   ddl m}  | dddƒ}t |tƒ ¡ d S )Nr   )ÚRendererAggr   é   é   )Úmatplotlib.backends.backend_aggrg   r   r   r   )rg   rE   r   r   r   Útest_renderer´   s   rk   c                  C   s^   ddl m}  | dƒ}|jj}| ddd¡}| t d¡ dd¡¡ |j 	¡  t
 |tƒ ¡ d S )Nr   )Únew_figure_manageriè  r$   é   r#   r&   )rj   rl   rA   r   rd   r/   r   r   r,   rB   r   r   r   )rl   Zmanagerr   r   r   r   r   Ú
test_imageº   s   
rn   c                  C   s4   t jdd t  ¡ } t | ¡}t |¡ t  ¡  d S )NT)r
   )r   r   Zgcfr   r>   r@   rB   )r   Úpfr   r   r   Ú
test_polarÆ   s
   

rp   c                   @   s   e Zd Zdd„ ZdS )ÚTransformBlobc                 C   sJ   t  ¡ | _t  ¡ | _t  | j| j¡| _t  | j¡| _t  | j| j¡| _d S ©N)	ÚmtransformsZIdentityTransformÚidentityZ	identity2ZCompositeGenericTransformÚ	compositeZTransformWrapperÚwrapperÚ
composite2)Úselfr   r   r   Ú__init__Ï   s   

þ
þzTransformBlob.__init__N)Ú__name__Ú
__module__Ú__qualname__ry   r   r   r   r   rq   Î   s    rq   c                  C   s|   t ƒ } t | ¡}~ t |¡} | jj| jksJ ‚dd„ | jj ¡ D ƒ| j	gks(J ‚| jj
| jj
ks2J ‚| jj| jjks<J ‚d S )Nc                 S   s   g | ]}|ƒ ‘qS r   r   )r9   r5   r   r   r   r=   ç   s    z"test_transform.<locals>.<listcomp>)rq   r   r>   r@   rv   Z_childru   Z_parentsÚvaluesrw   Z
input_dimsZoutput_dims)Úobjro   r   r   r   Útest_transformÞ   s   

"r   c                  C   s8   t dƒ} zt t | ¡¡ W d S  ty   tdƒ ‚ w )Nr%   z!rrulewrapper pickling test failed)r   r   r@   r>   ÚRecursionErrorrP   )Úrr   r   r   Útest_rrulewrapperí   s   þr‚   c                  C   sN   t jddd\} }t t | ¡¡} | jd  dd¡ | jd  ¡ dks%J ‚d S )	Nr%   T)Zsharexr   r   rh   r$   )r   rh   )r   Úsubplotsr   r@   r>   r   r0   Zget_xlim)r   Zaxsr   r   r   Útest_sharedö   s   r„   c                  C   sD   t  ¡ \} }| g d¢¡ |jdtjtjfd t t 	| ¡¡ d S )N)çš™™™™™¹?r…   ç333333Ó?r†   Útop)Z	functions)
r   rƒ   Z
inset_axesZsecondary_xaxisr   ZsquareZsqrtr   r@   r>   r   r   r   r   Útest_inset_and_secondaryý   s   rˆ   Úcmapc                 C   s   t  | ¡ d S rr   )r   r>   )r‰   r   r   r   Ú	test_cmap  s   rŠ   c                  C   sN   t  ¡ } | jd usJ ‚tƒ }t | |¡ | d¡ t |¡}|jd us%J ‚d S )Nr   )rb   rc   rA   r   r   r   ÚseekrO   )r   ÚoutZfig2r   r   r   Útest_unpickle_canvas	  s   

r   c                  C   s0   t  g d¢¡} tt t | ¡¡ƒt jksJ ‚d S )N)r   r   r$   r$   )r   Z	host_axesÚtyper   r@   r>   ZHostAxes)r   r   r   r   Útest_mpl_toolkits  s   "r   c                   C   s*   t t t tj ¡ ¡¡ƒtjjksJ ‚d S rr   )rŽ   r   r@   r>   ÚmplÚcolorsZLogNormr   r   r   r   Útest_standard_norm  s   ÿr’   c                  C   s<   t j t jjt jj¡ƒ } tt t 	| ¡¡ƒt| ƒksJ ‚d S rr   )
r   r‘   Zmake_norm_from_scaleZscaleZ
LogitScaleZ	NormalizerŽ   r   r@   r>   )Zlogit_norm_instancer   r   r   Útest_dynamic_norm  s   ÿÿr“   c                  C   s,   t jddgdd\} t t t| ƒ¡¡ d S )Nr   r$   T)Zpicker)r   r   r   r@   r>   r   )Úliner   r   r   Útest_vertexselector$  s   r•   c                  C   s^   t  ¡  ¡ } | jg d¢d |  ddg¡ t t | ¡¡} |  ddg¡\}| ¡ dks-J ‚d S )N)ÚcÚmr2   Úk)r–   r$   r%   r#   r&   r—   )	r   r   rd   Zset_prop_cycler   r   r@   r>   Z	get_color)r   Úlr   r   r   Útest_cycler)  s   rš   ):Úior   rW   r   r?   Znumpyr   ZpytestZ
matplotlibr   r   Zmatplotlib.testingr   Zmatplotlib.testing.decoratorsr   Zmatplotlib.datesr   Zmatplotlib.linesr   Zmatplotlib.pyplotZpyplotr   Zmatplotlib.transformsZ
transformsrs   Zmatplotlib.figurer   rb   Zmpl_toolkits.axes_grid1r   r   r6   ZstyleÚcontextrI   rS   r\   r`   rf   rk   rn   rp   rq   r   r‚   r„   rˆ   ZmarkZparametrizeZ
_colormapsr}   rŠ   r   r   r’   r“   r•   rš   r   r   r   r   Ú<module>   sV    
3



	

