o
    l’×iK»  ã                
   @   s‚  d dl mZ d dlmZ d dlZd dlZd dlm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Zd dlmZmZmZmZmZmZ d dlmZmZmZmZm Z  d d	l!m"Z"m#Z# d d
l$m%Z%m&Z&m'Z' d dl(m)Z* d dl+Z+e#dgdddd„ ƒZ,e#dgdddd„ ƒZ-e#dgddgdddd„ ƒZ.e+j/ 0dddg¡e#dgddgdd d!„ ƒƒZ1d"d#„ Z2e+j/ 0d$g d%¢¡e"dgdd&d'„ ƒƒZ3e"dgdd(d)„ ƒZ4e"ƒ d*d+„ ƒZ5d,d-„ Z6d.d/„ Z7e+j/ 0d0g d1¢¡d2d3„ ƒZ8e+j/ 0d0g d4¢¡d5d6„ ƒZ9d7d8„ Z:d9d:„ Z;d;d<„ Z<e#d=gddd>d?„ ƒZ=d@dA„ Z>e+j/ 0dBdCdDggdEfdFdGggdHfej?dIdJggdKfdIdLggdMfdNdNggdOfg¡dPdQ„ ƒZ@e#dRgdddSdT„ ƒZAe#dUgdddVdW„ ƒZBe#dXgdddYdZd[„ ƒZCe"dgdd\d]„ ƒZDd^d_„ ZEd`da„ ZFe#dbgdddcdd„ ƒZGe#degdddgdfdgdh„ ƒZHdidj„ ZIe#dkgdddYdldm„ ƒZJe#dngdddodp„ ƒZKe"dgddqdr„ ƒZLe#dsgddgdddtdudv„ ƒZMe#dwgdddYdxdy„ ƒZNdzd{„ ZOe#d|gdddYd}d~„ ƒZPdd€„ ZQdd‚„ ZRdƒd„„ ZSd…d†„ ZTd‡dˆ„ ZUd‰dŠ„ ZVe+j/ 0d‹ee WdŒ¡e Wd¡e WdŽ¡ Xd¡fe e WdŒ¡e Wd¡e Wd¡ Xd‘¡fg¡d’d“„ ƒZYd”d•„ ZZd–d—„ Z[e#d˜gddd™dš„ ƒZ\e#d›gdddœd„ ƒZ]dždŸ„ Z^d d¡„ Z_e#d¢gdd£d¤d¥d¦„ ƒZ`e#d§gddd¨d©„ ƒZadªd«„ Zbe#d¬gddd­d®„ ƒZce#d¯ge d¡ d°krxd nd±ddd²d³d´„ ƒZedµd¶„ Zfe+j/ 0d·d¸d¹„ dº g¡ D ƒ¡d»d¼„ ƒZhe#d½gdddYd¾d¿„ ƒZie#dÀgdddYdÁdÂ„ ƒZje#dÃgdddYdÄdÅ„ ƒZke+j/ 0dÆejlejmdÇdÈ„ dÉdÈ„ g¡dÊdË„ ƒZndÌdÍ„ ZodÎdÏ„ ZpdÐdÑ„ ZqdÒdÓ„ Zre+j/ 0dÔdÕdÖg¡e+j/ 0d×dØdÙg¡dÚdÛ„ ƒƒZsdÜdÝ„ ZtdÞdß„ Zudàdá„ Zvdâdã„ Zwe"ƒ dädå„ ƒZxdædç„ Zyej zd¡dèdé„ ƒZ{e#dêgddgdddtdëdì„ ƒZ|G dídî„ dîej}ƒZ~dïdð„ Zdñdò„ Z€e"dgddódô„ ƒZe+j/ 0dõdNdIg¡e"dgddöd÷„ ƒƒZ‚e"ƒ dødù„ ƒZƒe#dúgdddûdü„ ƒZ„e#dýgdde d¡ dþv rydÿnd d dd„ ƒZ…e+j/ †d¡e+j/ 0dddg¡e+j/ 0dg d¢g d	¢g¡e"d
ddd„ ƒƒƒƒZ‡e"dgddd„ ƒZˆdS (  é    )Ú	ExitStack)ÚcopyN)ÚPath)Úassert_array_equal)ÚImage)ÚcolorsÚimageÚpatchesÚpyplotÚstyleÚrcParams)Ú	AxesImageÚ	BboxImageÚFigureImageÚNonUniformImageÚPcolorImage)Úcheck_figures_equalÚimage_comparison)ÚBboxÚAffine2DÚTransformedBboxÚimage_interpsÚmpl20)r   c                  C   s†   dt jd< t d¡ dd¡} t  d¡\}\}}}|j| dd | d	¡ | d¡ |j| d
d | d
¡ |j| dd | d¡ dS )z5Make the basic nearest, bilinear and bicubic interps.é   ztext.kerning_factoréd   é   é   é   Únearest©Úinterpolationzthree interpolationsÚbilinearÚbicubicN)	Úpltr   ÚnpÚarangeÚreshapeÚsubplotsÚimshowÚ	set_titleÚ
set_ylabel)ÚXÚfigÚax1Úax2Úax3© r0   úY/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/matplotlib/tests/test_image.pyÚtest_image_interps   s   



r2   zinterp_alpha.pngT)Úremove_textc                  C   sf   t  dd¡\} \}}t d¡}t d¡|d< t tjdtjd¡|d< |j|dd	 |j|d
d	 dS )z:Test the interpolation of the alpha channel on RGBA imagesé   é   )r   r   é   ©r   r   ).r4   ©Údtype©.r   Únoner   r!   N)r#   r'   r$   ÚzerosÚonesÚtrilÚuint8r(   )r,   ÚaxlÚaxrÚimgr0   r0   r1   Útest_alpha_interp.   s   
rC   Úinterp_nearest_vs_noneÚpdfÚsvg)Ú
extensionsr3   c                  C   s|   dt d< tjg d¢g d¢gg d¢g d¢ggtjd} t dd	¡\}\}}|j| d
d | d¡ |j| dd | d¡ dS )z5Test the effect of "nearest" and "none" interpolationr   úsavefig.dpi)éÚ   é¥   é    )éz   ég   éî   )é   éÿ   r   )rP   éc   éG   r8   r4   r5   r;   r   zinterpolation noner   zinterpolation nearestN)r   r$   Úarrayr?   r#   r'   r(   r)   )r+   r,   r-   r.   r0   r0   r1   Útest_interp_nearest_vs_none;   s   ÿÿ
rT   ÚsuppressCompositeFÚfigimageÚpng)rG   c                 C   s  t jddd}| |_t t d¡d t d¡d ¡\}}t |d |d  ||  ¡}t d|d  d|d   ¡}||d  }|j|d	d	d
d |j|d d d…d d …f d	dd
d |j|d d …d d d…f dd	d
d |j|d d d…d d d…f ddd
d d S )N©r5   r5   r   ©ÚfigsizeÚdpig      Y@r5   r   é2   r   r   Úlower)ÚxoÚyoÚoriginéÿÿÿÿ)r#   ÚfigurerU   r$   Úix_r%   ÚsinrV   )rU   r,   ÚxÚyÚzÚcrB   r0   r0   r1   Útest_figimageM   s   $$$*ri   c                  C   sD   t  ¡ \} }| g d¢¡ t ¡ }|  |¡ | d¡ t  |¡ d S )N)r4   r5   r   r   )r#   r'   ÚplotÚioÚBytesIOÚsavefigÚseekÚimread)r,   ÚaxÚbufferr0   r0   r1   Útest_image_python_io]   s   

