o
    ä¸GfÛ   ã                   @   sF  d dl Z d dlmZ d dlmZmZmZmZm	Z	 d dlm
Z d dlmZmZ d dlmZmZ d dlZd dlmZ d dlZd dlZeƒ dd„ ƒZdd	„ Zed
gdddd„ ƒZdd„ Zdd„ Zedgdddd„ ƒZdd„ Zdd„ Zdd„ Z edgddddd d!„ ƒZ!d"d#„ Z"ed$gddd%d&d'„ ƒZ#d(d)„ Z$d*d+„ Z%G d,d-„ d-ƒZ&dS ).é    N)Ú
AsinhScaleÚAsinhTransformÚLogTransformÚInvertedLogTransformÚSymmetricalLogTransform)ÚAsinhLocatorÚLogFormatterSciNotation)Úcheck_figures_equalÚimage_comparison)Úassert_allclosec                 C   sz   | j dddd}| d¡ | d¡ | ¡ }| ¡ }|j dddd}|j||d | ddg|d¡ | |ddgd¡ d S )Néz   ÚlogÚsymlog)ÚyscaleZxscalegš™™™™8@)ÚxlimÚylimÚb)Zadd_subplotZaxvlineZaxhlineÚget_xlimÚget_ylimÚsetÚplot)Zfig_testZfig_refZax_testr   r   Zax_ref© r   ú;lib/python3.10/site-packages/matplotlib/tests/test_scale.pyÚtest_log_scales   s   

r   c                  C   s   t dddƒ} |  ¡ }t ddd¡}| |  |¡¡}t||ƒ t|ƒt|ƒu s(J ‚tj|d< | |  |¡¡}t||ƒ t|ƒt|ƒu sDJ ‚tj 	|¡}| |  |¡¡}t||ƒ t|ƒt|ƒu saJ ‚tjj
|d< | |  |¡¡}t||ƒ t|ƒt|ƒu s~J ‚d S )	Né
   é   é   g      ø¿é   ç      à?é   é   )r   ÚinvertedÚnpÚarangeÚtransform_non_affiner   ÚtypeÚnanZmaÚarrayZmasked)ZsltZsltiÚxÚoutr   r   r   Útest_symlog_mask_nan   s$   




r*   zlogit_scales.pngT)Úremove_textc                  C   sv   t  ¡ \} }t g d¢¡}d| }| ||¡ | d¡ | d¡ | | j 	¡ ¡}t 
|j¡s1J ‚t 
|j¡s9J ‚d S )N)gü©ñÒMbP?gú~j¼t“h?ç{®Gáz„?g¸…ëQ¸ž?gš™™™™™¹?çš™™™™™É?g333333Ó?gš™™™™™Ù?r   g333333ã?gffffffæ?gš™™™™™é?gÍÌÌÌÌÌì?g
×£p=
ï?g®Gáz®ï?g•C‹lçï?g+‡ÙÎ÷ï?g      ð?ZlogitT)ÚpltÚsubplotsr"   r'   r   Ú
set_xscaleZgridZget_tightbboxÚcanvasZget_rendererZisfiniteZx0Zy0)ÚfigÚaxr(   ÚyZbboxr   r   r   Útest_logit_scales:   s   

r5   c                  C   sx   t  d¡\} }t d¡}t d¡d }| ||¡ t ¡ }| j|dd t ¡ }| j|dd t ¡ }| j|dd dS )zIssue #1799r   r   Zpdf)ÚformatZepsZsvgN)r.   r/   r"   r#   ÚscatterÚioÚBytesIOZsavefig)r2   r3   r(   r4   Úbufr   r   r   Útest_log_scatterK   s   
r;   c                  C   s2   t  ¡ \} }|jdt g d¢¡d | j ¡  d S )Nr   )r   r    r   )Úsubs)r.   r/   Ú
set_yscaler"   r'   r1   Údraw©r2   r3   r   r   r   Útest_logscale_subs^   s   r@   zlogscale_mask.pngc                  C   sJ   t  ddd¡} t ¡ \}}| t  | d  ¡¡ |j ¡  |jdd d S )Nr   é2   ié  r   r   )r   )	r"   Úlinspacer.   r/   r   Úexpr1   r>   r   )Úxsr2   r3   r   r   r   Útest_logscale_maske   s
   
