o
    ‘i!¯  ã                   @   sf  d dl m Z  d dlZd dlZd dlZd dlmZ d dlZd dlm	Z	m
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mZmZm Z  d dl!m"Z"m#Z# ej$ddgd	d
d„ ƒZ%dd„ Z&e#dgƒdd„ ƒZ'e#dgƒdd„ ƒZ(e#dgƒdd„ ƒZ)e#dgƒdd„ ƒZ*e#dgƒdd„ ƒZ+e#dgƒdd„ ƒZ,e#d gƒd!d"„ ƒZ-e#d#gƒd$d%„ ƒZ.e#d&gƒd'd(„ ƒZ/e#d)gƒd*d+„ ƒZ0e#g d,¢ƒd-d.„ ƒZ1e#d/gƒd0d1„ ƒZ2d2d3„ Z3d4d5„ Z4d6d7„ Z5d8d9„ Z6ej7 8d:¡e"d;gd<d=d>„ ƒƒZ9d?d@„ Z:dAdB„ Z;e#dCgdDdEdFdG„ ƒZ<e#dHgdDd:dIdJdK„ ƒZ=e#dLgdDdEdMdN„ ƒZ>e#dOgdDdEdPdQ„ ƒZ?dRdS„ Z@dTdU„ ZAdVdW„ ZBdXdY„ ZCdZd[„ ZDd\d]„ ZEe#d^gdDdEd_d`„ ƒZFdadb„ ZGej7 8dc¡ddde„ ƒZHdfdg„ ZIej7 8dc¡dhdi„ ƒZJej7 8dc¡djdk„ ƒZKe#dlgƒdmdn„ ƒZLe#dogdDdcdIdpdq„ ƒZMdrds„ ZNdtdu„ ZOdvdw„ ZPdxdy„ ZQdzd{„ ZRd|d}„ ZSd~d„ ZTejU Vd€dd‚g¡dƒd„„ ƒZWd…d†„ ZXd‡dˆ„ ZYd‰dŠ„ ZZd‹dŒ„ Z[ddŽ„ Z\dd„ Z]d‘d’„ Z^d“d”„ Z_ej7 8dc¡d•d–„ ƒZ`d—d˜„ Zad™dš„ Zbd›dœ„ Zcddž„ ZddŸd „ Zed¡d¢„ Zfe#d£gd;gdDd:d¤d¥d£„ ƒZge"d;gd<d¦d§„ ƒZhd¨d©„ ZiejU Vdªd«d¬d­gg¡e"d;gd<e jd®d¯i¡d°d±„ ƒƒƒZkdS )²é    )ÚdatetimeN)ÚSimpleNamespace)Úassert_array_equalÚassert_array_almost_equal)Ú
CollectionÚLineCollectionÚEventCollectionÚPolyCollection)Úcheck_figures_equalÚimage_comparisonÚ
pcolormeshÚpcolor)Úparamsc                 C   s   | j S ©N)Úparam)Úrequest© r   úp/var/www/html/eduruby.in/lip-sync/lip-sync-env/lib/python3.10/site-packages/matplotlib/tests/test_collections.pyÚpcfunc   s   r   c               
   C   s®   t  g d¢¡} t  g d¢¡}d}d}d}d}g d¢}d}d	}t| |||||||d
}	t ¡ \}
}| |	¡ | d¡ | ||||||||dœ	}| dd¡ | dd¡ ||	|fS )z,Generate the initial collection and plot it.)ç        ç      ð?ç       @ç      @g      @g       @g      *@g      5@)g      A@g     €K@g     @V@Ú
horizontalé   ç      à?é   ©r   r   r   r   ÚsolidT)ÚorientationÚ
lineoffsetÚ
linelengthÚ	linewidthÚcolorÚ	linestyleÚantialiasedzEventCollection: default)	Ú	positionsÚextra_positionsr   r    r!   r"   r#   r$   r%   éÿÿÿÿé   r   )	ÚnpÚarrayr   ÚpltÚsubplotsÚadd_collectionÚ	set_titleÚset_xlimÚset_ylim)r&   r'   r   r    r!   r"   r#   r$   r%   ÚcollÚfigÚaxÚpropsr   r   r   Úgenerate_EventCollection_plot   sD   ù


ø

r6   ZEventCollection_plot__defaultc                  C   sÊ   t ƒ \} }}t||d |d |d |d ƒ tj |d | ¡ ¡ |d | ¡ ks+J ‚| ¡ s1J ‚|d | ¡ ks;J ‚|d | 	¡ ksEJ ‚| 
¡ dgksNJ ‚| ¡ g| ¡ ¢D ]}tj ||d ¡ qWd S )Nr&   r!   r    r   ©r   Nr#   )r6   Úcheck_segmentsr*   Útestingr   Úget_positionsÚget_orientationÚis_horizontalÚget_linelengthÚget_lineoffsetZget_linestyleÚ	get_colorÚ
get_colors)Ú_r2   r5   r#   r   r   r   Ú test__EventCollection__get_propsC   s    üÿrB   Z#EventCollection_plot__set_positionsc                  C   st   t ƒ \} }}t |d |d g¡}| |¡ tj || ¡ ¡ t|||d |d |d ƒ |  d¡ |  	dd¡ d S )	Nr&   r'   r!   r    r   zEventCollection: set_positionsr(   éZ   )
r6   r*   ÚhstackZset_positionsr9   r   r:   r8   r/   r0   ©Úspltr2   r5   Únew_positionsr   r   r   Ú$test__EventCollection__set_positions]   s   
ý
rH   Z#EventCollection_plot__add_positionsc                  C   s   t ƒ \} }}t |d |d d g¡}| ¡  | |d d ¡ | ¡  tj || ¡ ¡ t|||d |d |d ƒ |  	d¡ |  
dd	¡ d S )
Nr&   r'   r   r!   r    r   zEventCollection: add_positionsr(   é#   )r6   r*   rD   Úswitch_orientationZadd_positionsr9   r   r:   r8   r/   r0   rE   r   r   r   Ú$test__EventCollection__add_positionsk   s    

ÿü
rK   Z&EventCollection_plot__append_positionsc                  C   s€   t ƒ \} }}t |d |d d g¡}| |d d ¡ tj || ¡ ¡ t|||d |d |d ƒ |  d¡ |  	dd	¡ d S )
Nr&   r'   r   r!   r    r   z!EventCollection: append_positionsr(   rC   )
r6   r*   rD   Zappend_positionsr9   r   r:   r8   r/   r0   rE   r   r   r   Ú'test__EventCollection__append_positions}   s   

