a
    ضa4                     @   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d-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dd  Z e	j!j"d!d"d#gd$d%gd&d'd( Z#d)d* Z$d+d, Z%dS ).    )contextmanager)Path)TemporaryDirectoryN)pyplotstyle)USER_LIBRARY_PATHSSTYLE_EXTENSIONz
image.cmapZpinkc                 c   s   |st }d| tf }znt L}t||ddd | D  t| t	
  dV  W d   n1 sn0    Y  W t	
  n
t	
  0 dS )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   k/Users/vegardjervell/Documents/master/model/venv/lib/python3.9/site-packages/matplotlib/tests/test_style.py	<genexpr>       ztemp_style.<locals>.<genexpr>N)DUMMY_SETTINGSr   r   r   
write_textjoinitemsr   appendr   Zreload_library)Z
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,0    Y  t| jdkrV|| jd  v sZJ d S )NZfoobarbasename   r   )r   lenrecords
getMessage)ZcaplogZ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    n1 s.0    Y  d S )NZ_test_)r   r   r   	availabler   r   r   r   test_available0   s    r#   c                	   C   st   dt jt< tdtL td" t jt tks4J W d    n1 sH0    Y  W d    n1 sf0    Y  d S NgraytestmplrcParamsPARAMr   r   r   contextVALUEr   r   r   r   test_use5   s    
r-   c              	   C   s   t | d}|d tdtn dtjdkr0dnd |   }t	|" t
jd d	ks`J W d    n1 st0    Y  W d    n1 s0    Y  d S )
Nfilezaxes.facecolor: adeader&   zfile:win32z/// zaxes.facecolorz#adeade)r   r   r   r   sysplatformresolveas_posixr   r+   r(   r)   )r   pathurlr   r   r   test_use_url<   s    


r7   c                 C   s   dt jt< dt }t| |}|t dt  t|" t jt tksPJ W d    n1 sd0    Y  t jt dksJ d S )Nr%   ztext.z : )	r(   r)   r*   r   r   r   r,   r   r+   )r   r   r5   r   r   r   test_single_pathG   s    


0r8   c                	   C   s   dt jt< tdtL td" t jt tks4J W d    n1 sH0    Y  W d    n1 sf0    Y  t jt dksJ d S r$   r'   r   r   r   r   test_contextQ   s
    
Nr9   c                  C   sh   d} d}| t jt< tt|i" t jt |ks4J W d    n1 sH0    Y  t jt | ksdJ d S )Nr%   blue)r(   r)   r*   r   r+   original_valueother_valuer   r   r   test_context_with_dictZ   s    
0r>   c               	   C   s   d} d}| t jt< tdtT tdt|ig" t jt |ksDJ W d    n1 sX0    Y  W d    n1 sv0    Y  t jt | ksJ d S Nr%   r:   r&   )r(   r)   r*   r   r   r   r+   r;   r   r   r   'test_context_with_dict_after_namedstylec   s    
Nr@   c               	   C   s   d} d}| t jt< tdtT tt|idg" t jt tksDJ W d    n1 sX0    Y  W d    n1 sv0    Y  t jt | ksJ d S r?   r'   r;   r   r   r   (test_context_with_dict_before_namedstylen   s    
NrA   c               	   C   s   d} d}d}||i}| t jt< | t j|< tdtb td|g4 t jt tksXJ t j| |ksjJ W d    n1 s~0    Y  W d    n1 s0    Y  t jt | ksJ t j| | ksJ d S )Nr%   ztext.usetexTr&   r'   )r<   Zother_paramr=   dr   r   r   .test_context_with_union_of_dict_and_namedstyley   s    
NrC   c               
   C   s   d} d}t t|i tjt |ks*J t t| dd i}tt4 | W d    n1 sb0    Y  W d    n1 s0    Y  tjt |ksJ W d    n1 s0    Y  d S )Nr%   r:   Zbadparam)r   r+   r*   r(   r)   pytestZraisesKeyError)r<   r=   xr   r   r   test_context_with_badparam   s    <rG   equiv_styles)mpl20default)mpl15ZclassicrI   rK   )idsc              	   C   s   g }| D ]>}t |  |tj  W d    q1 s<0    Y  q|d }t| dd  |dd  D ]\}}||ksjJ qjd S )Nr   r   )r   r+   r   r(   r)   copyzip)rH   Zrc_dictsZstyZrc_basenmrcr   r   r   
test_alias   s    0"rQ   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sHJ d S Nzpath.sketch)r   d      )r(   r)   pltxkcdnpZtestingZbreak_cyclesr   r   r   r   test_xkcd_no_cm   s
    
rX   c                   C   sb   t jd d u sJ t " t jd dks.J W d    n1 sB0    Y  t jd d u s^J d S rR   )r(   r)   rU   rV   r   r   r   r   test_xkcd_cm   s    
0rY   )N)&
contextlibr   pathlibr   tempfiler   r1   ZnumpyrW   rD   Z
matplotlibr(   r   rU   r   Zmatplotlib.style.corer   r   r*   r,   r   r   r!   r#   r-   r7   r8   r9   r>   r@   rA   rC   rG   markZparametrizerQ   rX   rY   r   r   r   r   <module>   sB   

		