rr   z!img_size, fig_size, interpolation))r   r5   Úhanning)r   r   r   )r   é
   r   )r   ç333333@rs   )r   g333333"@r   c           	      C   sš   t j d¡ tjd }t j t|| ƒt|| ƒ¡}| |fD ]}| ||¡ q|  ¡ }| 	g d¢¡ |j
|dd | ¡ }| 	g d¢¡ |j
||d d S )Né!N,rH   ©r   r   r4   r4   Úantialiasedr   )r$   ÚrandomÚseedr#   r   ÚrandÚintÚset_size_inchesr'   Úset_positionr(   )	Úfig_testÚfig_refÚimg_sizeÚfig_sizer    r[   ÚAr,   rp   r0   r0   r1   Útest_imshow_antialiasedf   s   
r„   c                 C   s¶   t j d¡ tjd }t j t|d ƒt|d ƒ¡}| |fD ]}| dd¡ q|  ¡ }|j	|dd | 
ddg¡ | ddg¡ | ¡ }|j	|d	d | 
ddg¡ | ddg¡ d S )
Nrv   rH   r   ru   rx   r   rt   r   r   )r$   ry   rz   r#   r   r{   r|   r}   r'   r(   Úset_xlimÚset_ylim)r   r€   r[   rƒ   r,   rp   r0   r0   r1   Útest_imshow_zoom~   s   
r‡   c                 C   sŽ   t  d¡ ttƒjd }ttƒjd }|  d¡}|d  t |¡¡ |d  t |¡¡ | d¡}|d  t	 
|¡¡ |d  t	 
|¡¡ d S )NÚdefaultú%baseline_images/pngsuite/basn3p04.pngú%baseline_images/test_image/uint16.tifr5   r   r4   )r   Úuser   Ú__file__Úparentr'   r(   r   Úopenr#   ro   )r   r€   Úpng_pathÚ	tiff_pathÚaxsr0   r0   r1   Útest_imshow_pil   s   


r’   c                  C   sF   t  tj tj t¡ddd¡¡} | jtj	ksJ ‚t 
| ¡dks!J ‚d S )NÚbaseline_imagesÚ
test_imagez
uint16.tifé €ÿ)r#   ro   ÚosÚpathÚjoinÚdirnamerŒ   r9   r$   Úuint16Úsum©rB   r0   r0   r1   Útest_imread_pil_uint16   s
   ÿr   c                  C   s:   t  ttƒjd ¡} | jtjksJ ‚t | ¡dksJ ‚d S )NrŠ   r•   )	r#   ro   r   rŒ   r   r9   r$   rš   r›   rœ   r0   r0   r1   Útest_imread_fspath¤   s
   ÿrž   Úfmt)rW   ÚjpgÚjpegÚtiffc                 C   sÄ   | dv}t j d¡ t j dd¡}t ¡ }tj||| dd t ¡ }tj||| dd | d¡ tj	|| d}| d¡ tj	|| d}|j
ddd	| fksOJ ‚|j
ddd	| fks[J ‚t||ƒ d S )
N)r    r¡   r4   i@  r5   )Úformatr[   r   r   ©r£   r   )r$   ry   rz   r{   rk   rl   r#   Úimsavern   ro   Úshaper   )rŸ   Ú	has_alphaÚdataÚ	buff_dpi1Úbuff_dpi100Úarr_dpi1Ú
arr_dpi100r0   r0   r1   Útest_imsave«   s   	

r­   )rW   rE   ÚpsÚepsrF   c                 C   s&   t jttjƒt ddgg¡| d d S )Nr   r4   r¤   )r#   r¥   r   r–   Údevnullr$   rS   )rŸ   r0   r0   r1   Útest_imsave_fspathÎ   s   &r±   c                  C   s”   t j d¡ dD ]?} t j ddd¡}t ¡ }tj||| dd | d¡ t 	|¡}d|  
d	¡}| d
kr;|d d d… }d|  
d	¡}t||ƒ qd S )Nr4   )r]   Úupperé   r6   rW   )r`   r£   r   rP   r?   r]   ra   )r$   ry   rz   r{   rk   rl   r#   r¥   rn   ro   Úastyper   )r`   r¨   ÚbuffÚarr_bufr0   r0   r1   Útest_imsave_color_alphaÓ   s   

ðr·   c                  C   sh   ddl m}  t ¡ }| ƒ }| dd¡ tj|ddgddggdd	|id
 t |¡}|j	d dks2J ‚d S )Nr   )ÚPngInfoÚSoftwareÚtestr4   r5   r   rW   Úpnginfo©r£   Ú
pil_kwargs)
ÚPIL.PngImagePluginr¸   rk   rl   Úadd_textr#   r¥   r   rŽ   Úinfo)r¸   Úbufr»   Úimr0   r0   r1   Útest_imsave_pil_kwargs_pngì   s   ÿ
rÃ   c                     s€   ddl m‰  t ¡ } ddi}tj| ddgddggd|d	 t|ƒdks%J ‚t | ¡}‡ fd
d„|j	 
¡ D ƒ}|d dks>J ‚d S )Nr   )ÚTAGS_V2Údescriptionz
test imager4   r5   r   r¢   r¼   c                    s   i | ]
\}}ˆ | j |“qS r0   )Úname)Ú.0ÚkÚv©ÚTAGSr0   r1   Ú
<dictcomp>þ   ó    z/test_imsave_pil_kwargs_tiff.<locals>.<dictcomp>ÚImageDescription)ÚPIL.TiffTagsrÄ   rk   rl   r#   r¥   Úlenr   rŽ   Útag_v2Úitems)rÁ   r½   rÂ   Útagsr0   rÊ   r1   Útest_imsave_pil_kwargs_tiff÷   s   
rÔ   Úimage_alphac                  C   sd   t j d¡ t j dd¡} t dd¡\}\}}}|j| ddd |j| ddd |j| dd	d d S )
Nr   r   r4   r   ç      ð?r;   )Úalphar    ç      à?r   )r$   ry   rz   r{   r#   r'   r(   )ÚZr,   r-   r.   r/   r0   r0   r1   Útest_image_alpha  s   rÚ   c            
      C   sp  ddl m}  t ¡ \}}|jt d¡ dd¡dd}d\}}|j 	||g¡\}}| d|j