ÿü
rL   Z&EventCollection_plot__extend_positionsc                  C   sˆ   t ƒ \} }}t |d |d dd … g¡}| |d dd … ¡ tj || ¡ ¡ t|||d |d |d ƒ |  d¡ |  	dd	¡ d S )
Nr&   r'   r   r!   r    r   z!EventCollection: extend_positionsr(   rC   )
r6   r*   rD   Zextend_positionsr9   r   r:   r8   r/   r0   rE   r   r   r   Ú'test__EventCollection__extend_positions   s   
ÿü
rM   Z(EventCollection_plot__switch_orientationc                  C   sz   t ƒ \} }}d}| ¡  || ¡ ksJ ‚| ¡ rJ ‚| ¡ }t|||d |d |ƒ |  d¡ |  dd¡ |  dd¡ d S )	NÚverticalr!   r    z#EventCollection: switch_orientationr(   r)   r   r   )	r6   rJ   r;   r<   r:   r8   r/   r1   r0   )rF   r2   r5   Únew_orientationrG   r   r   r   Ú)test__EventCollection__switch_orientation   s   ý
rP   Z,EventCollection_plot__switch_orientation__2xc                  C   s€   t ƒ \} }}| ¡  | ¡  | ¡ }|d | ¡ ksJ ‚| ¡ s"J ‚tj |d |¡ t|||d |d |d ƒ |  	d¡ dS )zc
    Check that calling switch_orientation twice sets the orientation back to
    the default.
    r   r&   r!   r    z&EventCollection: switch_orientation 2xN)
r6   rJ   r:   r;   r<   r*   r9   r   r8   r/   rE   r   r   r   Ú,test__EventCollection__switch_orientation_2x®   s   ürQ   Z%EventCollection_plot__set_orientationc                  C   sx   t ƒ \} }}d}| |¡ || ¡ ksJ ‚| ¡ rJ ‚t||d |d |d |ƒ |  d¡ |  dd¡ |  dd	¡ d S )
NrN   r&   r!   r    z EventCollection: set_orientationr(   r)   r   r   )r6   Zset_orientationr;   r<   r8   r/   r1   r0   )rF   r2   r5   rO   r   r   r   Ú&test__EventCollection__set_orientationÃ   s   
ü
rR   Z$EventCollection_plot__set_linelengthc                  C   s`   t ƒ \} }}d}| |¡ || ¡ ksJ ‚t||d ||d |d ƒ |  d¡ |  dd¡ d S )Né   r&   r    r   zEventCollection: set_linelengthiìÿÿÿé   )r6   Zset_linelengthr=   r8   r/   r1   )rF   r2   r5   Znew_linelengthr   r   r   Ú%test__EventCollection__set_linelengthÔ   s   
ü
rU   Z$EventCollection_plot__set_lineoffsetc                  C   s`   t ƒ \} }}d}| |¡ || ¡ ksJ ‚t||d |d ||d ƒ |  d¡ |  dd¡ d S )Ng      Àr&   r!   r   zEventCollection: set_lineoffsetiúÿÿÿéüÿÿÿ)r6   Zset_lineoffsetr>   r8   r/   r1   )rF   r2   r5   Znew_lineoffsetr   r   r   Ú%test__EventCollection__set_lineoffsetã   s   
ü
rW   )Ú#EventCollection_plot__set_linestylerX   Z#EventCollection_plot__set_linewidthc                  C   sn   dddgfdddgfdfD ]'\} }}t ƒ \}}}|jdi | |i¤Ž t || ¡|ks,J ‚| d| › ¡ qd S )Nr$   Údashed)r   )ç      @rZ   )r"   é   r[   zEventCollection: set_r   )r6   Úsetr,   Zgetpr/   )ÚpropÚvalueÚexpectedrF   r2   rA   r   r   r   Útest__EventCollection__set_propò   s   

ýør`   ZEventCollection_plot__set_colorc                  C   sX   t ƒ \} }}t g d¢¡}| |¡ | ¡ g| ¡ ¢D ]	}tj ||¡ q|  d¡ d S )N)r   r   r   r   zEventCollection: set_color)	r6   r*   r+   Ú	set_colorr?   r@   r9   r   r/   )rF   r2   rA   Ú	new_colorr#   r   r   r   Ú test__EventCollection__set_color  s   
rc   c           
      C   sÎ   |   ¡ }| ¡ dks| ¡ dks|du rd}d}n| ¡ dkr$d}d}ntdƒ‚t|ƒD ]8\}}	|	d|f ||d  ks>J ‚|	d|f ||d  ksLJ ‚|	d|f || ksXJ ‚|	d|f || ksdJ ‚q,dS )	zp
    Test helper checking that all values in the segment are correct, given a
    particular set of inputs.
    r   ÚnoneNr   r   rN   z.orientation must be 'horizontal' or 'vertical'r   )Úget_segmentsÚlowerÚ
ValueErrorÚ	enumerate)
r2   r&   r!   r    r   ÚsegmentsZpos1Úpos2ÚiÚsegmentr   r   r   r8     s   ür8   c                  C   s4   t  g ¡} |  t ¡ ¡}t| ¡ tj ¡  ¡ ƒ d S r   )	ÚmcollectionsÚPathCollectionÚget_datalimÚmtransformsÚIdentityTransformr   Ú
get_pointsÚBboxÚnull)ÚcolZcol_data_limr   r   r   Útest_null_collection_datalim'  s
   
ÿrv   c                  C   sR   t  ¡ } t t ddg¡g¡}|  |¡ | t 	¡ ¡}t
| ¡ tj ¡  ¡ ƒ d S )N©r   r   )r   r   )r,   Úaxesrm   rn   ÚmpathÚPathr.   ro   rp   rq   r   rr   rs   rt   )r4   r2   Zcoll_data_limr   r   r   Útest_no_offsets_datalim.  s   
ÿr{   c                  C   sL   t  ¡  t  ¡ } |  ddgddg¡ | jj}|  g g ¡ | jj|ks$J ‚d S )Nr   r   )r,   Úfigurerx   ÚscatterÚdataLimÚbounds)r4   r   r   r   r   Útest_add_collection9  s   r€   Zmpl20Zpng)Ú
extensionsc                 C   s   g d¢}g d¢}t  ||¡\}}| ¡ }| ¡ }|  ¡ }| d¡ | d¡ d|_| ||¡ | ¡ }| d¡ | d¡ |j||ddd d S )N)g¥+ócÒ>gë®`8F×>gq[€;uÝ>gƒy7½/¤â>gÖÿ9Ì—ç>gWÇ!Üí>gÄ-–Iåò>g\p5ê÷>gd‘jDþ>g—y¹´K'?gÈ~Í+»=?gFEAL ®?g|™0j?gÇ÷GKe’?gÞ´=B?)r   çš™™™™™¹?g²ï§ÆKÇ?gsh‘í|?Õ?g‡ÙÎ÷Sã?gš™™™™™ñ?r   g…ëQ¸@gÂõ(\@g333333(@g      6@gÍÌÌÌÌÌC@g33333ÓQ@Úlogr   ÚoÚ )ÚmarkerÚls)	r*   ÚmeshgridÚflattenr-   Z
set_xscaleZ
set_yscaleZmarginsr}   Zplot)Úfig_testÚfig_refZx_valsZy_valsÚxÚyÚax_testÚax_refr   r   r   Útest_collection_log_datalimD  s   