rE   c               	   C   sR   t  ¡ \} }dD ]}t t¡ |j|dd W d   ƒ n1 s!w   Y  qd S )N)Zlinearr   r   Úmask)Zfoo)r.   r/   ÚpytestÚraisesÚ	TypeErrorr=   ©r2   r3   Úscaler   r   r   Útest_extra_kwargs_raiseq   s   ÿ€ÿrL   c                  C   sX   t  ¡ \} }| d¡ |j|j ¡   ¡ }tdd ¡ }t|tƒs#J ‚|j	dks*J ‚d S )Nr   r   )Úbase)
r.   r/   r=   Z	transAxesÚ	transDatar!   r   Ú
isinstancer   rM   )r2   r3   ZtformZinverted_transformr   r   r   Útest_logscale_invert_transformy   s   
rP   c                  C   s4   t  ¡ \} }| d¡ t|jƒ ttdddƒ d S )Nr   r   Zclip©Únonpositive)r.   r/   r=   ÚreprrN   r   r?   r   r   r   Útest_logscale_transform_repr…   s   

rT   zlogscale_nonpos_values.pngg{®Gáz”?Zmpl20)r+   ZtolÚstylec                  C   s  t j d¡ t jjtdƒd} t dd¡\}\\}}\}}|j| ddd | d¡ |j| ddd |jdd	d
 t  	ddd¡}t  
| ¡}dd|  t  d| ¡ t  
| ¡ }| ||| || ¡ | d¡ t  dd¡}	|	d }
|	d }|j|	|
|d | d¡ | d¡ d S )Ni!N,g     @@)Úsizer   )éûÿÿÿr   r   )ÚrangeZbinsr   rF   rQ   r   r,   r-   r   éÿÿÿÿr   r    )Úyerr)r"   ZrandomZseedZnormalÚintr.   r/   Zhistr=   r#   rC   ZcosZfill_betweenZlogspaceZerrorbarr0   )rD   r2   Zax1Zax2Zax3Zax4ZxdataZydataZedatar(   r4   rZ   r   r   r   Útest_logscale_nonpos_valuesŒ   s$   
&

r\   c                  C   sd  t  ¡ \} }| tddƒtddƒ¡ | d¡ | ¡ }t t¡ |j	dd W d   ƒ n1 s1w   Y  | ¡ |ks>J ‚t t¡ |j	dd W d   ƒ n1 sTw   Y  | ¡ |ksaJ ‚| 
d¡ | ¡ }t t¡ |jdd W d   ƒ n1 s€w   Y  | ¡ |ksJ ‚t t¡ |jdd W d   ƒ n1 s£w   Y  | ¡ |ks°J ‚d S )	Nr   r   r   )ÚleftrY   )Úright)Zbottom)Útop)r.   r/   r7   rX   r0   r   rG   ZwarnsÚUserWarningÚset_xlimr=   r   Zset_ylim)r2   r3   Zoriginal_xlimZoriginal_ylimr   r   r   Útest_invalid_log_lims§   s,   
ÿÿ
ÿÿrb   zfunction_scales.png)r+   rU   c                  C   sV   dd„ } dd„ }t  ¡ \}}t dd¡}| ||¡ |jd|| fd | dd¡ d S )	Nc                 S   ó   | d S )Nr   r   ©r(   r   r   r   ÚinverseÁ   ó   z$test_function_scale.<locals>.inversec                 S   rc   )Nr   r   rd   r   r   r   ÚforwardÄ   rf   z$test_function_scale.<locals>.forwardr   iè  Úfunction)Z	functions)r.   r/   r"   r#   r   r0   ra   )re   rg   r2   r3   r(   r   r   r   Útest_function_scale¿   s   ri   c                  C   s`   t  ¡ \} }tjd d}| |¡ tjd d}| |¡ |j ¡ dks%J ‚|j ¡ dks.J ‚d S )N©Úaxisr   )	r.   r/   ÚmscaleÚLogScaler0   r=   ZxaxisZ	get_scaleZyaxisrJ   r   r   r   Útest_pass_scaleÐ   s   

