a
    ضa                     @   s   d dl Zd dlZd dlZd dlmZ d dlmZm	Z	 d dl
mZ edsVejdZe	dgddgd	d
dd Ze dd Ze dd Zdd Zejdg ddd Zejdddgdd Zejdddgdd Zd d! ZdS )"    N)_has_tex_package)check_figures_equalimage_comparisonTz$Missing TeX of Ghostscript or dvipngtest_usetexZpdfZpngmpl20)Zbaseline_images
extensionsstylec                  C   s   dt jd< t \} }ddtddddd	}|jd
i | |jdi | |jdi | |jdi | |jdi | dd |jD D ]}|| qdd |jD D ]}|| q|	  d S )NTtext.usetexZbaseline   r   knone)padZ	edgecolorZ	facecolor)ZverticalalignmentsizeZbbox)皙?gffffff?zO\LaTeX\ $\left[\int\limits_e^{2e}\sqrt\frac{\log^3 x}{x}\,\mathrm{d}x \right\}$)r   333333?Zlg)g?r   z$\frac{1}{2}\pi$)g333333?r   z	$p^{3^A}$)g?r   z	$p_{3_2}$c                 S   s   h | ]}|  d  qS )r   get_position.0t r   l/Users/vegardjervell/Documents/master/model/venv/lib/python3.9/site-packages/matplotlib/tests/test_usetex.py	<setcomp>"       ztest_usetex.<locals>.<setcomp>c                 S   s   h | ]}|  d  qS    r   r   r   r   r   r   $   r   )
mplrcParamspltsubplotsdicttextZtextsZaxvlineZaxhlineZset_axis_off)figaxkwargsxyr   r   r   r      s"    
c                 C   s   dt jd< | ddd d S )NTr	         ?z% a commentr   r   r!   Zfig_testZfig_refr   r   r   
test_empty)   s    
r*   c                 C   s*   dt jd< | ddd |ddd d S )NTr	   r'   z$-$u   −r(   r)   r   r   r   test_unicode_minus/   s    
r+   c                  C   s:   dt jd< t  } |  dd dtjd< | j  d S )NTzaxes.formatter.use_mathtextr   r	   )r   r   figureZadd_subplotZset_xlimr   canvasdraw)r"   r   r   r   test_mathdefault6   s
    

r0   fontsize)   
      c                 C   s   t jd | t jd< i }t }dD ]b}|  |D ]}|jddd| ddd q6|j	  t
|j d d	kjd
d ||< q&th | d
ksJ d S )Nr   z	font.size)r   )r,   )r,   r   r'   $TZusetex).r      r   )Zaxis)r   r   Zuser   r   r-   Zclfr!   r.   r/   nparrayZbuffer_rgbaanysumlenvalues)r1   Zheightsr"   valsr%   r   r   r   test_minus_no_descent@   s    

r?   pkgZxcolorZchemformulac                 C   s   t | st|  d dtjd< t }|ddd}|j	  d|  tjd< t }|ddd}|j	  t
j| |  d S )Nz is not availableTr	   r'   zSome text 0123456789z8\PassOptionsToPackage{dvipsnames}{xcolor}\usepackage{%s}text.latex.preamble)r   pytestskipr   r   r   r-   r!   r.   r/   r8   ZtestingZassert_array_equalZget_window_extent)r@   r"   r!   Ztext2r   r   r   test_usetex_packagesT   s    


rD   preamblez\usepackage[full]{textcomp}z\usepackage{underscore}c                 C   s2   | t jd< t  }|jddddd |j  d S )NrA   r'   zhello, worldTr6   )r   r   r-   r!   r.   r/   )rE   r"   r   r   r   test_latex_pkg_already_loadedg   s    
rF   c                  C   sh   dt jd< tdd d d tdd} t  \}}|jdd| d |  |jd	d	d
dd t   d S )NTr	      r,   )a_bcrI   rH   )datar   Zfoo_barr6   )r   r   ranger   ZplotZlegendr!   r/   )Zdfr"   r#   r   r   r   test_usetex_with_underscorer   s    
rL   )Znumpyr8   rB   Z
matplotlibr   Zmatplotlib.testingr   Zmatplotlib.testing.decoratorsr   r   Zmatplotlib.pyplotZpyplotr   Zcheckdep_usetexmarkrC   Z
pytestmarkr   r*   r+   r0   Zparametrizer?   rD   rF   rL   r   r   r   r   <module>   s8   