r   c                  C   sä   t  ¡ } t d¡t d¡}}t ddd¡ dd¡ }}t  ||||¡}| | j¡j	dks/J ‚t  
¡  t  ¡ } t ddd¡}t dd	d¡}t ||¡\}}t ¡  d
d¡| j }t j||t |¡t |¡|d | jj	dkspJ ‚d S )Né   é
   r   éP   )r   r   g      @g      "@éûÿÿÿrT   éþÿÿÿé   é   é    ©Ú	transform)g      4@g      >@g      .@rZ   )r,   rx   r*   ÚarangeÚlinspaceÚreshapeZquiverro   Ú	transDatar   r|   rˆ   rp   ÚAffine2DÚ	translateÚsinÚcosr~   )r4   rŒ   r   ÚuÚvÚqÚtransr   r   r   Útest_quiver_limits^  s    r§   c                  C   s€   t  ¡ } t ddd¡}t ddd¡}t ||¡\}}t ¡  dd¡| j }t j	||t 
|¡t |¡|d t| jjd	d
d d S )Nr”   r’   rT   r•   r–   r—   r˜   r™   )rT   é   rS   é   r   ©Údecimal)r,   rx   r*   rœ   rˆ   rp   rŸ   r    rž   Zbarbsr¡   r¢   r   r~   r   )r4   rŒ   r   r¦   r   r   r   Útest_barb_limitso  s    

ÿr¬   z EllipseCollection_test_image.pngT)Úremove_textc               	   C   sž   t  ¡ \} }t d¡}t d¡}t ||¡\}}t | ¡ | ¡ f¡j}||d  }||d  }t |¡d }	t	j
|||	d||jdd}
| |
¡ | ¡  d S )Nr–   é   r(   rT   rŒ   rd   )ÚunitsÚoffsetsÚoffset_transformÚ
facecolors)r,   r-   r*   r›   rˆ   ZvstackÚravelÚTZ	ones_likerm   ÚEllipseCollectionrž   r.   Úautoscale_view)r3   r4   rŒ   r   ÚXÚYZXYZwwÚhhÚaaÚecr   r   r   Útest_EllipseCollection}  s   

þ
r¼   zpolycollection_close.png)r­   Ústylec                  C   sH  ddl m}  ddgddgddgddggddgddgddgddggddgddgddgddggddgddgddgddggg}t ¡ }| | |ƒ¡}g d¢}ttd	ƒƒ}tj|t	|ƒ d
d}| 
d¡ g }g }t||ƒD ]\}	}
| |	gt	|ƒ ¡ | |
gt	|ƒ ¡ qj| |¡ |j||dd | dd¡ | dd¡ | dd¡ d S )Nr   )ÚAxes3Dr   r   r   r   ç      @)ÚrÚgÚbr   Úkr[   ç      Ð?©r"   gffffffæ?r   )ÚzsZzdirr–   r®   )Zmpl_toolkits.mplot3dr¾   r,   r|   Zadd_axesÚlistÚrangerm   r	   ÚlenÚ	set_alphaÚzipÚextendra   Zadd_collection3dZ
set_xlim3dZ
set_zlim3dZ
set_ylim3d)r¾   Z	vertsQuadr3   r4   ÚcolorsZzposZpolyrÆ   ÚcsÚzÚcr   r   r   Útest_polycollection_close‘  s0   üÿ