||ƒ}| |¡d	ks9J ‚d
\}}|j 	||g¡\}}| d|j
||ƒ}| |¡d u sXJ ‚| ¡  |jt d¡ dd¡dd}d\}}|j 	||g¡\}}| d|j
||ƒ}| |¡d	ks‰J ‚t ¡ \}}|jt d¡ dd¡g d¢d}d\}}|j 	||g¡\}}| d|j
||ƒ}| |¡dks¾J ‚d\}}|j 	||g¡\}}| d|j
||ƒ}| |¡d u sÝJ ‚d\}}|j 	||g¡\}}| d|j
||ƒ}| |¡d u süJ ‚tƒ  d¡ d¡}	|jt d¡ dd¡|	|j d}d\}}|j 	||g¡\}}| d|j
||ƒ}| |¡d	ks6J ‚d S )Nr   ©Ú
MouseEventr   rt   r²   )r`   ©r6   r6   Úmotion_notify_eventé,   )g333333$@r6   r]   )r   rØ   r   rØ   ©Úextent)ç      Ð?râ   é7   )ç      è?râ   )ç{®Gáz„?g{®Gáz„¿r5   rØ   )Ú	transform)r   rt   )Úmatplotlib.backend_basesrÜ   r#   r'   r(   r$   r%   r&   Ú	transDataræ   ÚcanvasÚget_cursor_dataÚclearr   ÚscaleÚrotate)
rÜ   r,   rp   rÂ   re   rf   ÚxdispÚydispÚeventÚtransr0   r0   r1   Útest_cursor_data  sN   
 ÿrò   z
data, texti'  i'  z[10001.000]g°rh‘í|¿?g/Ý$•ï?z[0.123]r4   r5   ú[]g     ð?z[1.0000000000000000]ra   z[-1.0000000000000000]c           	      C   sb   ddl m} t ¡ \}}| | ¡}|j ddg¡\}}|d|j||ƒ}| | 	|¡¡|ks/J ‚d S )Nr   rÛ   rÞ   )
rç   rÜ   r#   r'   r(   rè   ræ   ré   Úformat_cursor_datarê   )	r¨   ÚtextrÜ   r,   rp   rÂ   rî   rï   rð   r0   r0   r1   Útest_format_cursor_dataT  s   	
rö   Ú
image_clipc                  C   sF   ddgddgg} t  ¡ \}}| | ¡}tjdd|jd}| |¡ d S )Nr4   r5   r   r6   ©r   r   )Úradiusræ   )r#   r'   r(   r	   ÚCirclerè   Úset_clip_path)Údr,   rp   rÂ   Úpatchr0   r0   r1   Útest_image_clipg  s
   
rþ   Úimage_cliprectc                  C   sN   t  ¡ \} }ddgddgg}|j|dd}tjddd|jjd}| |¡ d S )	Nr4   r5   r   r6   ©r   r   r   r   rà   ©r4   r4   )ÚxyÚwidthÚheightræ   )r#   r'   r(   r	   Ú	RectangleÚaxesrè   rû   )r,   rp   rü   rÂ   Úrectr0   r0   r1   Útest_image_cliprectq  s   ÿr  r(   )r3   r   c                  C   sH   t  ¡ \} }t d¡ d¡}|j|ddd | dd¡ | dd¡ d S )Nr   ©rt   rt   r!   ©r4   r5   r4   r5   ©r    rá   r   r   )r#   r'   r$   r%   r&   r(   r…   r†   ©r,   rp   Úarrr0   r0   r1   Útest_imshow}  s
   r  c                 C   s†   t  d¡ d¡}| ¡ }|j|d d …d d …df ddd | dd¡ | dd¡ |  ¡ }|j|ddd | dd¡ | dd¡ d S )Nr   )rt   rt   r4   r   r!   r
  r  r   )r$   r%   r&   r'   r(   r…   r†   )r   r€   r  rp   r0   r0   r1   Útest_imshow_10_10_1†  s   "r  c                  C   óV   t  ¡ \} }t d¡ d¡}t t¡ | |¡ W d   ƒ d S 1 s$w   Y  d S )NéÈ   )rt   rt   r5   ©	r#   r'   r$   r%   r&   ÚpytestÚraisesÚ	TypeErrorr(   r  r0   r0   r1   Útest_imshow_10_10_2•  ó
   "ÿr  c                  C   r  )Niô  )rt   rt   r   r  r  r0   r0   r1   Útest_imshow_10_10_5œ  r  r  Úno_interpolation_originc                  C   sP   t  d¡\} }|d jt d¡ d¡ddd |d jt d¡ d¡dd	 d S )
Nr5   r   r   )r5   r\   r]   r;   )r`   r    r4   r   )r#   r'   r(   r$   r%   r&   )r,   r‘   r0   r0   r1   Útest_no_interpolation_origin£  s
   ÿ"r  Úimage_shift)r3   rG   c                  C   sT   dd„ t ddƒD ƒ} d}d}t ¡ \}}|j| t ¡ d||ddfd | d	¡ d S )
