U
    Âmœd  ã                   @   sŒ   d dl Zd dlmZ d dlZd dlZd dlm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dS )é    N)Úcyclerc                  C   sh   t  ¡ \} }| tddddgƒ¡ tdƒD ]}| tdƒtdƒ¡ q*dd„ |jD ƒddddgksdt‚d S )	NÚcolorÚrÚgÚyé   é
   c                 S   s   g | ]}|  ¡ ‘qS © ©Ú	get_color©Ú.0Úlr	   r	   úU/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/matplotlib/tests/test_cycles.pyÚ
<listcomp>   s     z)test_colorcycle_basic.<locals>.<listcomp>©ÚpltÚsubplotsÚset_prop_cycler   ÚrangeÚplotÚlinesÚAssertionError©ÚfigÚaxÚ_r	   r	   r   Útest_colorcycle_basic	   s
    r   c                  C   s˜   t  ¡ \} }| tddddgƒtddddgƒ ¡ td	ƒD ]}| td
ƒtd
ƒ¡ q:dd„ |jD ƒddddgkstt‚dd„ |jD ƒddddgks”t‚d S )NÚcr   r   r   ÚmarkerÚ.Ú*Úxr   r   c                 S   s   g | ]}|  ¡ ‘qS r	   r
   r   r	   r	   r   r      s     z%test_marker_cycle.<locals>.<listcomp>c                 S   s   g | ]}|  ¡ ‘qS r	   ©Z
get_markerr   r	   r	   r   r      s     r   r   r	   r	   r   Útest_marker_cycle   s    ÿ r$   c                  C   s–   t  ¡ \} }|jt dddg¡tdddgƒd� tdƒD ]}| td	ƒtd	ƒ¡ q8d
d„ |jD ƒddddgksrt	‚dd„ |jD ƒddddgks’t	‚d S )Nr   r   r   r    r!   r"   )r   r   r   r   c                 S   s   g | ]}|  ¡ ‘qS r	   r
   r   r	   r	   r   r   !   s     z=test_marker_cycle_kwargs_arrays_iterators.<locals>.<listcomp>c                 S   s   g | ]}|  ¡ ‘qS r	   r#   r   r	   r	   r   r   "   s     )
r   r   r   ÚnpÚarrayÚiterr   r   r   r   r   r	   r	   r   Ú)test_marker_cycle_kwargs_arrays_iterators   s    ÿ r(   c                  C   sh   t  ¡ \} }| tddddgƒ¡ tdƒD ]}| tdƒtdƒ¡ q*dd„ |jD ƒddddgksdt‚d S )	NÚlsú-ú--ú:r   r   c                 S   s   g | ]}|  ¡ ‘qS r	   ©Zget_linestyler   r	   r	   r   r   *   s     z-test_linestylecycle_basic.<locals>.<listcomp>r   r   r	   r	   r   Útest_linestylecycle_basic%   s
    r.   c                  C   sÊ   t  ¡ \} }| tddddgƒtddddgƒ td	d