rÑ   z regularpolycollection_rotate.pngc            	      C   sš   t jd d…d d…f \} }t  |  ¡ | ¡ g¡}t  ddt j t|ƒ¡}t ¡ \}}t	||ƒD ]\}}t
jdd||g|jd}|j|dd q/| ¡  d S )	Nr’   r   r   r–   )éd   )ÚsizesÚrotationr°   r±   T©Zautolim)r*   ZmgridZ	transposer‰   rœ   ÚpirÉ   r,   r-   rË   rm   ÚRegularPolyCollectionrž   r.   r¶   )	ÚxxÚyyZ	xy_pointsZ	rotationsr3   r4   ÚxyÚalpharu   r   r   r   Ú!test_regularpolycollection_rotate¶  s   þrÜ   zregularpolycollection_scale.pngc                  C   s`   G dd„ dt jƒ} t ¡ \}}dg}tjd g}| |||jd}|j|dd | g d¢¡ d S )	Nc                       s$   e Zd Z‡ fdd„Zdd„ Z‡  ZS )z:test_regularpolycollection_scale.<locals>.SquareCollectionc                    s    t ƒ jddtjd i|¤Ž d S )Nr–   rÔ   r¿   )r–   )ÚsuperÚ__init__r*   rÖ   )ÚselfÚkwargs©Ú	__class__r   r   rÞ   Ê  s    zCtest_regularpolycollection_scale.<locals>.SquareCollection.__init__c                 S   sJ   | j }d|jj }||jj |jj }||jj |jj }t ¡  	||¡S )z4Return transform scaling circle areas to data space.g      R@)
rx   r|   ÚdpiZbboxÚwidthZviewLimÚheightrp   rŸ   Úscale)rß   r4   Z
pts2pixelsZscale_xZscale_yr   r   r   Úget_transformÍ  s
   zHtest_regularpolycollection_scale.<locals>.SquareCollection.get_transform)Ú__name__Ú
__module__Ú__qualname__rÞ   rç   Ú__classcell__r   r   rá   r   ÚSquareCollectionÉ  s    rì   rw   r   )rÓ   r°   r±   TrÕ   )r(   r   r(   r   )	rm   r×   r,   r-   r*   rÖ   rž   r.   Zaxis)rì   r3   r4   rÚ   Zcircle_areasZsquaresr   r   r   Ú test_regularpolycollection_scaleÅ  s   ÿrí   c                  C   sn   t  ¡ \} }|jdgdgdgdd}| jt ¡ | jd tddd}| |¡\}}|s-J ‚t	|d	 dgƒ d S )
Nr   iè  T)Zpicker)rã   iE  éð   )rŒ   r   Úind)
r,   r-   r}   ZsavefigÚioÚBytesIOrã   r   Úcontainsr   )r3   r4   ru   Úmouse_eventÚfoundÚindicesr   r   r   Útest_pickingâ  s   rö   c                  C   sô   t  d¡} | d d …d f | d d d …f  }t ¡ \}}| |¡}| ¡  d\}}| ¡  ||f¡\} }t||| |d}| 	|¡\}	}
|	sFJ ‚t
|
d dgƒ d\}}| ¡  ||f¡\} }t||| |d}| 	|¡\}	}
|	spJ ‚t
|
d dgƒ d S )Nr–   ©r   r   ©ÚxdataÚydatarŒ   r   rï   r   ©ç      ø?rü   r[   )r*   r›   r,   r-   r   Údraw_without_renderingrç   rš   r   rò   r   )rŒ   r·   r3   r4   Úmeshrù   rú   r   ró   rô   rõ   r   r   r   Útest_quadmesh_containsì  s"   
 
rÿ   c                  C   s¦   ddgddgg} ddgddgg}t  ¡ \}}| | |dgg¡}| ¡  g d¢}|D ]'}|\}}}	| ¡  ||f¡\} }t||| |d}
| |
¡\}}||	u sPJ ‚q)d S )Nr   r(   r   ))ç      à¿rÄ   T)r   rÄ   F)r   rÄ   T)r   g      Ð¿Trø   )r,   r-   r   rý   rç   rš   r   rò   )rŒ   r   r3   r4   rþ   ZpointsÚpointrù   rú   r_   ró   rô   rõ   r   r   r   Útest_quadmesh_contains_concave  s   
ûr  c            	      C   s¶   t  d¡} | d d …d f | d d d …f  }t ¡ \}}| |¡}d |_| ¡  d\}}| ¡  ||f¡\} }t	||| |d}| 
|¡d u sGJ ‚| t  |j¡¡ t| 
|¡dgƒ d S )Nr–   r÷   rø   r   )r*   r›   r,   r-   r   Z_Arý   rç   rš   r   Úget_cursor_dataÚ	set_arrayÚonesÚshaper   )	rŒ   r·   r3   r4   rþ   rù   rú   r   ró   r   r   r   Útest_quadmesh_cursor_data  s   
 
r  c                  C   sx   g d¢} t  ¡ \}}| | | t d¡¡}| ¡  d\}}| ¡  ||f¡\} }t||| |d}t	| 
|¡t d¡ƒ d S )N)r   r   r   r   ©r®   r®   rû   rø   é	   )r,   r-   r   r*   r  rý   rç   rš   r   r   r  )rŒ   r3   r4   rþ   rù   rú   r   ró   r   r   r   Ú)test_quadmesh_cursor_data_multiple_points+  s   r
  c                   C   s,   t jg d¢g d¢dddgfd t  ¡  d S )N©r   r   r   r   r   )r$   )r,   r}   Údrawr   r   r   r   Útest_linestyle_single_dashes7  s    r  zsize_in_xy.pngc                  C   s`   t  ¡ \} }d\}}}d}ddg}tj|||d||jd}| |¡ | dd¡ | dd¡ d S )N)©r’   r’   r’   r   r  )rS   rS   rÚ   )r¯   r°   r±   r   r¨   )r,   r-   rm   rµ   rž   r.   r0   r1   )r3   r4   ÚwidthsZheightsZanglesÚcoordsÚer   r   r   Útest_size_in_xy<  s   
þ
r  c                 C   sŠ   g d¢}| j g d¢|d }}| j g d¢|d}| j g d¢|d}| j g d¢|d}t|d t|d t|d	 t|d
 t|d d S )N)é   é   é   )ÚredÚblueÚgreen)Úindex©r   r   r®   )r   rY   Zdashdot)TFT)Ú
edgecolors)r²   )Ú
linewidths)Ú
linestyles)Zantialiaseds)ZSeriesr   )Úpdr  r»   ZfcZlwr‡   rº   r   r   r   Útest_pandas_indexingM  s   



r  Údefaultc                  C   s”   t  g ¡} |  ddg¡ |  g d¢¡ |  ¡ dgd ksJ ‚|  ¡ g d¢d ks*J ‚|  g d¢¡ |  ¡ dgd ks<J ‚|  ¡ g d¢k ¡ sHJ ‚d S )Nú-r  r7   r©   r   )r!  r!  r!  r®   )rm   rn   Zset_linestylesZset_linewidthsZget_linestylesZget_linewidthsÚall©ru   r   r   r   Útest_lslw_bcast^  s   
r$  c                  C   sD   t ƒ } tjtdd |  d¡ W d   ƒ d S 1 sw   Y  d S )Nz"Do not know how to convert 'fuzzy'©ÚmatchZfuzzy)r   ÚpytestÚraisesrg   Zset_linestyle©rÐ   r   r   r   Útest_set_wrong_linestylel  s   "ÿr*  c                  C   óV   t  g ¡} |  ¡ d u sJ ‚t jg dd} |  ¡ dksJ ‚|  d¡ |  ¡ dks)J ‚d S )NÚround)ZcapstyleZbutt)rm   rn   Zget_capstyleÚset_capstyler#  r   r   r   Útest_capstyler  ó   