Nc                    s$   g | ]‰ ‡ fd d„t ddƒD ƒ‘qS )c                    s   g | ]
}d | d ˆ   ‘qS )r4   r0   )rÇ   re   ©rf   r0   r1   Ú
<listcomp>­  rÍ   z/test_image_shift.<locals>.<listcomp>.<listcomp>r4   r   )Úrange)rÇ   r0   r  r1   r  ­  s   $ z$test_image_shift.<locals>.<listcomp>r4   r   gMòãûk&Ag˜Šäûk&Ar;   )Únormr    rá   Úauto)r  r#   r'   r(   r   ÚLogNormÚ
set_aspect)ÚimgDataÚtMinÚtMaxr,   rp   r0   r0   r1   Útest_image_shift«  s   
ÿr&  c                  C   s  t jddgd} | jg d¢dd}t t d¡d¡ dd	¡}|j|d
g d¢ddd}d }}| | |g¡ | 	| |g¡ | 
g ¡ | g ¡ t ¡ }| j|dd | d¡ t  |¡}t|d d …df ƒ\}}}	}
t|d d …df ƒ\}}}	}
|dksJ dƒ‚d S )Nr4   ©rZ   rw   F)Úframeoné   é   r   é	   r²   ©éöÿÿÿrt   r-  rt   r;   Úgray)r`   rá   r    Úcmapr5   )r   r4   r   )Ú	facecolorr   ra   r   z.Expected a non-green edge - but sadly, it was.)r#   rb   Úadd_axesr$   Útiler%   r&   r(   r…   r†   Ú
set_xticksÚ
set_yticksrk   rl   rm   rn   ro   r›   )r,   rp   r¨   rÂ   re   rf   rÁ   ÚrÚgÚbÚar0   r0   r1   Útest_image_edges·  s$   ÿ



r9  Úimage_composite_backgroundc                  C   s^   t  ¡ \} }t d¡ dd¡}|j|g d¢d |j|g d¢d | d¡ | ddg¡ d S )	Nr)  r6   r   )r   r5   r*  r   rà   )r6   r   r*  r   )r4   r   r   rØ   r   )r#   r'   r$   r%   r&   r(   Úset_facecolorr…   r  r0   r0   r1   Útest_image_composite_backgroundÓ  s   
r<  Úimage_composite_alphac                  C   sŠ  t  ¡ \} }t d¡}d|dd…dd…df< t t ddd¡t ddd¡ddd… f¡|dd…dd…df< t d	¡}d|dd…dd…df< d|dd…dd…df< t t ddd¡t ddd¡ddd… f¡dd…tjf |dd…dd…df< |j|g d
¢dd |j|g d¢dd |j|g d¢d |j|g d¢d |j|g d¢dd |j|g d¢dd | d¡ | 	ddg¡ | 
ddg¡ dS )z{
    Tests that the alpha value is recognized and correctly applied in the
    process of compositing images together.
    )é   é   r6   r4   Nr   çš™™™™™ñ?çš™™™™™¹?ra   r   )r?  r>  r6   )r4   r5   r   r   ç333333Ó?)rá   r×   )r5   r   r   r   ç333333ã?)r   r6   r   r   rà   )r   r   r4   r5   )r   r   r5   r   )r   r   r   r6   )r   rØ   r   r4   r   )r#   r'   r$   r<   Úconcatenater%   Únewaxisr(   r;  r…   r†   )r,   rp   r  Úarr2r0   r0   r1   Útest_image_composite_alphaÞ  s.   
$ÿ
$ÿÿ
rG  c                 C   sp   t  d¡ d¡}| |fD ]&}| ¡ }|j|tj g d¢¡|jfd |j|tj g d¢¡|jfd qd|_	d S )Nr+  ©r   r   )rø   ©r   r4   )r4   r   )Ú	clip_path)r  ©r4   r5   )r5   r4   T)
r$   r%   r&   Úadd_subplotr(   Úmplr—   r   rè   rU   )r   r€   Útr,   rp   r0   r0   r1   Ú#test_clip_path_disables_compositingù  s   ÿ
ÿ
rO  Úrasterize_10dpi)rG   r3   r   c                  C   sÔ   t  ddgddgg¡} tjdddd\}}|d  | ¡ |d jddgddgdd	d
 |d jddd |d jddgddgdd |d jddd |D ]}| g ¡ | g ¡ |j	d d …  
d¡ qMdtd< d S )Nr4   r5   r   r6   )r   r4   r'  r   g      4@T)Ú	linewidthÚ
rasterizedrI  )ra   r5   ©ÚxlimÚylim)rQ  Frt   rH   )r$   Úasarrayr#   r'   r(   rj   Úsetr3  r4  ÚspinesÚset_visibler   )rB   r,   r‘   rp   r0   r0   r1   Útest_rasterize_dpi  s   	

rZ  Úbbox_image_invertedc                  C   sÈ   t  d¡ d¡} t ¡ \}}tttddgddggƒ|jƒdd}| 	| ¡ | 
d¡ | dd¡ | dd¡ | |¡ t  d¡} tttdd	gd
dggƒ|jjƒdd}| 	| ¡ | 
d¡ | |¡ d S )Nr   r	  r   r   r   Frt   rA  çš™™™™™É?rB  râ   )r$   r%   r&   r#   r'   r   r   r   rè   Úset_dataÚset_clip_onr…   r†   Ú
add_artistÚidentityrb   ÚtransFigure)r   r,   rp   Úbbox_imr0   r0   r1   Útest_bbox_image_inverted$  s(   þ



ÿþ

rc  c                  C   s.  t  g d¢g d¢g d¢g d¢g¡} tjddd\}}| g d¢¡ | d	d
¡ | d	d
¡ |j| g d¢dd}|j 	¡  |jj
}| |¡}t| ¡ ddgddggƒ tjddd\}}| g d¢¡ | d
d¡ | d	d
¡ |j| g d¢d|jd}|j 	¡  |jj
}| |¡}t| ¡ ddgddggƒ d S )N)râ   rä   rÖ   rä   )rA  gÍÌÌÌÌÌä?rØ   çš™™™™™Ù?)rC  rB  ç        r\  )çffffffæ?çÍÌÌÌÌÌì?rd  rC  r	  r   rY   rw   r   r4   )rd  rf  r\  rg  r   )rá   r    i  r  i¼  i„  r5   )rá   r    ræ   )r$   rS   r#   r'   r~   r…   r†   r(   ré   ÚdrawÚrendererÚget_window_extentr   Ú
get_pointsÚ	transAxes)rÂ   r,   rp   Úim_objri  Úim_bboxr0   r0   r1   Ú$test_get_window_extent_for_AxisImage=  s4   ÿ
ÿ


þ

ro  zzoom_and_clip_upper_origin.pngc                  C   sF   t  d¡} |  d¡} t ¡ \}}| | ¡ | dd¡ | dd¡ d S )Nr   r	  ç       @ç      à¿)r$   r%   r&   r#   r'   r(   r†   r…   )r   r,   rp   r0   r0   r1   Útest_zoom_and_clip_upper_origin`  s   


rr  c                  C   s   t  ¡ } t| ƒ}| d¡ d S )NÚBlues)r#   Úgcar   Úset_cmap©rp   rÂ   r0   r0   r1   Útest_nonuniformimage_setcmapl  s   rw  c                  C   s"   t  ¡ } t| ƒ}| t  ¡ ¡ d S ©N)r#   rt  r   Úset_normÚ	Normalizerv  r0   r0   r1   Útest_nonuniformimage_setnormr  s   r{  c                  C   sn   t jddd} tdƒD ]}t  ddd¡d | |d d …f< qt dd¡}| |  ¡ ¡ t 	¡ \}}| 
|¡ d S )	Nr	  r?   r8   rt   re  rÖ   rP   ÚL)r$   r=   r  Úlinspacer   ÚnewÚputdataÚflattenr#   r'   r(   )ÚimdÚirÂ   r,   rp   r0   r0   r1   Útest_jpeg_2dx  s    rƒ  c                  C   sª   t jddd tjdtd} t ddd¡| d< t  | ¡ t ¡ }t j	|d	d
dd | 
d¡ t |¡}t| d¡ƒ}d|  krEdksHJ ‚ J ‚| d¡}|dksSJ ‚d S )Nr  é,  rY   )r„  r„  r6   r8   re  rÖ   r:   Úredr    )r0  r£   r[   r   é   é¯   éÒ   rø   )éþ   r   r   )r#   rb   r$   r<   Úfloatr}  rV   rk   rl   rm   rn   r   rŽ   rÐ   Ú	getcolorsÚgetpixel)rÂ   rµ   r   Ú
num_colorsÚcorner_pixelr0   r0   r1   Útest_jpeg_alphaƒ  s   



r  c                  C   óP   t  ¡ } t| ƒ}tjdtd d¡}| |¡ d|d< |jd dks&J dƒ‚d S ©Nr)  r8   ©r6   r   çÍÌÌÌÌÌ#@rø   r   úvalue changed)	r#   rt  r   r$   r%   rŠ  r&   r]  Ú_A)rp   rÂ   rg   r0   r0   r1   Útest_axesimage_setdata›  ó   
r–  c                  C   r  r‘  )	r#   Úgcfr   r$   r%   rŠ  r&   r]  r•  )r,   rÂ   rg   r0   r0   r1   Útest_figureimage_setdata¤  r—  r™  zimage_cls,x,y,aç      @ç      @g      (@r’  g      @)r   r5   c                 C   sš   t  ¡ }| |ƒ}| |||¡ d |d<  |d< |d< |jd |jd   kr6|jd   kr6dks;J dƒ‚ J dƒ‚| ||| g |j¢d‘R ¡¡ d S )Nr“  r   rø   r”  ra   )r#   rt  r]  r•  Ú_AxÚ_Ayr&   r¦   )Ú	image_clsre   rf   r8  rp   rÂ   r0   r0   r1   Útest_setdata_xya­  s   @$rŸ  c            
      C   sÞ   ddl m}  tj d¡ tj dd¡}t dd¡\}}|d  |¡}|d  |¡}tj	||d d tj	||d d t
 ¡ }tj|dd t
 | ¡ ¡}|  |¡}d }| d	¡D ]}	|d u rd|	d
 }qY|	d
 |krlJ ‚qYd S )Nr   )ÚElementTreert   r4   r5   )rp   rF   r¤   r   r  )Ú	xml.etreer   r$   ry   rz   r{   r#   r'   Ú
pcolormeshÚcolorbarrk   rl   rm   ÚgetvalueÚparseÚiter)
r   r¨   r,   rp   Úp1Úp2rµ   Útreer  r   r0   r0   r1   Útest_minimized_rasterized½  s(   

ÿürª  c                  C   s¤   t tƒjd } dtjdkrdnd |  ¡  ¡  }tjt	dd t
 |¡ W d   ƒ n1 s/w   Y  tj |¡}t
 |¡ W d   ƒ d S 1 sKw   Y  d S )Nr‰   zfile:Úwin32z///Ú zPlease open the URL©Úmatch)r   rŒ   r   ÚsysÚplatformÚresolveÚas_posixr  r  Ú
