o
    ˆ‚¡e³!  ã                   @   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)Ú
projectioné
   Úfoobar©Úlabel)ÚxÚheightÚlog)ÚpltÚfigureÚpickleÚdumpr   ÚHIGHEST_PROTOCOLÚsubplotÚaxesÚplotÚnpÚarangeÚlegendÚbarÚgcaÚ
set_yscale©ÚfigÚax© r$   úK/var/www/myenv/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ã¿é   é   Úhelloé   z//Úooo)Úhatchesé   zhello
world!é   r'   é   é	   z$x^2$r   z
upper left)Úlocg      à¿gš™™™™™É?gš™™™™™Ù?)ÚxerrÚyerrT)Ú	draggable)Úset_size_inchesr   r   Úsuptitler   r   ÚlinspaceÚreshapeÚsinr   r   ÚlistÚrangeÚylabelÚcontourfÚcolorbarÚ
pcolormeshÚimshowÚpcolorÚset_xlimÚset_ylimÚ
streamplotÚquiverÚscatterr   ÚerrorbarÚalign_ylabels)Úfig_refr   ÚyÚdataÚuÚvr#   r$   r$   r%   Ú_generate_complete_test_figure0   sD   







rQ   ÚdefaultÚpng)Ú
extensionsc                 C   st   t |ƒ t |tj¡}ddd„ t |¡D ƒvsJ ‚t |¡}|j ¡  |  	| 
¡ ¡ |  |jj ¡ ¡ t |¡ d S )NÚFigureCanvasAggc                 S   s   g | ]\}}}|‘qS r$   r$   )Ú.0ÚopÚargÚposr$   r$   r%   Ú
<listcomp>m   s    z!test_complete.<locals>.<listcomp>)rQ   r   Údumpsr   ÚpickletoolsÚgenopsÚloadsÚcanvasÚdrawr8   Úget_size_inchesÚfigimageÚrendererÚbuffer_rgbar   Úclose)Úfig_testrL   ÚpklÚ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   ÚpathÚblobr"   r$   r$   r%   Ú_pickle_load_subprocessw   s   
ÿrt   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é<   ÚAgg)rj   Ú
MPLBACKEND)ÚtimeoutÚ	extra_env)rQ   Úexistsrn   r   r   r   r   rt   rq   r^   ÚastÚliteral_evalÚstdoutr_   r`   r8   ra   rb   rc   rd   r   re   )rf   rL   Útmp_pathÚfpÚfileÚprocÚ
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   re   Ú_pylab_helpersÚGcfÚfigsr^   Ú	getbufferÚ	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*   )
Úmatplotlib.backends.backend_pdfr   ÚmfigureÚFigureÚadd_subplotr   r   r   r   r   )r   r"   Ú_r#   r$   r$   r%   Útest_no_pyplotª   s   r“   c                  C   s*   ddl m}  | dddƒ}t |tƒ ¡ d S )Nr   )ÚRendererAggr   é   é   )Úmatplotlib.backends.backend_aggr”   r   r   r   )r”   rc   r$   r$   r%   Útest_renderer´   s   r˜   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*   r0   )r—   r™   r_   r   r‘   rC   r   r   r;   r`   r   r   r   )r™   Úmanagerr"   r#   r$   r$   r%   Ú
test_imageº   s   
rœ   c                  C   s4   t jdd t  ¡ } t | ¡}t |¡ t  ¡  d S )NT)r
   )r   r   Úgcfr   r[   r^   r`   )r"   Úpfr$   r$   r%   Ú
test_polarÆ   s
   

rŸ   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)	ÚmtransformsÚIdentityTransformÚidentityÚ	identity2ÚCompositeGenericTransformÚ	compositeÚTransformWrapperÚwrapperÚ
composite2)Úselfr$   r$   r%   Ú__init__Ï   s   

þ
þzTransformBlob.__init__N)Ú__name__Ú
__module__Ú__qualname__r¬   r$   r$   r$   r%   r    Î   s    r    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$   )rV   rP   r$   r$   r%   rZ   ç   s    z"test_transform.<locals>.<listcomp>)r    r   r[   r^   r©   Ú_childr§   Ú_parentsÚvaluesrª   Ú
input_dimsÚoutput_dims)Úobjrž   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)Úsharexr   r   r•   r+   )r   r•   )r   Úsubplotsr   r^   r[   r   rE   Úget_xlim)r"   Ú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)Ú	functions)
r   r»   Ú
inset_axesÚsecondary_xaxisr   ÚsquareÚsqrtr   r^   r[   r!   r$   r$   r%   Útest_inset_and_secondaryý   s   rÇ   Úcmapc                 C   s   t  | ¡ d S r¡   )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   )r   r   r_   r   r   r   Úseekro   )r"   ÚoutÚ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   Ú	host_axesÚtyper   r^   r[   Ú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 r¡   )rÏ   r   r^   r[   ÚmplÚcolorsÚ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 r¡   )
rÒ   rÓ   Úmake_norm_from_scaleÚscaleÚ
LogitScaleÚ	NormalizerÏ   r   r^   r[   )Ú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)Ú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ÚmrM   Úk)rß   r+   r-   r*   r0   rà   )	r   r   r‘   Úset_prop_cycler   r   r^   r[   Ú	get_color)r#   Úlr$   r$   r%   Útest_cycler)  s   rå   ):Úior   r|   r   r\   Únumpyr   ÚpytestÚ
matplotlibrÒ   r   Úmatplotlib.testingr   Úmatplotlib.testing.decoratorsr   Úmatplotlib.datesr   Úmatplotlib.linesr   Úmatplotlib.pyplotÚpyplotr   Úmatplotlib.transformsÚ
transformsr¢   Úmatplotlib.figurer   r   Úmpl_toolkits.axes_grid1r   r&   rQ   ÚstyleÚcontextri   rt   r„   rŒ   r“   r˜   rœ   rŸ   r    r¶   r¹   r¾   rÇ   ÚmarkÚparametrizeÚ
_colormapsr²   rÉ   rÍ   rÑ   rÕ   rÛ   rÞ   rå   r$   r$   r$   r%   Ú<module>   sV    
3



	