r.  c                  C   r+  )Nr,  )Z	joinstyleÚmiter)rm   rn   Zget_joinstyleÚset_joinstyler#  r   r   r   Útest_joinstyle|  r/  r2  zcap_and_joinstyle.pngc                  C   sÎ   t  ¡ \} }| ddg¡ | ddg¡ t g d¢¡}t dgdgdgg¡t g d¢g¡ }t d	¡}||d d …d d …d
f< ||d d …d d …df< t|g d¢d}| d¡ | 	d¡ | 
|¡ | d¡ d S )Nr   rü   ç      @)r   r   r   r   r   r   )r   r   r   )r®   r®   r   r   r   )r’   rS   rT   rÅ   r,  r0  z2Line collection with customized caps and joinstyle)r,   r-   r0   r1   r*   r+   Zzerosr   r-  r1  r.   r/   )r3   r4   rŒ   ZysZsegsZline_segmentsr   r   r   Útest_cap_and_joinstyle_image†  s   &



r4  zscatter_post_alpha.pngc                  C   s6   t  ¡ \} }|jtdƒtdƒtdƒd}| d¡ d S )Nr[   r)  r‚   )r,   r-   r}   rÈ   rÊ   )r3   r4   Úscr   r   r   Útest_scatter_post_alphaš  s   r6  c                  C   sŒ  t  d¡} | d }t d¡\}\}}|j| | | |d}|j| | | d}| |¡ t ¡  t| ¡ d d …df |ƒ t| ¡ d d …df |ƒ t d¡\}\}}|j| | g d¢|d}|j| | d|d}t ¡  t| ¡ d d …df |ƒ t| ¡ d d …df |ƒ t d¡\}\}}|j| | g d¢d	}| |¡ |j| | dd	}| |¡ t ¡  t| ¡ d d …df |ƒ t| ¡ d d …df |ƒ d S )
Nr[   r   )rÐ   rÛ   r)  r(   )rÀ   rÁ   rÂ   rÐ   Úm)r#   rÛ   rÀ   )r#   )	r*   r›   r,   r-   r}   rÊ   r  r   Úget_facecolors)rŒ   rÛ   r3   Úax0Úax1Zsc0Zsc1r   r   r   Útest_scatter_alpha_array¢  s.   



r;  c               	   C   s^  t j d¡ t j dd¡\} }t j d¡}t jjdddd}t jjdddd}t ¡ \}}|j| |||dd	dd
}|jdd\}}t	|ƒdksIJ ‚|g d¢ksQJ ‚t  
dd„ |D ƒ¡}	| t  d¡d ¡}
t|	|
ƒ |j||dd}|jdd\}}t	|ƒdks‚J ‚|j||dd}|jdddd\}}tdd„ |D ƒƒsŸJ ‚tdd„ |D ƒƒsªJ ‚|j||dd}|jddddd„ d\}}d d„ |D ƒ}t  t  
|t¡d ¡}t||ƒ |j||d!d}tjjdd"g d#¢d$}|j|d\}}t	|ƒt	|ƒksûJ ‚g d%¢}|j|ddd&\}}d'd„ |D ƒ|dd … ksJ ‚||||fD ]}| |¡ q|j ¡  d S )(Né!N,r   r’   r   r[   )Úsizei,  Zjetr„   )rÐ   ÚsÚcmapr†   r  z{x:g})Úfmt)Ú0Ú1Ú2Ú3Ú4c                 S   ó   g | ]}|  ¡ ‘qS r   )r?   ©Ú.0Úliner   r   r   Ú
<listcomp>Ì  ó    z7test_pathcollection_legend_elements.<locals>.<listcomp>r–   r   )Úlocr	  )ÚnumrÓ   r   r  )r]   rÛ   r#   c                 s   ó    | ]	}|  ¡ d kV  qdS )r   N)Z	get_alpharG  r   r   r   Ú	<genexpr>Ö  ó   € z6test_pathcollection_legend_elements.<locals>.<genexpr>c                 s   rN  )r  N)Zget_markerfacecolorrG  r   r   r   rO  ×  rP  z{x:.2f}c                 S   s   d|  S )Nr   r   )rŒ   r   r   r   Ú<lambda>Û  ó    z5test_pathcollection_legend_elements.<locals>.<lambda>)r]   rM  r@  Úfuncc                 S   rF  r   ©Zget_markersizerG  r   r   r   rJ  Ü  rK  r®   r‘   )r   r   r3  r®   r[   r©   r‘   r’   )ZnbinsZmin_n_ticksZsteps)r(   r   g33333³K@i  )rM  r]   r@  c                 S   s   g | ]}t |ƒ‘qS r   )Úfloat)rH  Úlr   r   r   rJ  è  rK  )r*   ÚrandomÚseedZrandÚrandintr,   r-   r}   Úlegend_elementsrÉ   r+   r?  r›   r   Zlegendr"  ÚsqrtrU  r   ÚmplZtickerZMaxNLocatorZ
add_artistÚcanvasr  )rŒ   r   rÐ   r>  r3   r4   r5  ÚhrV  rÍ   Zcolors2Úl1Zh2Zlab2Úl2Úl3ZactsizesZlabeledsizesÚl4rL  Zh5Zlab5ZlevelsZh6Zlab6r   r   r   Ú#test_pathcollection_legend_elements¿  sN   


ÿ

