o
    Ďi&                     @   s   d 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mZ ddl	Z	e	j
ddd ddd	Zd
d Z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 )z2Tests for tinypages build using sphinx extensions.    N)Path)subprocess_run_for_testingsphinxz4.1.3)
minversionc              
   C   s   |d u rg n|}t jdddddt|t| t|g	|}t|ddi tjddid	}|j}|j}|jd
ksAJ d| d| d|rKt	
d|  | sQJ d S )N-msphinx-W-bhtml-dT
MPLBACKEND capture_outputtextenvr   z!sphinx build failed with stdout:
z	
stderr:

z-sphinx build emitted the following warnings:
)sys
executablestrr   osenvironstdoutstderr
returncodepytestfailis_dir)
source_dirdoctree_dirhtml_dir
extra_argscmdprocouterr r%   [/home/jeff/fluffinator/venv/lib/python3.10/site-packages/matplotlib/tests/test_sphinxext.pybuild_sphinx_html   s$   r'   c              
      s.  t jttjd | dd | d d }|d | d  tjdd	d
ddt tttjd t|g	}t|ddi t	j
dt	jdd}|j}|j}t|  | fdd fddfddtddD \}}}t|dstJ t|ds~J t|dsJ t|dsJ t|dsJ t|dsJ |d jdd}	d |	v sJ t|d! sJ t|d" sJ d#|	v sJ d$|	v sJ d%|	v sJ d&|	v sJ d'|	v sJ |	d(d)ksJ t|d*sJ t|d+ sJ | d,  }
|
d-d.}
| d, |
 fd/d0d1D }t|  | t|d*s1J d j|d2 ks?J d) j|d ksMJ d3 j|d) ks[J t|dsfJ t|d)sqJ t|d3s|J d j|d3 ksJ t|dsJ d S )4N	tinypagesTdirs_exist_ok_buildr	   _imagesdoctreesr   r   r   r
   r   )r   GCOV_ERROR_FILEr   c                    s    d|  d S Nsome_plots-.pngr%   numimg_dirr%   r&   	plot_file:   s   z!test_tinypages.<locals>.plot_filec                    s    j d d|  d S )Nplot_directiver0   r1   )parentr2   )r   r%   r&   plot_directive_file=   s   z+test_tinypages.<locals>.plot_directive_filec                 3   s    | ]} |V  qd S Nr%   .0i)r6   r%   r&   	<genexpr>A   s    z!test_tinypages.<locals>.<genexpr>               12_0012_01   some_plots.htmlutf-8encodingz# Only a commentz
range4.pngzrange6_range6.pngz This is the caption for plot 15.z7Plot 17 uses the caption option,
with multi-line input.z3Plot 17 uses the alt option, with multi-line input.z This is the caption for plot 18.z&plot-directive my-class my-other-classz#This caption applies to both plots.      zrange6_range10.pngzincluded_plot_21.rsts   plt.plot(range(6))s   plt.plot(range(4))c                    s   g | ]	} |  jqS r%   )statst_mtimer;   )r9   r%   r&   
<listcomp>l   s    z"test_tinypages.<locals>.<listcomp>)r?   rK      rA   r   rP   )shutilcopytreer   __file__r8   r   r   r   r   r   r   devnullr   r   r'   rangefilecmpcmp	read_textcount
read_bytesreplacewrite_bytesrM   rN   )tmp_pathr   r!   r"   r#   r$   range_10range_6range_4html_contentscontentsmodification_timesr%   )r   r5   r9   r6   r&   test_tinypages#   sl   
rd   c                 C   s   t tj}t|d | d  t|d | d  | d }| d d | d d	 }t| || tt	|
d
dks=J | d d }t| ||ddgd tt	|
d
dksZJ d S )Ntinypages/conf.pyconf.pytinypages/_static_staticr-   	index.rstz#
.. plot::

    plt.plot(range(2))
r+   html1**/index-1.pyr?   html2-Dzplot_html_show_source_link=0r    r   r   rS   r8   rQ   copyfilerR   
write_textr'   lenlistglob)r]   r8   r   	html_dir1	html_dir2r%   r%   r&   test_plot_html_show_source_link   s   
rw   plot_html_show_source_linkr?   c                 C      t tj}t|d | d  t|d | d  | d }| d d | d d	 }t| ||d
d| gd tt	|
ddksDJ d S )Nre   rf   rg   rh   r-   ri   z?
.. plot::
    :show-source-link: true

    plt.plot(range(2))
r+   r	   rm   plot_html_show_source_link=rn   rk   r?   ro   r]   rx   r8   r   r   r%   r%   r&   test_show_source_link_true      

r|   c                 C   ry   )Nre   rf   rg   rh   r-   ri   z@
.. plot::
    :show-source-link: false

    plt.plot(range(2))
r+   r	   rm   rz   rn   rk   r   ro   r{   r%   r%   r&   test_show_source_link_false   r}   r~   c                    sZ  t jttjd | dd | d d }|d  | d }t| ||dd	gd
 d! fdd	}dD ]}|| s8J ||dd sBJ q. d  sKJ  d  sSJ  d  s[J  d  scJ  d  skJ  d  ssJ  d  s{J  d  sJ d|d jddv sJ d}||d jddv sJ d}||d  jddv sJ d S )"Nr(   Tr)   r+   r	   r,   r-   rm   zplot_srcset=2xrn   r   c                    s    d|  | d S r/   r%   )r3   suffr4   r%   r&   r6      s   z&test_srcset_version.<locals>.plot_file)	r?   rK   rP   rA   rB   rC   rF      rL   z.2x)r   znestedpage-index-1.pngznestedpage-index-1.2x.pngznestedpage-index-2.pngznestedpage-index-2.2x.pngznestedpage2-index-1.pngznestedpage2-index-1.2x.pngznestedpage2-index-2.pngznestedpage2-index-2.2x.pngzDsrcset="_images/some_plots-1.png, _images/some_plots-1.2x.png 2.00x"rG   rH   rI   zVsrcset="../_images/nestedpage-index-1.png, ../_images/nestedpage-index-1.2x.png 2.00x"znestedpage/index.htmlzXsrcset="../_images/nestedpage2-index-2.png, ../_images/nestedpage2-index-2.2x.png 2.00x"znestedpage2/index.html)r   )rQ   rR   r   rS   r8   r'   existsrX   )r]   r   r   r6   indstr%   r4   r&   test_srcset_version   s8   r   r:   )__doc__rV   r   pathlibr   rQ   r   matplotlib.testingr   r   importorskipr'   rd   rw   markparametrizer|   r~   r   r%   r%   r%   r&   <module>   s"    
\

