U
    md                     @   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st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st	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stdd |jD ddddgkstdd |jD d
ddd
gkst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stdd |jD ddddgkst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stdd |jD d
d
d
ddgkst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   