ÿ rc  c                  C   s2   t  g d¢¡} t| ƒ}t j | t  g d¢¡¡ d S )N)r®   r   r   r’   )r*   r+   r   r9   r   )Zarrr2   r   r   r   Útest_EventCollection_nosortð  s   rd  c                  C   s  t jdt jd ddd¡} t| ƒ}tt| ƒƒ}t|jƒt|jƒks#J ‚t|j|jƒD ]\}}t  	|j
|j
¡s8J ‚t  	|j|j¡sBJ ‚q*t jdtd}dd„ | D ƒ|d d …< t|ƒ}t|jƒt|jƒkseJ ‚t|j|jƒD ]\}}t  	|j
|j
¡szJ ‚t  	|j|j¡s„J ‚qld S )Nr“   ©Zdtyper’   r–   r   c                 S   s   g | ]}t d d„ |D ƒƒ‘qS )c                 s   s    | ]}t |ƒV  qd S r   ©Útuple)rH  r   r   r   r   rO    s   € z=test_collection_set_verts_array.<locals>.<listcomp>.<genexpr>rf  ©rH  rŒ   r   r   r   rJ    s    z3test_collection_set_verts_array.<locals>.<listcomp>)r*   r›   Údoubler   r	   rÇ   rÉ   Ú_pathsrË   Úarray_equalÚ	_verticesÚ_codesÚemptyÚobject)ZvertsZcol_arrZcol_listZapZlpZverts_tupleZcol_arr_tupleZatpr   r   r   Útest_collection_set_verts_array÷  s   þrp  c                  C   sx   g t dƒ¢} tƒ }| | ¡ tjtdd | d¡ W d   ƒ n1 s%w   Y  d| d< t | | ¡ ¡ 	¡ s:J ‚d S )Nr’   z^Image data of dtyper%  Zwrong_inputé-   r[   )
rÈ   r   r  r'  r(  Ú	TypeErrorr*   Ú	not_equalÚ	get_arrayÚany)ÚvalsrÐ   r   r   r   Útest_collection_set_array	  s   
ÿrw  c                  C   sœ   g d¢} d}t  t  | d¡t  d|gt| ƒ¡g¡}| t| ƒddg¡}t ¡ \}}t 	|j
|j¡}| t||d¡ |jddd t j | ¡ d	d
g¡ d S )N)r   r   r–   çš™™™™™É?r   r   r™   TF)ZscalexZscaleyr   r¿   )r*   Úcolumn_stackÚrepeatÚtilerÉ   r   r,   r-   rp   Zblended_transform_factoryrž   Ú	transAxesr.   r   r¶   r9   Úassert_allcloseÚget_xlim)Úarå   Zxy_pairsZ	line_segsÚfr4   r¦   r   r   r   Útest_blended_collection_autolim  s   &r  c                  C   sH   t  ¡ \} }| dd¡ tj | ¡ ddg¡ tj | ¡ ddg¡ d S )Nr   g¸…ëQ¸®¿g¸…ëQ¸®?)r,   r-   r}   r*   r9   r}  Úget_ylimr~  )r3   r4   r   r   r   Útest_singleton_autolim'  s   rƒ  ztransform, expected)rž   )r   g      @)r|  )gffffff@gš™™™™™	@c                 C   sV   t  ¡ \}}|jddt|| ƒd | dd¡ tj | ¡ |¡ tj | ¡ |¡ d S )Nr   r™   r®   )	r,   r-   r}   Úgetattrr*   r9   r}  r‚  r~  )rš   r_   r3   r4   r   r   r   Útest_autolim_with_zeros.  s
   	r…  c                 C   s.  t  d¡}t  d¡}t j d¡}t ¡ \}}t|| ƒ|||ƒ}tjtt	 
d¡d | | dd¡¡ W d   ƒ n1 s=w   Y  t  d¡ d	¡}tjtt	 
d
¡d | |¡ W d   ƒ n1 sdw   Y  tjtt	 
d¡d | | ¡ ¡ W d   ƒ n1 s…w   Y  t  d¡}tjtt	 
d¡d | |¡ W d   ƒ n1 s©w   Y  t  d¡}tjtt	 
d¡d | |¡ W d   ƒ n1 sÍw   Y  t  d¡}tjtt	 
d¡d | |¡ W d   ƒ n1 sñw   Y  t  d¡}t  d¡}t j d¡}t ¡ \}}|j|||dd}d S )Nr  r‘   )é   r’   zuFor X (11) and Y (8) with flat shading, A should have shape (7, 10, 3) or (7, 10, 4) or (7, 10) or (70,), not (10, 7)r%  r’   r†  é6   )r©   r	  ztFor X (11) and Y (8) with flat shading, A should have shape (7, 10, 3) or (7, 10, 4) or (7, 10) or (70,), not (6, 9)zsFor X (11) and Y (8) with flat shading, A should have shape (7, 10, 3) or (7, 10, 4) or (7, 10) or (70,), not (54,))r	  r©   r®   zwFor X (11) and Y (8) with flat shading, A should have shape (7, 10, 3) or (7, 10, 4) or (7, 10) or (70,), not (9, 6, 3))r	  r©   r–   zwFor X (11) and Y (8) with flat shading, A should have shape (7, 10, 3) or (7, 10, 4) or (7, 10) or (70,), not (9, 6, 4))r†  r’   r   zxFor X (11) and Y (8) with flat shading, A should have shape (7, 10, 3) or (7, 10, 4) or (7, 10) or (70,), not (7, 10, 2)Úgouraud©Zshading)r*   r›   rW  r,   r-   r„  r'  r(  rg   ÚreÚescaper  r   r³   r  r   )r   rŒ   r   rÏ   r3   r4   r2   r   r   r   Ú"test_quadmesh_set_array_validation>  sX   



ÿý

ÿý

ÿý


ÿý


ÿý


ÿý

rŒ  c                  C   s   t  g d¢g d¢¡\} }| | d d…d d…f }t | ||¡}| ¡  | ¡ dd …d d …f }t j | dk|dk@ | ¡} t | ||¡}| ¡  t	| 
¡ ƒdksQJ ‚t|| ¡ ƒ t j | dk|dk@ |¡}t | ||¡}| ¡  t	| 
¡ ƒdkszJ ‚t|| ¡ ƒ t j | d d…d d…f dk|d d…d d…f dk@ |¡}t |¡}| ¡  t	| 
¡ ƒdks±J ‚t|| ¡ ƒ tjtjdd | t  d¡¡ W d   ƒ n1 sÓw   Y  t  d	¡ d
¡}| |¡ | ¡  t	| 
¡ ƒd	ksóJ ‚t j |d¡}| |¡ | ¡  t	| 
¡ ƒdksJ ‚d S )Nr  )r   r   r   r®   r(   r   r   r[   zSetting a PolyQuadMeshr%  r©   )r®   r   r   r–   )r*   rˆ   r,   r   Úupdate_scalarmappableÚget_facecolorÚmaZmasked_wherer   rÉ   Ú	get_pathsr   r'  Zwarnsr\  ZMatplotlibDeprecationWarningr  r  r›   r   Zmasked_less)rØ   rÙ   ÚzzZquadmeshZquadmesh_fcZpolymeshr   r   r   Ú'test_polyquadmesh_masked_vertices_arrayn  sB   :
ÿþ

