o
    :a                     @   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   b   t  \} }|tdg d tdD ]}|tdtd qdd |jD g dks/J d S )Ncolorrgy   
   c                 S      g | ]}|  qS  Z	get_color.0lr   r   >/usr/lib/python3/dist-packages/matplotlib/tests/test_cycles.py
<listcomp>       z)test_colorcycle_basic.<locals>.<listcomp>r   r   r   r   pltsubplotsset_prop_cycler   rangeplotlinesfigax_r   r   r   test_colorcycle_basic	   
    r    c                  C   s   t  \} }|tdg dtdg d  tdD ]}|tdtd qdd |jD g d	ks6J d
d |jD g dksDJ d S )Ncr   marker.*xr	   r
   c                 S   r   r   r   r   r   r   r   r      r   z%test_marker_cycle.<locals>.<listcomp>r   c                 S   r   r   Z
get_markerr   r   r   r   r      r   r%   r&   r'   r%   r   r   r   r   r   test_marker_cycle   s    r*   c                  C   s   t  \} }|jtg dtg dd tdD ]}|tdtd qdd |jD g dks5J d	d |jD g d
ksCJ d S )Nr   r$   )r"   r#   r	   r
   c                 S   r   r   r   r   r   r   r   r   !   r   z=test_marker_cycle_kwargs_arrays_iterators.<locals>.<listcomp>r   c                 S   r   r   r(   r   r   r   r   r   "   r   r)   )	r   r   r   nparrayiterr   r   r   r   r   r   r   )test_marker_cycle_kwargs_arrays_iterators   s   
 r.   c                  C   r   )Nls---:r	   r
   c                 S   r   r   Zget_linestyler   r   r   r   r   *   r   z-test_linestylecycle_basic.<locals>.<listcomp>r1   r2   r3   r1   r   r   r   r   r   test_linestylecycle_basic%   r!   r6   c                  C   s   t  \} }|tdg dtdg d tdg d  tdD ]}|tdtd q"d	d
 |jD dd
 dD ks@J dd
 |jD g dksNJ dd
 |jD g dks\J d S )Nr"   r   hatchxxO|-Z	linestyler0   r	   r
   c                 S   r   r   Zget_facecolorr   pr   r   r   r   4   r   z(test_fillcycle_basic.<locals>.<listcomp>c                 S      g | ]}t j|qS r   mplZcolorsZto_rgbar   r"   r   r   r   r   5       r   c                 S   r   r   Z	get_hatchr=   r   r   r   r   6   r   )r9   r:   r;   r9   c                 S   r   r   r4   r=   r   r   r   r   7   r   r5   r   r   r   r   r   fillpatchesr   r   r   r   test_fillcycle_basic-   s    rH   c                  C   s   t  \} }|tdg dtdg d tdg d  td}|j||dd	d
 |j||dd
 ||| ||| dd |jD dd dD ksPJ dd |jD g dks^J d S )Nr   r   r7   r8   r#   )r%   r&   Dr
   r   r9   )r7   r:   c                 S   r   r   r<   r=   r   r   r   r   H   r   z)test_fillcycle_ignore.<locals>.<listcomp>c                 S   r?   r   r@   rB   r   r   r   r   I   rC   )r   r   r   r   c                 S   r   r   rD   r=   r   r   r   r   J   r   )r9   r:   r:   r;   rE   )r   r   tr   r   r   test_fillcycle_ignore:   s    rK   c                  C   s|   t  \} }|dddg td}tddD ]
}|j||dd q||| ||| dd	 |jD g d
ks<J d S )N	linewidth   r	   r
      皙?lwc                 S   r   r   Zget_linewidthr   r   r   r   r   U   r   z0test_property_collision_plot.<locals>.<listcomp>)rO   rO   rO   rM   r	   )r   r   r   r   r   r   r   r   rJ   r"   r   r   r   test_property_collision_plotM   s    rT   c                  C   s   t  \} }|jg ddd td}tddD ]
}|j||dd q||| ||| d	d
 |jD dd
 dD ks@J dd
 |jD g dksNJ d S )N)rM      r	         Zbgcmy)rL   Z	facecolorr
   rN   r	   rO   rP   c                 S   r   r   r<   r=   r   r   r   r   `   r   z0test_property_collision_fill.<locals>.<listcomp>c                 S   r?   r   r@   rB   r   r   r   r   a   rC   c                 S   r   r   rR   r=   r   r   r   r   b   r   )rO   rO   rO   rV   rW   )r   r   r   r   rF   rG   rS   r   r   r   test_property_collision_fillX   s    rX   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g dg d	g d
g |dg ddgg d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 )NrL   rN   rM   r   ZrgywkbcmrQ   )rN   rM   )rN   r   r   )r   rN   r   )r   r   rN   Zdashes   )   rU   rN   rU   kwr1   r2   )rQ   r   r/   )r   r   r   r   r-   r+   r,   r   r   r   r   r   test_valid_input_formse   s$   

r^   c                  C   sb   t  \} }t|jj}|jg dd |t|jjksJ |d  t|jj}||ks/J d S )N)r
   	   r	   )rL   )r   r   nextZ
_get_linesZprop_cyclerr   )r   r   ZpropZgotr   r   r   test_cycle_resety   s   
ra   c                  C   sr  t  \} }tttf |d W d    n1 sw   Y  tttf |ddg W d    n1 s;w   Y  tttf |dd W d    n1 sXw   Y  tttf |dd W d    n1 suw   Y  tttf |dddh W d    n1 sw   Y  tttf |jddd W d    n1 sw   Y  tttf |dddg W d    n1 sw   Y  tttf |jddgd	 W d    n1 sw   Y  tttf |tddgd	 W d    n	1 sw   Y  tt |td
dd W d    d S 1 s2w   Y  d S )NrN   rM   r   ZfishrL   r   )rL   r   foobar)rb   ZrgbZcmy)r   r"   )r   r   pytestZraises	TypeError
ValueErrorr   r   r]   r   r   r   test_invalid_input_forms   s>   $rf   )Z
matplotlibrA   Zmatplotlib.pyplotZpyplotr   Znumpyr+   rc   r   r    r*   r.   r6   rH   rK   rT   rX   r^   ra   rf   r   r   r   r   <module>   s     

