o
    Õj÷  ã                   @   sÎ   d dl Z d dlZd dlZd dlmZ d dlmZ d dl	m
Z d dlmZ edgdddd	d
„ ƒZedgddddd„ ƒZedgdde  ¡ dkrId ndddd„ ƒZdd„ Zdd„ Zdd„ Zdd„ Zdd„ ZdS )é    N)Ú
same_color)Úimage_comparison)Úart3dzlegend_plot.pngTZmpl20)Úremove_textÚstylec                  C   s\   t jtddd\} }t d¡}|j|d| dddd	 |j||d ddd
d	 | ¡  d S )NÚ3d©Ú
projection©Z
subplot_kwé
   é   ÚoÚyzz=1)ÚzdirÚlabelzz=-1)ÚpltÚsubplotsÚdictÚnpÚarangeÚplotÚlegend)ÚfigÚaxÚx© r   úi/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/mpl_toolkits/mplot3d/tests/test_legend3d.pyÚtest_legend_plot   s
   
r   zlegend_bar.pngc                  C   sr   t jtddd\} }t d¡}|j||dddd}|j||d d d	… d
ddd}| |d |d gddg¡ d S )Nr   r   r
   r   r   ÚedgeÚm)r   ÚalignÚcoloréÿÿÿÿr   Úgr   ÚupZdown)r   r   r   r   r   Úbarr   )r   r   r   Úb1Úb2r   r   r   Útest_legend_bar   s
   
 r(   z	fancy.pngÚx86_64gºI+‡†?)r   r   Ztolc                  C   s¢   t jtddd\} }|jt d¡t dd¡t dd¡ddd |jt d¡t dd	d
¡dd |jt dd¡t d¡t dd¡dddd |j	ddddd d S )Nr   r   r
   r   r   zo--Úline©r   r   r"   Úscatterç      à?Úerrorbar)ZxerrZzerrr   z
lower lefté   z	My legendé   )ÚlocZncolsÚtitleÚ	numpoints)
r   r   r   r   r   r   Úfullr,   r.   r   )r   r   r   r   r   Ú
test_fancy   s   *  ÿr5   c                  C   s  ddgddgg} ddgddggd	dgd
dggg}d
dgddggd	dgddggg}t j| ddd}t j|dd}t j|dd	d}tjtddd\}}| |¡ | |¡ | |¡ | |||gg d¢¡}|j\}	}
}t|||f|	|
|fƒD ]\}}| 	¡ d |j
ks„J ‚qud S )N)r   r-   )r-   r0   )g333333Ó?ç333333ã?)çš™™™™™É?r7   gffffffæ?r7   gš™™™™™é?gš™™™™™Ù?r-   r6   gš™™™™™¹?z--é   )Ú
linestylesZlwz-.)r9   ú:r   r   r
   )Úline1Úline2zline 3r   )r   ZLine3DCollectionr   r   r   Zadd_collectionr   Úlegend_handlesÚzipZget_linestylesZ_dash_pattern)Zlines1Zlines2Zlines3Zlc1Zlc2Zlc3r   r   ÚlegZh1Úh2Zh3ZohZlhr   r   r   Ú!test_linecollection_scaled_dashes)   s     


ÿrA   c                  C   s|   t jtddd\} }|jddgddgdd tjdgdgdgddg}|j|dgdd	}|d  ¡ |jd  ¡ ks<J ‚d S )
Nr   r   r
   r   r0   Úv)ÚmarkerZAardvark)r3   )	r   r   r   r,   r   ZLine3Dr   Z
get_markerr=   )r   r   Zhandlesr?   r   r   r   Útest_handlerline3d=   s
   "rD   c            	      C   s¬   t jdd…dd…f \} }| | }g d¢}tjtddd\}}|j| ||g d¢|dd	}| ¡ \}}|g d
¢ks;J ‚tdd„ |D ƒƒsFJ ‚tdd„ t||ƒD ƒƒsTJ ‚d S )Nr0   r   )Úbluez#00FF00Úredr   r   r
   ©r   é   é2   Úboth©ZlevelsÚcolorsÚextend)z
$x = 10.0$z
$x = 30.0$z
$x = 50.0$c                 s   ó    | ]
}t |tjjƒV  qd S ©N)Ú
isinstanceÚmplÚlinesZLine2D©Ú.0Úar   r   r   Ú	<genexpr>P   ó   € z/test_contour_legend_elements.<locals>.<genexpr>c                 s   ó"    | ]\}}t | ¡ |ƒV  qd S rO   )r   Ú	get_color©rT   rU   Úcr   r   r   rV   Q   ó   € ÿ)	r   Úmgridr   r   r   ZcontourÚlegend_elementsÚallr>   )	r   r   ÚhrL   r   r   ÚcsÚartistsÚlabelsr   r   r   Útest_contour_legend_elementsF   s   ÿrd   c            	      C   sÌ   t jdd…dd…f \} }| | }tjtddd\}}|j| ||g d¢g d¢dd	}|j d
¡ |j d¡ | 	¡  | 
¡ \}}|g d¢ksIJ ‚d}tdd„ |D ƒƒsVJ ‚tdd„ t||ƒD ƒƒsdJ ‚d S )Nr0   r   r   r   r
   rG   )ú#FFFF00ú#FF00FFz#00FFFFrJ   rK   rF   rE   )z$x \leq -1e+250s$z$10.0 < x \leq 30.0$z$30.0 < x \leq 50.0$z$x > 1e+250s$)rE   re   rf   rF   c                 s   rN   rO   )rP   rQ   ZpatchesZ	RectanglerS   r   r   r   rV   f   rW   z0test_contourf_legend_elements.<locals>.<genexpr>c                 s   rX   rO   )r   Úget_facecolorrZ   r   r   r   rV   g   r\   )r   r]   r   r   r   ZcontourfZcmapZset_overZ	set_underÚchangedr^   r_   r>   )	r   r   r`   r   r   ra   rb   rc   Zexpected_colorsr   r   r   Útest_contourf_legend_elementsU   s"   þÿri   c                  C   s~   t  g d¢g d¢g d¢g¡} tj| gdd}tjddid\}}| d	¡ | |¡}| ¡ }|j	d
  
¡ | 
¡ k ¡ s=J ‚d S )N)r   r   r   )r   r0   r0   )r0   r   r0   Zsurfacer+   r	   r   r
   Úkr   )r   Zasarrayr   ZPoly3DCollectionr   r   Zset_edgecolorZadd_collection3dr   r=   rg   r_   )ZvertsZmeshr   r   Úhandler?   r   r   r   Útest_legend_Poly3dCollectionk   s   

ÿ
ÿrl   )ÚplatformÚnumpyr   Z
matplotlibrQ   Zmatplotlib.colorsr   Zmatplotlib.testing.decoratorsr   Zmatplotlib.pyplotZpyplotr   Zmpl_toolkits.mplot3dr   r   r(   Úmachiner5   rA   rD   rd   ri   rl   r   r   r   r   Ú<module>   s(    


ÿ
		