r’  c                 C   sd   g d¢}g d¢}t jdd}t  ||¡\}}tt| ƒ|||ƒ}t  |j|jg¡j}t| ¡ |ƒ d S )Nr  )r   r–   r©   )r   r   )r  )	r*   r  rˆ   r„  r,   Ústackr´   r   Zget_coordinates)r   rŒ   r   rÏ   rØ   rÙ   r2   r  r   r   r   Útest_quadmesh_get_coordinates£  s   r”  c                  C   s8  t  d¡} t  d¡}t  d¡ d¡}t ¡ \}}| | |t  |j¡¡}| |¡ |j	 
¡  t  | ¡ |¡s7J ‚| t  d¡¡ |j	 
¡  t  | ¡ t  d¡¡sQJ ‚t  d¡ d¡}t ¡ \}}|j| |t  |j¡dd}| |¡ |j	 
¡  t  | ¡ |¡s€J ‚| t  d¡¡ |j	 
¡  t  | ¡ t  d¡¡sšJ ‚d S )Nr–   r	  r  é   )r–   r–   rˆ  r‰  )r*   r›   r   r,   r-   r   r  r  r  r]  r  rk  rt  )rŒ   r   rÏ   r3   r4   r2   r   r   r   Útest_quadmesh_set_array¯  s(   







r–  c                 C   s²   t  ¡ \}}tjd }tjjddd}t|| ƒdgg||d}|j ¡  t	 
| ¡ dd d …f ||dƒƒ¡s8J ‚d\|_|_|j ¡  t	 
| ¡ dd d …f ||dƒƒ¡sWJ ‚d S )NZplasmar   r   )ÚvminÚvmax)r?  Únorm)r   r   )r,   r-   r\  Z	colormapsrÍ   Z	Normalizer„  r]  r  r*   rk  r8  r—  r˜  )r   r3   r4   r?  r™  r2   r   r   r   Útest_quadmesh_vmin_vmaxÍ  s   

(
,rš  c                 C   s<  t  d¡}t  d¡}t  d¡ d¡}|| ¡  }| ¡ }t d¡\}\}}t|| ƒ||||d}	t|| ƒ|||ƒ}
|
 |¡ t 	¡  t
|	 ¡ d d …df |ƒ t
|
 ¡ d d …df |ƒ t d¡\}\}}t|| ƒ||||d}	t|| ƒ|||ƒ}
|
 |¡ t 	¡  t
|	 ¡ d d …df |ƒ t
|
 ¡ d d …df |ƒ d S )Nr–   r	  r  r   )rÛ   r(   )r*   r›   r   Úmaxr³   r,   r-   r„  rÊ   r  r   r8  )r   rŒ   r   rÏ   rÛ   Z
alpha_flatr3   r9  r:  Zcoll1Zcoll2r   r   r   Útest_quadmesh_alpha_arrayÝ  s&   



rœ  c                 C   sp   t  ¡ \}}t|| ƒt d¡ d¡ƒ}tjtdd | 	ddg¡ | 
¡  W d   ƒ d S 1 s1w   Y  d S )Nr  )r®   r–   z^Data array shaper%  r   ç333333ã?)r,   r-   r„  r*   r›   r   r'  r(  rg   rÊ   r  )r   r3   r4   Úpcr   r   r   Útest_alpha_validationõ  s   
"þrŸ  c                  C   s¦   t j d¡ t j d¡} t j d¡}dt j d¡ }d| }g d¢}t ¡ \}}|j| ||d}|jd|dd	„ d
\}}	dd„ |D ƒ}
dd„ |
D ƒ}
t|
|dd dS )z|
    Ensure legend markers scale appropriately when label and size are
    inversely related.
    Here label = 5 / size
    r<  é2   r   r[   )rx  gš™™™™™Ù?r  gš™™™™™é?©r>  rÓ   c                 S   s   d|  S )Nr[   r   r¡  r   r   r   rQ    rR  z=test_legend_inverse_size_label_relationship.<locals>.<lambda>)r]   rM  rS  c                 S   rF  r   rT  rh  r   r   r   rJ    rK  z?test_legend_inverse_size_label_relationship.<locals>.<listcomp>c                 S   s   g | ]}d |d  ‘qS )r[   r   r   rh  r   r   r   rJ    s    rª   N)r*   rW  rX  r,   r-   r}   rZ  r   )r·   r¸   ÚCÚSZlegend_sizesr3   r4   r5  ZhandlesÚlabelsZhandle_sizesr   r   r   Ú+test_legend_inverse_size_label_relationshipþ  s   

ÿr¥  c                 C   sd  t t| ƒ} t d¡ dd¡}| |ddd}| ¡  t | ¡ ¡}| 	¡ | 
| ¡ ¡ƒ}t | ¡ d¡s5J ‚| |ƒ}| d¡ | d¡ | ¡  t | ¡ d¡sQJ ‚t | ¡ g d¢g¡s^J ‚| d¡ | ¡  t | ¡ g d	¢g¡stJ ‚| d ¡ | ¡  t | ¡ g d¢g¡sŠJ ‚| d ¡ | ¡  t | ¡ |¡sJ ‚| d ¡ | ¡  t | ¡ |¡s°J ‚t | ¡ d¡sºJ ‚| d ¡ | ¡  t | ¡ d¡sÍJ ‚t | ¡ |¡s×J ‚| |¡ | ¡  t | ¡ |¡sêJ ‚t | ¡ d¡sôJ ‚| |d
dd}| ¡  t | ¡ |¡s
J ‚t | ¡ g d¢g¡sJ ‚| |d
t d¡d}| ¡  t | ¡ |¡s1J ‚t | ¡ g d¢g¡s?J ‚| d ¡ | ¡  t | ¡ t d¡¡sVJ ‚t | ¡ g d¢g¡sdJ ‚| |d
t d¡d}| ¡  t | ¡ |¡s}J ‚t | ¡ g d¢g¡s‹J ‚| d ¡ | ¡  t | ¡ t d¡¡s¢J ‚t | ¡ g d¢g¡s°J ‚d S )Nr  r®   r–   r  rd   )r  r²   r   r   )r   r   r   r   )r   r   r   )r   r   r   )r  r®   )r  r–   )r„  r,   r*   r›   r   r  ÚmcolorsZto_rgba_arrayZ_get_default_facecolorZget_cmapr™  r³   Ú
same_colorÚget_edgecolorZset_facecolorZset_edgecolorrŽ  rÊ   rk  r  r  )r   rÏ   rž  Zface_defaultZmappedr   r   r   Útest_color_logic  sl   










 r©  c                  C   sŒ   t d dddg d¢d} |  ¡ d dksJ ‚t |  ¡ d¡sJ ‚|  ¡ dks'J ‚t |  ¡ g d¢g¡s4J ‚t d d d} t |  ¡ d¡sDJ ‚d S )	Ngš™™™™™@rÀ   r®   )r   r   r   r   )r"   Z	edgecolorZzorderr²   r   )Z	facecolorrd   )r   Zget_linewidthr¦  r§  r¨  Z