ValueErrorr#   ro   ÚurllibÚrequestÚurlopen)r—   ÚurlÚfiler0   r0   r1   Útest_load_from_urlÞ  s   ÿ
þÿ"ÿr¹  Úlog_scale_imagec                  C   sN   t  d¡} d| d d d…< t ¡ \}}|j| g d¢ddddd |jd	d
 d S )Nr	  r4   r5   )r4   r   r4   r   Úviridisra   r   )rá   r/  ÚvmaxÚvminÚaspectÚlog)Úyscale)r$   r<   r#   r'   r(   rW  )rÙ   r,   rp   r0   r0   r1   Útest_log_scale_imageé  s   
ÿrÁ  Úrotate_imagec                  C   s,  d} t  dd| ¡ }}t  ||¡\}}t  |d |d   d ¡dt j  }t  |d d d |d d d   d ¡dt j d d  }|| }t dd¡\}}	|	j|dd	d
g d¢dd}
tƒ  	d¡|	j
 }|
 |¡ |
 ¡ \}}}}|	j|||||g|||||gdd|d |	 dd¡ |	 dd¡ d S )Nrâ   ç      Àrš  r5   r4   ç      ø?rØ   r;   r»  r]   )éþÿÿÿr6   éýÿÿÿr5   T)r    r/  r`   rá   Úclip_oné   zr--r   )Úlwræ   r   r   r6   )r$   r%   ÚmeshgridÚexpÚpir#   r'   r(   r   Ú
rotate_degrè   Úset_transformÚ
get_extentrj   r…   r†   )Údeltare   rf   r+   ÚYÚZ1ÚZ2rÙ   r,   r-   Úim1Útrans_data2Úx1Úx2Úy1Úy2r0   r0   r1   Útest_rotate_imageô  s*   &*ÿ
þ
 ÿrÚ  c                  C   sR   t  ¡ } t d¡}tj| |dd |  d¡ t | ¡}|jd d… |jks'J ‚d S )N)iá  iA  rW   r¤   r   r5   )	rk   rl   r$   r<   r#   r¥   rn   ro   r¦   )rµ   rÂ   rB   r0   r0   r1   Útest_image_preserve_size  s   


rÛ  c                  C   sÈ   d} t  | t¡}tj| | fdd}t |g d¢¡}| ¡  | |¡ |j|dddd t	 
¡ }|j|d	d
 | d¡ t |¡}|jdksGJ ‚tt  |d d …d d …df t¡t  | t¡d d d… ƒ d S )Né   F)rZ   r(  )re  re  rÖ   rÖ   r   r]   r   )r    r`   r¾  r4   )r[   r   )rÜ  rÜ  r6   ra   )r$   r`  rŠ  r#   rb   ÚAxesÚset_axis_offr1  r(   rk   rl   rm   rn   ro   r¦   r   rV  Úbool)Únr¨   r,   rp   rµ   rB   r0   r0   r1   Útest_image_preserve_size2  s   


ÿrá  zmask_image_over_under.pngrÖ   )r3   Útolc               	   C   sl  dt jd< d} t dd| ¡ }}t ||¡\}}t |d |d   d ¡dtj  }t |d d d |d d	 d   d ¡dtj d	 d  }d
||  }t jjj	dddd}tj
 |dk|¡}	t  dd¡\}
\}}|j|	d|tjdddddg d¢d}| d¡ |
j|dd|d
d |j|	d|tjg d¢ddddg d¢d}| d¡ |
j|dd d|d
d! d S )"NFzpcolormesh.snapgš™™™™™™?rÃ  rš  r5   r4   rÄ  rØ   rt   r5  r6  r7  ©ÚoverÚunderÚbadç333333ó?r!   g      ð¿rÖ   )r½  r¼  Úclipr]   )rÆ  r   rÆ  r   )r    r/  r  r`   rá   zGreen=low, Red=high, Blue=badÚbothÚ
horizontal)ÚextendÚorientationrp   r¾  r   )ra   rq  gš™™™™™É¿r   r\  rØ   r4   r†  )Úncolorsrè  zWith BoundaryNormÚproportional)rë  Úspacingrì  rp   r¾  )r#   r   r$   r%   rÊ  rË  rÌ  Úcmr.  Úwith_extremesÚmaÚmasked_wherer'   r(   r   rz  r)   r£  ÚBoundaryNorm)rÐ  re   rf   r+   rÑ  rÒ  rÓ  rÙ   ÚpaletteÚZmr,   r-   r.   rÂ   r0   r0   r1   Útest_mask_image_over_under2  s@   
&*ÿý

ÿ
ÿü