ddgƒ ¡ tdƒD ]}| tdƒtdƒ¡ qJdd„ |jD ƒdd„ dD ƒks†t‚dd„ |jD ƒddddgks¦t‚dd„ |jD ƒd
ddd
gksÆt‚d S )Nr   r   r   r   ÚhatchÚxxÚOú|-Z	linestyler*   r+   r,   r   r   c                 S   s   g | ]}|  ¡ ‘qS r	   ©Zget_facecolor©r   Úpr	   r	   r   r   4   s     z(test_fillcycle_basic.<locals>.<listcomp>c                 S   s   g | ]}t j |¡‘qS r	   ©ÚmplÚcolorsZto_rgba©r   r   r	   r	   r   r   5   s     )r   r   r   r   c                 S   s   g | ]}|  ¡ ‘qS r	   ©Z	get_hatchr4   r	   r	   r   r   6   s     c                 S   s   g | ]}|  ¡ ‘qS r	   r-   r4   r	   r	   r   r   7   s     ©r   r   r   r   r   ÚfillÚpatchesr   r   r	   r	   r   Útest_fillcycle_basic-   s    ÿþÿ r>   c                  C   sÊ   t  ¡ \} }| tddddgƒtddddgƒ td	d
ddgƒ ¡ tdƒ}|j||ddd� |j||dd� | ||¡ | ||¡ dd„ |jD ƒdd„ dD ƒks¦t‚dd„ |jD ƒddddgksÆt‚d S )Nr   r   r   r   r/   r0   r1   r2   r   r    r!   ÚDr   )r/   c                 S   s   g | ]}|  ¡ ‘qS r	   r3   r4   r	   r	   r   r   H   s     z)test_fillcycle_ignore.<locals>.<listcomp>c                 S   s   g | ]}t j |¡‘qS r	   r6   r9   r	   r	   r   r   I   s     )r   r   r   r   c                 S   s   g | ]}|  ¡ ‘qS r	   r:   r4   r	   r	   r   r   J   s     r;   )r   r   Útr	   r	   r   Útest_fillcycle_ignore:   s    ÿþÿrA   c                  C   s‚   t  ¡ \} }| dddg¡ tdƒ}tddƒD ]}|j||dd� q.| ||¡ | ||¡ dd	„ |jD ƒdddddgks~t‚d S )
NÚ	linewidthé   r   r   é   çš™™™™™¹?©Úlwc                 S   s   g | ]}|  ¡ ‘qS r	   ©Zget_linewidthr   r	   r	   r   r   U   s     z0test_property_collision_plot.<locals>.<listcomp>)r   r   r   r   r   r   r   ©r   r   r@   r   r	   r	   r   Útest_property_collision_plotM   s    rJ   c                  C   s¬   t  ¡ \} }|jdddddgdd� tdƒ}td	dƒD ]}|j||d
d� q6| ||¡ | ||¡ dd„ |jD ƒdd„ dD ƒks†t‚dd„ |jD ƒd
d
d
ddgks¨t‚d S )NrC   é   r   é   é   Zbgcmy)rB   Z	facecolorr   rD   rE   rF   c                 S   s   g | ]}|  ¡ ‘qS r	   r3   r4   r	   r	   r   r   `   s     z0test_property_collision_fill.<locals>.<listcomp>c                 S   s   g | ]}t j |¡‘qS r	   r6   r9   r	   r	   r   r   a   s     c                 S   s   g | ]}|  ¡ ‘qS r	   rH   r4   r	   r	   r   r   b   s     )r   r   r   r   r<   r=   r   rI   r	   r	   r   Útest_property_collision_fillX   s    ÿrN   c               
   C   s  t  ¡ \} }| d ¡ | tdddgƒ¡ | dd¡ | dd¡ | dddg¡ | dtddgƒ¡ | dt ddg¡¡ | dt dddgdddgdddgg¡¡ | d	g d
dgddddgg¡ |jddgddgddgd� |jt ddg¡t ddg¡t ddg¡d� d S )NrB   rD   rC   r   ZrgywkbcmrG   )rD   rC   r   Údashesé   é   rK   ÚkÚwr*   r+   )rG   r   r)   )r   r   r   r   r'   r%   r&   ©r   r   r	   r	   r   Útest_valid_input_formse   s$    
þþrU   c                  C   sd   t  ¡ \} }t|jjƒ}|jdddgd� |t|jjƒks>t‚| d ¡ t|jjƒ}||ks`t‚d S )Nr   é	   r   )rB   )r   r   ÚnextZ
_get_linesZprop_cyclerr   r   )r   r   ÚpropÚgotr	   r	   r   Útest_cycle_resety   s    
rZ   c               	   C   s¤  t  ¡ \} }t ttf¡� | d¡ W 5 Q R X t ttf¡� | ddg¡ W 5 Q R X t ttf¡� | dd¡ W 5 Q R X t ttf¡� | dd¡ W 5 Q R X t ttf¡� | dddh¡ W 5 Q R X t ttf¡� |jddd� W 5 Q R X t ttf¡� | dddg¡ W 5 Q R X t ttf¡� |jddgd	� W 5 Q R X t ttf¡� | tddgd	�¡ W 5 Q R X t t¡� | td
dd�¡ W 5 Q R X d S )NrD   rC   r   ÚfishrB   r   )rB   r   Úfoobar)r\   ÚrgbZcmy)r   r   )r   r   ÚpytestZraisesÚ	TypeErrorÚ
ValueErrorr   r   rT   r	   r	   r   Útest_invalid_input_forms†   s*    ra   )Z
matplotlibr7   Zmatplotlib.pyplotZpyplotr   Únumpyr%   r^   r   r   r$   r(   r.   r>   rA   rJ   rN   rU   rZ   ra   r	   r	   r	   r   Ú<module>   s   