get_zorderrŽ  )Úlcr   r   r   Útest_LineCollection_args_  s   ÿr«  c                 C   sx   t  d¡ dd¡}tt| ƒ|ƒ}| | ¡ ¡ | ¡  | |¡ | ¡  t jdt jd ddd¡}| |¡ | ¡  d S )Nr  r®   r–   é$   re  )	r*   r›   r   r„  r,   r  r³   r  Zuint8)r   rÏ   rž  r   r   r   Útest_array_dimensionsm  s   

r­  c                  C   sB   t  t  ddd¡d¡j} t| gƒ}| ¡ \}t  | |k¡sJ ‚d S )Nr   r   é   )r   r   )r*   r{  rœ   r´   r   re   r"  )ri   rª  Zreadbackr   r   r   Útest_get_segments}  s   

r¯  c                  C   s~   t  ¡ } dg}tj|d}tj|dd}tj|d}| d¡ | | ¡j}| | ¡j}| | ¡j}||ks7J ‚||ks=J ‚d S )Nr   )rÓ   r  )rÓ   r°   )rp   rq   rm   ZCircleCollectionÚset_offsetsro   r   )ÚidentityrÓ   rt   ÚinitÚlateZnull_boundsZinit_boundsZlate_boundsr   r   r   Útest_set_offsets_late†  s   
r´  c                  C   sV   t  ¡  dd¡} tj| d}t ¡ }| | ¡ | | ¡   kr&| ¡ ks)J ‚ J ‚d S )Nr   )r±   )rp   rŸ   Úskewrm   r   Zset_offset_transformZget_offset_transform)rµ  r²  r³  r   r   r   Útest_set_offset_transform›  s
   
(r¶  c                  C   s¶   t  ddd¡} t  | ¡}| t  dd¡ t  d¡ }t ||¡}| ¡ }| t	t
||ƒƒ¡ t j || ¡ ¡ t ¡ \}}| ||¡}| ¡ }| t	t
||ƒƒ¡ t j || ¡ ¡ d S )Nr   r’   r[   é   r^  z
2021-11-29)r*   rœ   r¡   Ztimedelta64Z
datetime64r,   r}   Úget_offsetsr°  rÇ   rË   r9   r}  r-   )rŒ   r   Údr5  Zoff0r3   r4   r   r   r   Útest_set_offset_units¥  s   
rº  Útest_check_masked_offsets)Zbaseline_imagesr   r­   r½   c                  C   sˆ   t ddddddƒt ddddddƒt ddddddƒt dddddd	ƒt dddddd
ƒg} tjjg d¢g d¢d}t ¡ \}}| | |¡ d S )Niæ  r  rS   r–   é1   é4   é5   r‡  é7   é8   ©r   r   r®   r–   r[   )r   r   r   r   r   ©Úmask)r   r*   r  r+   r,   r-   r}   )Z
unmasked_xZmasked_yr3   r4   r   r   r   r»  ¹  s   ûc                 C   s–   t jjg d¢g d¢d}t  dd¡}| ¡ }| ||¡}| t j ||g¡¡ | g ¡ | 	g ¡ |  ¡ }| g d¢g d¢¡ | g ¡ | 	g ¡ d S )NrÁ  ©r   r   r   r   r   rÂ  r   r©   )r   r   r[   )
r*   r  r+   r›   Úadd_subplotr}   r°  ry  Z
set_xticksZ
set_yticks)r‹   rŠ   rŒ   r   rŽ   Úscatr   r   r   r   Útest_masked_set_offsetsÌ  s   


rÇ  c                  C   sž   t jjg d¢g d¢d} t  dd¡}t ¡ \}}| | |¡}t j | |g¡}| |¡ t	| 
¡ t|ƒƒs6J ‚t  | |g¡}| |¡ t	| 
¡ t|ƒƒsMJ ‚d S )NrÁ  rÄ  rÂ  r   r©   )r*   r  r+   r›   r,   r-   r}   ry  r°  Ú
isinstancer¸  Útype)rŒ   r   r3   r4   rÆ  Zmasked_offsetsZunmasked_offsetsr   r   r   Útest_check_offsets_dtypeÝ  s   

rÊ  ÚgapcolorZorangerÀ   rÃ   zlines.linewidthrT   c           
   	   C   s°   |   d¡}|  d¡}||fD ]}| dd¡ | dd¡ qtddƒ}g d¢}|j|dd||dd t|tƒr9|g}t|t 	|¡t 	|¡ƒD ]\}}}	|j
|dd|	|dd qEd S )Néo   r   r©   r   )ú:r!  z--r   )r$   rË  rÛ   )rÅ  r0   r1   rÈ   ZvlinesrÈ  ÚstrrË   Ú	itertoolsÚcycleZaxvline)
rŠ   r‹   rË  rŽ   r   r4   rŒ   r  Zgcolr‡   r   r   r   Útest_striped_linesí  s   



ÿþrÑ  )lr   rð   rÏ  rŠ  Útypesr   Únumpyr*   Znumpy.testingr   r   r'  Z
matplotlibr\  Zmatplotlib.pyplotZpyplotr,   Zmatplotlib.collectionsÚcollectionsrm   Zmatplotlib.colorsrÍ   r¦  Zmatplotlib.pathÚpathry   Zmatplotlib.transformsZ
transformsrp   r   r   r   r	   Zmatplotlib.testing.decoratorsr
   r   Zfixturer   r6   rB   rH   rK   rL   rM   rP   rQ   rR   rU   rW   r`   rc   r8   rv   r{   r€   r½   Úcontextr   r§   r¬   r¼   rÑ   rÜ   rí   rö   rÿ   r  r  r
  r  r  r  r$  r*  r.  r2  r4  r6  r;  rc  rd  rp  rw  r  rƒ  ÚmarkZparametrizer…  rŒ  r’  r”  r–  rš  rœ  rŸ  r¥  r©  r«  r­  r¯  r´  r¶  rº  r»  rÇ  rÊ  Z
rc_contextrÑ  r   r   r   r   Ú<module>   sò    
(












	



$








	
	
ÿ
1þ
05	

E	
ÿ