ÿr÷  Ú
mask_imagec                  C   s   t  dd¡\} \}}t d¡}tj|dd…dd…f< |j|dd tjdtd}d|dd…dd…f< tj 	tjdtj
d|¡}|j|dd d S )Nr4   r5   r7   r   r   r8   T)r#   r'   r$   r=   Únanr(   r<   rß  rò  Úmasked_arrayrš   )r,   r-   r.   rƒ   r0   r0   r1   Útest_mask_imageT  s   
rû  c                  C   s2   t  dt j¡} t ¡ \}}| | ¡ |j ¡  d S )NrX   )r$   Úfullrù  r#   r'   r(   ré   Ú	draw_idle)r¨   r,   rp   r0   r0   r1   Útest_mask_image_allf  s   
rþ  zimshow_endianess.pngc                  C   s„   t  d¡} t  | | ¡\}}t  |d |d ¡}t dd¡\}\}}tdddd}|j| d	¡fi |¤Ž |j| d
¡fi |¤Ž d S )Nrt   r   r4   r5   r]   r   r»  )r`   r    r/  z<f8z>f8)	r$   r%   rÊ  Úhypotr#   r'   Údictr(   r´   )re   r+   rÑ  rÙ   r,   r-   r.   Úkwargsr0   r0   r1   Útest_imshow_endianessn  s   
r  Úimshow_masked_interpolationÚx86_64rå   )râ  r3   r   c            
      C   sð   t jd jdddd} d}tjd|| d d	}tj|| td
 ||¡}d|d< d|d< tj	|d< t 
|¡ d¡}d|d< tj ||¡}t dd¡\}}ttjƒ}| d¡ t|| ¡ ƒD ]\}}	|	 |¡ |	j||| |d |	 d¡ q^d S )Nr»  r5  r7  rÈ   rã  r   r   r4   ©r½  r¼  r8   ra   r7   g     jø@)r*  r   )r*  r*  rß  T)r   r*  r   r   rx   )r  r/  r    Úoff)rM  Ú	colormapsrñ  r   rz  r$   r%   rŠ  r&   ÚinfÚ
zeros_liker´   rò  rú  r#   r'   ÚsortedÚmimageÚ	_interpd_ÚremoveÚzipÚravelr)   r(   Úaxis)
r/  ÚNrà  r¨   Úmaskr,   Úax_gridÚinterpsÚinterprp   r0   r0   r1   Ú test_imshow_masked_interpolation|  s$   



ýr  c                   C   s   t  ddgdtjgg¡ d S )Nr4   r5   r   )r#   r(   r$   rù  r0   r0   r0   r1   Útest_imshow_no_warn_invalid   s   r  r9   c                 C   s   g | ]}t  |¡‘qS r0   )r$   r9   )rÇ   Úsr0   r0   r1   r  ¥  s    r  zu2 u4 i2 i4 i8 f4 f8c                 C   sÐ   t jd| d d¡}| jdkr|d8 }|dk }|dk}| jdkr$|d }t ¡ \}}| |¡ ¡ }|| dk ¡ s;J ‚| jdkrT|| d	k ¡ sJJ ‚|j	jdksRJ ‚d S || dk ¡ s^J ‚|j	t j
ksfJ ‚d S )
Nr„  r8   )rt   rt   r   Úurt   r   rP   Úfr4   )r$   r%   r&   Úkindr#   r'   r(   Ú	get_arrayÚallr9   r?   )r9   r  Útoo_lowÚtoo_highÚ_rp   Úoutr0   r0   r1   Ú$test_imshow_clips_rgb_to_valid_range¤  s   


r"  zimshow_flatfield.pngc                  C   s0   t  ¡ \} }|jt d¡dd}| dd¡ d S )Nr7   r   r   rØ   rÄ  )r#   r'   r(   r$   r=   Úset_clim©r,   rp   rÂ   r0   r0   r1   Útest_imshow_flatfield¹  s   r%  zimshow_bignumbers.pngc                  C   sJ   dt d< t ¡ \} }tjg d¢g d¢gtjd}| |¡}| dd¡ d S )Nr   úimage.interpolation)r4   r5   g   ¢”mB)r   r4   r6   r8   r   r   )r   r#   r'   r$   rS   Úuint64r(   r#  ©r,   rp   rB   Úpcr0   r0   r1   Útest_imshow_bignumbersÀ  s
   
r*  zimshow_bignumbers_real.pngc                  C   sD   dt d< t ¡ \} }t g d¢g d¢g¡}| |¡}| dd¡ d S )Nr   r&  )rp  rÖ   g’ÕMÏð€D)r›  rÖ   rš  r   r   )r   r#   r'   r$   rS   r(   r#  r(  r0   r0   r1   Útest_imshow_bignumbers_realË  s
   
r+  Ú	make_normc                   C   ó
   t  d¡S ©Nr4   )r   Ú
SymLogNormr0   r0   r0   r1   Ú<lambda>Û  ó   
 r0  c                   C   r-  r.  )r   Ú	PowerNormr0   r0   r0   r1   r0  Ü  r1  c                 C   s¤   t  ¡ \}}tjtdd |jg g| ƒ d}W d   ƒ n1 s!w   Y  | g d¢¡ |j ¡  t 	t
¡ | |j ¡ ¡ W d   ƒ d S 1 sKw   Y  d S )Nz.Attempting to set identical low and high xlimsr­  ©r  )éûÿÿÿr   r4  r   )r#   r'   r  ÚwarnsÚUserWarningr(   Ú
set_extentré   rh  r  ÚRuntimeErrorÚ
make_imageÚget_renderer)r,  r,   rp   rÂ   r0   r0   r1   Útest_empty_imshow×  s   ÿþ
"ÿr;  c                  C   s0   t  ¡ \} }| tjdtjd¡ | j ¡  d S )NrH  r8   )r#   r'   r(   r$   r<   Úfloat16ré   rh  ©r,   rp   r0   r0   r1   Útest_imshow_float16é  s   r>  c                  C   st   t  ¡ \} }| tjdtjd¡ t tjtjd¡rtƒ nt	 
t¡ | j ¡  W d   ƒ d S 1 s3w   Y  d S )NrH  r8   Úequiv)r#   r'   r(   r$   r<   Ú
longdoubleÚcan_castÚfloat64r   r  r5  r6  ré   rh  r=  r0   r0   r1   Útest_imshow_float128ð  s   ÿ"ýrC  c                  C   s0   t  ¡ \} }| tjddgddggtd¡ d S )NTFr8   )r#   r'   r(   r$   rS   rß  r=  r0   r0   r1   Útest_imshow_boolù  s   $rD  c                  C   s.   t  ¡ \} }| t dtj¡¡ | j ¡  d S )Nr	  )r#   r'   r(   r$   rü  rù  ré   rh  r=  r0   r0   r1   Útest_full_invalidþ  s   rE  zfmt,counted)r®   s    colorimage)rF   s   <imagezcomposite_image,count)Tr4   )Fr5   c           
      C   s²   t  t  ddd¡t  ddd¡¡\}}t  |d ¡}t ¡ \}}| dd¡ |j|g d¢d |j|d d d	… g d
¢d |tjd< t	 
¡ }	|j|	| d |	 ¡  |¡|ksWJ ‚d S )Nr4  r   r4   r5   r   r   ©r   r4   r   r4   rà   ra   )r5   r   r   r4   zimage.composite_imager¤   )r$   rÊ  r%   rd   r#   r'   r…   r(   r   rk   rl   rm   r¤  Úcount)
rŸ   ÚcountedÚcomposite_imagerG  r+   rÑ  rÙ   r,   rp   rÁ   r0   r0   r1   Útest_composite  s   $
rJ  c                  C   sV   t  ¡ \} }|jdggdd | ¡  | ¡  | ¡ | ¡   kr&dks)J ‚ J ‚d S )Nr   rF  rà   rI  )r#   r'   r(   ÚrelimÚ	autoscaleÚget_xlimÚget_ylimr=  r0   r0   r1   Ú
test_relim  s
   (rO  c                  C   sx   t  ¡ \} }| ¡  |jddgddggddddd}|jddd | j ¡  t | j 	¡ ¡d	d d
…f dk 
¡ s:J ‚d S )Nr   r   r,  r.  F)r¾  rá   r/  rÇ  rI  rS  .r   )r#   r'   rÞ  r(   rW  ré   rh  r$   rS   Úbuffer_rgbar  r$  r0   r0   r1   Útest_unclipped   s   ÿ
,rQ  c                  C   sÄ   t  d¡\} }|D ]}| ¡  q	|d jddgddggddd}| d ¡ | |d j¡ t ¡ }| j	|dd	 h | 
¡ £d
hksCJ ‚|d jdd t ¡ }| j	|dd	 | 
¡ | 
¡ ks`J ‚d S )Nr5   r4   r   r   r   rF  )r¾  rá   Úrgbar¤   rP   )ra   r   )rU  )r#   r'   rÞ  r(   rû   Úset_clip_boxÚbboxrk   rl   rm   r¤  rW  )r,   r‘   rp   rÂ   Ú
buf_beforeÚ	buf_afterr0   r0   r1   Útest_respects_bbox,  s   
 
