o
    e%                     @   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ejdk r,dn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sphinx)   
   z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'   N/var/www/myenv/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  }	d|	v sJ t|d sJ t|d  sJ d!|	v sJ d"|	v sJ d#|	v sJ d$|	v sJ |	d%d&ksJ t|d'sJ t|d( sJ | d)  }
|
d*d+}
| d) |
 fd,dd-D }t|  | t|d's(J d j|d. ks6J d& j|d ksDJ d/ j|d& ksRJ t|ds]J t|d&shJ t|d/ssJ d j|d/ ksJ t|dsJ d S )0N	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_directiver2   r3   )parentr4   )r    r'   r(   plot_directive_file>   s   z+test_tinypages.<locals>.plot_directive_filec                    s   g | ]} |qS r'   r'   .0i)r8   r'   r(   
<listcomp>B   s    z"test_tinypages.<locals>.<listcomp>               12_0012_01   some_plots.htmls   # Only a commentz
range4.pngzrange6_range6.pngs    This is the caption for plot 15.s    Plot 17 uses the caption option.s    This is the caption for plot 18.s&   plot-directive my-class my-other-classs#   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<   )r;   r'   r(   r?   j   s    )r@   rI   r   rB   r   r   )shutilcopytreer   __file__r:   r   r   r   r   r   r   devnullr   r   r)   rangefilecmpcmp
read_bytescountreplacewrite_bytesrK   rL   )tmp_pathr!   r#   r$   r%   r&   range_10range_6range_4html_contentscontentsmodification_timesr'   )r    r7   r;   r8   r(   test_tinypages$   sj   
r_   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   rO   r:   rM   copyfilerN   
write_textr)   lenlistglob)rX   r:   r    	html_dir1	html_dir2r'   r'   r(   test_plot_html_show_source_link}   s   
rr   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 )Nr`   ra   rb   rc   r/   rd   z?
.. plot::
    :show-source-link: true

    plt.plot(range(2))
r-   r   rh   plot_html_show_source_link=ri   rf   r@   rj   rX   rs   r:   r    r!   r'   r'   r(   test_show_source_link_true      

rw   c                 C   rt   )Nr`   ra   rb   rc   r/   rd   z@
.. plot::
    :show-source-link: false

    plt.plot(range(2))
r-   r   rh   ru   ri   rf   r   rj   rv   r'   r'   r(   test_show_source_link_false   rx   ry   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/   rh   zplot_srcset=2xri   r   c                    s    d|  | d S r1   r'   )r5   suffr6   r'   r(   r8      s   z&test_srcset_version.<locals>.plot_file)	r@   rI   r   rB   rC   rD   rG      rJ   z.2x)rz   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"rH   zutf-8)encodingzVsrcset="../_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   )rM   rN   r   rO   r:   r)   exists	read_text)rX   r!   r    r8   indstr'   r6   r(   test_srcset_version   s8   r   )N)__doc__rR   r   pathlibr   rM   r   matplotlib.testingr   r   importorskipversion_infor)   r_   rr   markparametrizerw   ry   r   r'   r'   r'   r(   <module>   s&    
Y