rn   c                  C   sH   t jddd} t | ¡}t|  ¡ ƒt| ¡ ƒksJ ‚| j|jus"J ‚d S )Nr(   r   ©rk   rM   )rl   rm   ÚcopyZdeepcopyÚstrÚget_transformZ
_transform)ZscZsc2r   r   r   Útest_scale_deepcopyÛ   s   
rs   c                   @   s4   e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd	d
„ ZdS )ÚTestAsinhScalec           	      C   sn   d}t  ddd¡}t|ƒ}| ¡ }| ¡ }| |¡}| |¡}t||ƒ | |¡}t||t  || ¡ ƒ d S )Ng      1@iÎÿÿÿrA   éd   )r"   rB   r   r!   r$   r   Zarcsinh)	ÚselfZa0Úarg   re   ZinvinvZ	a_forwardZ
a_invertedZa_invinvr   r   r   Útest_transformsã   s   



zTestAsinhScale.test_transformsc                 C   sl   t  ¡ \}}td dd}|jdksJ ‚|jdksJ ‚|jdks!J ‚| ¡ }t|tƒs,J ‚|j|jks4J ‚d S )Ng      7@©rk   Úlinear_widthé   r   )r   r   )	r.   r/   r   rz   Ú_baseÚ_subsrr   rO   r   )rv   r2   r3   ÚsZtxr   r   r   Ú	test_initò   s   zTestAsinhScale.test_initc                 C   sb   t  ¡ \}}td dd}|jdksJ ‚|jdksJ ‚td ddd}|jdks(J ‚|jdks/J ‚d S )Nr    ro   )r   é   )r   r   )rk   rM   r<   )r.   r/   r   r|   r}   )rv   r2   r3   Zs3Ús7r   r   r   Útest_base_initþ   s   zTestAsinhScale.test_base_initc                 C   sš   G dd„ dƒ}|ƒ }t |dd}| |¡ t|jd tƒsJ ‚t|jd tƒs)J ‚|ƒ }t |dd}| |¡ t|jd tƒsAJ ‚t|jd tƒsKJ ‚d S )Nc                   @   s$   e Zd Zdd„ Zdd„ Zdd„ ZdS )z-TestAsinhScale.test_fmtloc.<locals>.DummyAxisc                 S   s
   i | _ d S )N©Úfields)rv   r   r   r   Ú__init__  s   
z6TestAsinhScale.test_fmtloc.<locals>.DummyAxis.__init__c                 [   s   | j jdi |¤Ž d S )Nr   )r„   Úupdate)rv   Úkwargsr   r   r   r     s   z1TestAsinhScale.test_fmtloc.<locals>.DummyAxis.setc                 S   s   || j d< d S )NÚmajor_formatterrƒ   )rv   Úfr   r   r   Úset_major_formatter  s   zATestAsinhScale.test_fmtloc.<locals>.DummyAxis.set_major_formatterN)Ú__name__Ú
__module__Ú__qualname__r…   r   rŠ   r   r   r   r   Ú	DummyAxis
  s    rŽ   r   ro   Zmajor_locatorrˆ   r   )r   Z#set_default_locators_and_formattersrO   r„   r   rq   r   )rv   rŽ   Zax0Ús0Zax5r   r   r   r   Útest_fmtloc	  s   


ÿzTestAsinhScale.test_fmtlocc                 C   s’   t  ¡ \}}t t¡ td dd W d   ƒ n1 sw   Y  t t¡ td dd W d   ƒ n1 s7w   Y  td d}td dd}d S )Nr   ry   rY   rj   g      @)r.   r/   rG   rH   Ú
ValueErrorr   )rv   r2   r3   r   Ús1r   r   r   Útest_bad_scale  s   ÿÿ
zTestAsinhScale.test_bad_scaleN)r‹   rŒ   r   rx   r   r‚   r   r“   r   r   r   r   rt   â   s    rt   )'rp   Zmatplotlib.pyplotZpyplotr.   Zmatplotlib.scaler   r   r   r   r   rK   rl   Zmatplotlib.tickerr   r   Zmatplotlib.testing.decoratorsr	   r
   Znumpyr"   Znumpy.testingr   r8   rG   r   r*   r5   r;   r@   rE   rL   rP   rT   r\   rb   ri   rn   rs   rt   r   r   r   r   Ú<module>   s@    


ÿ