rW  c                  C   s„   t  ¡ \} }| t d¡¡}tjjdgdgd}| |¡dks!J ‚tjjdgdgd}| |¡dks4J ‚tj}| |¡dks@J ‚d S )	NrÝ   r   T©r  ró   Fz[0]z[nan])	r#   r'   r(   r$   r<   rò  rú  rô   rù  )r,   rp   rÂ   r¨   r0   r0   r1   Útest_image_cursor_formatting>  s   rY  c           
      C   sœ   t  dd¡}t  ||¡\}}t  d|d d  |d  ¡}|| ¡  }tjd }|  ¡ }|j|||dd	 | ¡ }|t	 
¡ |ƒƒ}	||	d
< |j|	dd dS )zPer-pixel alpha channel test.r   r4   rÆ  rØ   r5   g[Âõ(\ß?r»  r   )r×   r/  r    ).ra   r   N)r$   r}  rÊ  rË  ÚmaxrM  r  rL  r(   r   rz  )
r   r€   re   ÚxxÚyyÚzzr×   r/  rp   rR  r0   r0   r1   Útest_image_array_alphaM  s   
r^  c                   C   sL   t jtdd tjt d¡ddgd W d   ƒ d S 1 sw   Y  d S )Nzalpha must be a float, two-dr­  rX   r4   )r×   )r  r  r  r#   r(   r$   r<   r0   r0   r0   r1   Ú!test_image_array_alpha_validation`  s   "ÿr_  c                  C   sÊ   t tjd ƒ} | jdd tjddd}| g d¢¡}tjg d¢gt	d	}|j
|d
| ddd}| d¡ |j ¡  | |jj¡d d }| | |dggd  j ¡ ¡¡d  t¡}t ||k¡scJ ‚d S )NÚautumn_rÚ	lightgrey)Úcolor)gffffffþ?rA  r   rY   rw   )ra   ra   ra   r   r   r   r   é+   éO   é_   éB   r4   ra   ra   ra   r   r   r   é"   r8   r   r   )r¾  r/  r½  r¼  r  r4   rt   rP   )r   rM  r  Ú	set_underr#   rb   r1  r$   rS   rŠ  r(   r  ré   rh  r9  ri  r/  r  ÚTr  r´   r|   r  )r/  r,   rp   r¨   rÂ   Ú
from_imageÚdirect_computationr0   r0   r1   Útest_exact_vmine  s    þ

$þrl  Úimage_placementc                  C   sn   t  ¡ \} }|jg d¢g d¢ddd tj d¡ |jtj dd¡dd	d
ddd | dd¡ | 	dd¡ dS )zK
    The red box should line up exactly with the outside of the image.
    )r   r   r4   r4   r   )r   r4   r4   r   r   r5  rA  )rb  rÉ  rv   r³   rs  rF  r;   ra   r4   )r/  rá   r    r½  r¼  gš™™™™™¹¿r@  N)
r#   r'   rj   r$   ry   rz   r(   Úrandnr…   r†   r=  r0   r0   r1   Útest_image_placement‚  s   ÿro  c                       s@   e Zd Zdd„ Zdd„ Z‡ fdd„Zdd„ Zed	d
„ ƒZ‡  Z	S )Ú
QuantityNDc                 C   s   t  |¡ | ¡}||_|S rx  )r$   rV  ÚviewÚunits)ÚclsÚinput_arrayrr  Úobjr0   r0   r1   Ú__new__–  s   zQuantityND.__new__c                 C   s   t |dd ƒ| _d S ©Nrr  )Úgetattrrr  )Úselfru  r0   r0   r1   Ú__array_finalize__›  s   zQuantityND.__array_finalize__c                    s<   t | dd ƒ}tt| ƒ |¡}t|tƒs|d urt||ƒ}|S rw  )rx  Úsuperrp  Ú__getitem__Ú
isinstance)ry  Úitemrr  Úret©Ú	__class__r0   r1   r|  ž  s
   
zQuantityND.__getitem__c                 O   sp  t ||ƒ}d|v rtS t|ƒdkr'|d }t |ddƒ}|t |¡fi |¤Ž}nt|ƒdkr¤|d }|d }	t |ddƒ}
t |	ddƒ}|
d u rG|n|
}
|d u rO|
n|}|tjtjfv rb|
|kr_t‚|
}n0|tjkro|
› d|› }n#|tj	kr}|
› d|› d	}n|tj
tjtjtjtjtjfv rd }ntS || tj¡|	 tj¡fi |¤Ž}ntS |d u r±t |¡}|S t||ƒ}|S )
Nr!  r4   r   rr  Údimensionlessr5   Ú*z/(ú))rx  ÚNotImplementedrÐ   r$   rV  ÚaddÚsubtractr³  ÚmultiplyÚdivideÚgreaterÚgreater_equalÚequalÚ	not_equalÚlessÚ
less_equalrq  ÚndarrayrS   rp  )ry  ÚufuncÚmethodÚinputsr  ÚfuncÚi0ÚunitÚout_arrÚi1Úu0Úu1r0   r0   r1   Ú__array_ufunc__¥  sF   



þ$

ÿzQuantityND.__array_ufunc__c                 C   s   |   tj¡S rx  )rq  r$   r  )ry  r0   r0   r1   rÉ   Ì  s   zQuantityND.v)
Ú__name__Ú
__module__Ú__qualname__rv  rz  r|  r›  ÚpropertyrÉ   Ú__classcell__r0   r0   r€  r1   rp  •  s    'rp  c                  C   sÌ   t ddgdƒ} | d d … \}}t | jt ddg¡k¡sJ ‚| jdks%J ‚t || jt dg¡k¡s5J ‚|| jdks>J ‚|| jdksGJ ‚t t¡ |t ddƒ  W d   ƒ d S 1 s_w   Y  d S )Nr4   r5   Úmr   zm*mzm/(m)r  )	rp  r$   r  rÉ   rV  rr  r  r  r³  )ÚqÚq0Úq1r0   r0   r1   Útest_quantityndÑ  s    "ÿr¥  c                  C   s4   t t d¡dƒ} t ¡ \}}| | ¡ |j ¡  d S )NrX   r¡  )rp  r$   r=   r#   r'   r(   ré   rh  )r  r,   rp   r0   r0   r1   Útest_imshow_quantityndÝ  s   
r¦  c                 C   s.  t jddt jd}d|dd…d d …f< t jj|dd}d	|jdd…dd…f< tjd
 jdd}|  	¡ }|j
|tjddddd|d}| tjddd¡ |j
|tjddddd|d}| tjddd¡ |jddd | 	¡ }|j
|tjddddd|d |j
|tjddddd|d |jddd d S )Nr7   r4   r8   ra   r   r5   FrX  Tr»  Úw©rå  rØ   r  r   r   )r  rá   r    r/  rÅ  )r   rt   r   rt   ©r   rt   rS  )r$   rü  rB  rò  rS   r  rM  r  rñ  r'   r(   r   r!  ry  rz  rW  )r   r€   r¨   Úmasked_datar/  rp   rÂ   r0   r0   r1   Útest_norm_changeæ  s.   ÿÿÿÿr«  re   c                 C   s²   t jd|t jd}d|dd…d d …f< |  ¡ }|j|tjd| ¡ ddd	d
 t jd|t jd}d|dd…d d …f< | ¡ }tj	d	 j
dd}|j|tjd| ¡ dd|d
 d S )Nr7   r8   g@Œµx¯Dr   r5   r4   r  r   r»  )r  r    r/  iè  r§  r¨  )r$   rü  rB  r'   r(   r   r!  rZ  rM  r  rñ  rz  )r   r€   re   r¨   rp   r/  r0   r0   r1   Útest_huge_range_log  s   ÿ
