o
    liE                     @   s(  d dl mZ d dlm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mZ d dlmZmZ dZdZeeiZed3d	d
Zdd Zdd Zdd Zdd Zdd Zdd Zdd Zdd Zdd Zdd Zdd  Z e	j!j"d!d"d#gd$d%gd&d'd( Z#d)d* Z$d+d, Z%d-d. Z&d/d0 Z'd1d2 Z(dS )4    )contextmanager)Path)TemporaryDirectoryN)pyplotstyle)USER_LIBRARY_PATHSSTYLE_EXTENSIONz
image.cmappinkc                 c   s    |st }d| tf }zAt (}t||jddd | D dd t| t	
  dV  W d   n1 s:w   Y  W t	
  dS W t	
  dS t	
  w )zAContext manager to create a style sheet in a temporary directory.z%s.%s
c                 s   s     | ]\}}d  ||V  qdS )z{}: {}N)format).0kv r   Y/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/matplotlib/tests/test_style.py	<genexpr>   s    ztemp_style.<locals>.<genexpr>utf-8encodingN)DUMMY_SETTINGSr   r   r   
write_textjoinitemsr   appendr   reload_library)
style_namesettings	temp_filetmpdirr   r   r   
temp_style   s"   



r   c                 C   s^   ddi}d}t || W d    n1 sw   Y  t| jdkr+|| jd  v s-J d S )Nfoobarbasename   r   )r   lenrecords
getMessage)caplogSETTINGSr"   r   r   r   )test_invalid_rc_warning_includes_filename'   s   r)   c                   C   s>   t dt dtjv sJ W d    d S 1 sw   Y  d S )N_test_)r   r   r   	availabler   r   r   r   test_available1   s   "r,   c                	   C   s   dt jt< tdt/ td t jt tksJ W d    n1 s$w   Y  W d    d S W d    d S 1 s<w   Y  d S NgraytestmplrcParamsPARAMr   r   r   contextVALUEr   r   r   r   test_use6   s   
"r6   c              	   C   s   t | d}|jddd tdt@ dtjdkrdnd	 |   }t	| t
jd
 dks2J W d    n1 s<w   Y  W d    d S W d    d S 1 sTw   Y  d S )Nfilezaxes.facecolor: adeader   r   r/   zfile:win32z/// zaxes.facecolorz#adeade)r   r   r   r   sysplatformresolveas_posixr   r4   r1   r2   )r   pathurlr   r   r   test_use_url=   s   

"r@   c                 C   s   dt jt< dt }t| |}|jt dt dd t| t jt tks*J W d    n1 s4w   Y  t jt dksBJ d S )Nr.   ztext.z : r   r   )	r1   r2   r3   r   r   r   r5   r   r4   )r   r   r>   r   r   r   test_single_pathH   s   


rA   c                	   C   s   dt jt< tdt& td t jt tksJ W d    n1 s$w   Y  W d    n1 s3w   Y  t jt dksAJ d S r-   r0   r   r   r   r   test_contextR   s   
rB   c                  C   sh   d} d}| t jt< tt|i t jt |ksJ W d    n1 s$w   Y  t jt | ks2J d S )Nr.   blue)r1   r2   r3   r   r4   original_valueother_valuer   r   r   test_context_with_dict[   s   
rG   c               	   C   s   d} d}| t jt< tdt* tdt|ig t jt |ks"J W d    n1 s,w   Y  W d    n1 s;w   Y  t jt | ksIJ d S Nr.   rC   r/   )r1   r2   r3   r   r   r   r4   rD   r   r   r   'test_context_with_dict_after_namedstyled      
rI   c               	   C   s   d} d}| t jt< tdt* tt|idg t jt tks"J W d    n1 s,w   Y  W d    n1 s;w   Y  t jt | ksIJ d S rH   r0   rD   r   r   r   (test_context_with_dict_before_namedstyleo   rJ   rK   c               	   C   s   d} d}d}||i}| t jt< | t j|< tdt1 td|g t jt tks,J t j| |ks5J W d    n1 s?w   Y  W d    n1 sNw   Y  t jt | ks\J t j| | ksfJ d S )Nr.   ztext.usetexTr/   r0   )rE   other_paramrF   dr   r   r   .test_context_with_union_of_dict_and_namedstylez   s   
rN   c               
   C   s   d} d}t t|iK tjt |ksJ t t| dd i}tt | W d    n1 s1w   Y  W d    n1 s@w   Y  tjt |ksNJ W d    d S 1 sYw   Y  d S )Nr.   rC   badparam)r   r4   r3   r1   r2   pytestraisesKeyError)rE   rF   xr   r   r   test_context_with_badparam   s   "rT   equiv_styles)mpl20default)mpl15classicrV   rX   )idsc              	   C   s   g }| D ]}t | |tj  W d    n1 sw   Y  q|d }t| dd  |dd  D ]
\}}||ks?J q5d S )Nr   r#   )r   r4   r   r1   r2   copyzip)rU   rc_dictsstyrc_basenmrcr   r   r   
test_alias   s   "rb   c                   C   sL   t jd d u s	J t  t jd dksJ tj  t jd dks$J d S Nzpath.sketch)r#   d      )r1   r2   pltxkcdnptestingbreak_cyclesr   r   r   r   test_xkcd_no_cm   s
   
rk   c                   C   sb   t jd d u s	J t  t jd dksJ W d    n1 s!w   Y  t jd d u s/J d S rc   )r1   r2   rf   rg   r   r   r   r   test_xkcd_cm   s
   
rl   c                  C   s   t jd t j } W d    n1 sw   Y  t j| ks"J tt jj t j	d W d    n1 s:w   Y  t j| ksFJ tt jj t jj
d  W d    d S 1 s_w   Y  d S )Nzseaborn-v0_8-brightzseaborn-bright)r1   r   r4   r2   r[   rP   warns_apiMatplotlibDeprecationWarninguselibrary)seaborn_brightr   r   r   test_deprecated_seaborn_styles   s   "rs   c                   C   s   t jjjh t jjksJ d S N)r1   r   coreSTYLE_BLACKLISTrcsetup_validatorsr   r   r   r   test_up_to_date_blacklist   s   ry   c                 C   s   | |  ||  | d }|  |d jddd |djddd tjd tjd	 d
ks4J tjd tjd	 dksCJ tjd tjd	 dksRJ d S )Nmpl_test_style_pkgztest_style.mplstylezlines.linewidth: 42r   r   z	.mplstylezlines.linewidth: 84zmpl_test_style_pkg.test_stylezlines.linewidth*   zmpl_test_style_pkg.mplstyleT   z./mpl_test_style_pkg.mplstyle)	syspath_prependchdirmkdirr   with_suffixr1   r   rp   r2   )tmp_pathmonkeypatchpkg_pathr   r   r   test_style_from_module   s    


r   rt   ))
contextlibr   pathlibr   tempfiler   r:   numpyrh   rP   
matplotlibr1   r   rf   r   matplotlib.style.corer   r   r3   r5   r   r   r)   r,   r6   r@   rA   rB   rG   rI   rK   rN   rT   markparametrizerb   rk   rl   rs   ry   r   r   r   r   r   <module>   sJ    

		