ÿr¬  c           	   
   C   sB  |   dd¡}|  dd¡}ddgddggddgddggddgddggf}g d¢}tt||ƒƒD ]p\}\}}||  |¡ ||  |¡ ||  |¡ || j|ddddddd	 ||  d
d¡ ||  dd
¡ || j 	¡  || j
 d¡ || j d¡ || j tjdg d¢dd¡ || j tjdg d¢dd¡ q.d S )Nr4   r   r   )r=   r<   Úmixedr   rŒ  r²   ÚGreys)r    r¾  r`   r/  r½  r¼  rq  rÄ  gÍÌÌÌÌÌð?ré  r+  )r4   r5   r   rt   T)ÚnbinsÚstepsÚinteger)r'   Ú	enumerater  r)   Úspyr(   r…   r†   ÚxaxisÚtick_topÚtitleÚset_yÚset_ticks_positionÚset_major_locatorÚmtickerÚMaxNLocatorÚyaxis)	r   r€   Úax_testÚax_refÚ	plot_dataÚplot_titlesr‚  rg   r¶  r0   r0   r1   Útest_spy_box  s6   ýþ
ÿ
ÿñrÁ  znonuniform_and_pcolor.pngc                  C   sÈ   t jddjdddd} t| ddgƒD ]&\}}t||d}| t d¡d	 t d¡d	 t d
¡ d¡¡ | 	|¡ q| d	  
t d¡d	 t d¡d	 t d
¡ d¡¡ | D ]}| ¡  |jdd qUd S )NrH  r'  r   T)ÚsharexÚshareyr   r!   r   r5   r+  r6   r©  )rT  )r#   rb   r'   r  r   r]  r$   r%   r&   Ú	add_imageÚ
pcolorfastrÞ  rW  )r‘   rp   r    rÂ   r0   r0   r1   Útest_nonuniform_and_pcolor7  s   ÿ&ÿýrÆ  zrgba_antialias.png)Úaarch64Úppc64leÚs390xgyé&1¬|?)r   r3   râ  c                  C   s¢  t jddddddd\} }d}t ||f¡}d|d d d…d d …f< t |¡| d }t |¡| d }t ||¡\}}t |d |d  ¡}d	}	d
}
t tjd |	| |
|d  d   ¡}d|d t	|d ƒ…d d …f |d t	|d ƒ…d d …f dk < d|d t	|d ƒ…d d …f |d t	|d ƒ…d d …f dk < |d d …t	|d ƒd …f |d d …t	|d ƒd …f< tj
|dd…dd…f< d|dd…dd…f< d|dd…dd…f< d|dd…dd…f< d|dd…dd…f< tt jjƒ}| d¡ | d¡ | ¡ }|d j|d|ddd |d  |d d |d d g¡ |d  |d d |d d	 g¡ |d j|d|ddd |d j|dd |ddd! |d" j|dd#|ddd! d S )$Nr5   )ç      @rÊ  FT)rZ   rÂ  rÃ  Úconstrained_layoutéú   ra   rØ   rt   éK   rd  r4   rB  r   r\   g    €„.AéF   éZ   g    €„.ÁrÈ  éÃ   é×   ÚyellowÚcyanr   r   g333333ó¿rç  )r    r/  r½  r¼  é   rx   r¨   )r    Úinterpolation_stager/  r½  r¼  r   rR  )r#   r'   r$   r=   r%   rÊ  Úsqrtrd   rÌ  r|   rù  r   rð  ÚRdBu_rÚset_overrh  r€  r(   r…   r†   )r,   r‘   r  Úaare   rf   r+   rÑ  ÚRÚf0rÈ   r8  r/  r0   r0   r1   Útest_rgba_antialiasG  sF   
ÿ(<<4

""ÿ
ÿrÜ  z<ignore:Data with more than .* cannot be accurately displayedr`   r²   r]   zdim, size, msg)Úrowi  € z2\*\*23 columns)Úcoli   z2\*\*24 rows)rW   c           
   	   C   sæ   |   ¡ }|  ¡ }t d|d f¡}d|d d …|jd d …f< |dkr%|j}|j|ddddd|d}	tjtd	|› d
d | j	 
¡  W d   ƒ n1 sLw   Y  t d¡}d|d d …df< |dkre|j}|j|ddddd|d}	d S )Nr4   r5   rÞ  r   r   rF  r;   )r½  r¼  r¾  rá   r    r`   zData with more than z  cannot be accurately displayed.r­  rK  )r'   r$   r<   Úsizeri  r(   r  r5  r6  ré   rh  )
r   r€   Údimrß  Úmsgr`   r½  r¾  rS   rÂ   r0   r0   r1   Útest_large_image  s2   
ý
ÿý

ýrâ  c                 C   sn  t j dd¡d d }|  dd¡}|d j|dd |d j|dd	d
 |d j|dd |d j|dddd |d j|dddd | dd¡}|d j|t ¡ d |d j|tjd	dd |d j|tjddd |d j|tjddddd |d j|ddd t|d j	d j
ƒtju s–J ‚t t¡ |d j|dd W d   ƒ d S 1 s°w   Y  d S )Nrt   gš™™™™™é?rA  r4   r   r   r¿  r3  r\  )r  r½  r5   Úsymlogr   rB  rf  )r  r½  r¼  r6   Úlogit)r½  )Ú	linthresh)rå  r½  r¼  )rB  rf  )r  ÚclimÚfoobar)r$   ry   r{   r'   r(   r   r!  r/  ÚtypeÚimagesr  r  r  r³  )r   r€   rN  ÚaxtsÚaxrsr0   r0   r1   Útest_str_norms§  s"   "ÿrì  )‰Ú
contextlibr   r   rk   r–   Úpathlibr   r°  r¯  Úurllib.requestr´  Únumpyr$   Únumpy.testingr   ÚPILr   Ú
matplotlibrM  r   r   r  r	   r
   r#   r   r   Úmatplotlib.imager   r   r   r   r   Úmatplotlib.testing.decoratorsr   r   Úmatplotlib.transformsr   r   r   Úmatplotlib.tickerÚtickerrº  r  r2   rC   rT   ÚmarkÚparametrizeri   rr   r„   r‡   r’   r   rž   r­   r±   r·   rÃ   rÔ   rÚ   rò   rù  rö   rþ   r  r  r  r  r  r  r&  r9  r<  rG  rO  rZ  rc  ro  rr  rw  r{  rƒ  r  r–  r™  r%   r&   rŸ  rª  r¹  rÁ  rÚ  rÛ  rá  r÷  rû  rþ  r  Úmachiner  r  Úsplitr"  r%  r*  r+  rz  r!  r;  r>  rC  rD  rE  rJ  rO  rQ  rW  rY  r^  r_  Úcontextrl  ro  r  rp  r¥  r¦  r«  r¬  rÁ  rÆ  rÜ  Úfilterwarningsrâ  rì  r0   r0   r0   r1   Ú<module>   sš    

ÿ
	þ




"


Gûÿ

	





ÿ
	



ÿ

#ÿ

		ÿÿýÿ
	!



!

þ
!ÿ



ÿ

ýþ
	ÿ


ÿ
<
	


 
þ7ÿÿ

