a
    ض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   sb   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   s   g | ]}|  qS  Z	get_color.0lr
   r
   l/Users/vegardjervell/Documents/master/model/venv/lib/python3.9/site-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	   s
    r   c                  C   s   t  \} }|tdg dtdg d  tdD ]}|tdtd q6dd |jD g d	kslJ d
d |jD g dksJ d S )Ncr   marker.*xr   r	   c                 S   s   g | ]}|  qS r
   r   r   r
   r
   r   r      r   z%test_marker_cycle.<locals>.<listcomp>r   c                 S   s   g | ]}|  qS 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 q4dd |jD g dksjJ d	d |jD g d
ksJ d S )Nr   r!   )r   r    r   r	   c                 S   s   g | ]}|  qS r
   r   r   r
   r
   r   r   !   r   z=test_marker_cycle_kwargs_arrays_iterators.<locals>.<listcomp>r   c                 S   s   g | ]}|  qS 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   sb   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ls---:r   r	   c                 S   s   g | ]}|  qS r
   Zget_linestyler   r
   r
   r   r   *   r   z-test_linestylecycle_basic.<locals>.<listcomp>r.   r/   r0   r.   r   r   r
   r
   r   test_linestylecycle_basic%   s
    r3   c                  C   s   t  \} }|tdg dtdg d tdg d  tdD ]}|tdtd qDd	d
 |jD dd
 dD ksJ dd
 |jD g dksJ dd
 |jD g dksJ d S )Nr   r   hatchxxO|-Z	linestyler-   r   r	   c                 S   s   g | ]}|  qS r
   Zget_facecolorr   pr
   r
   r   r   4   r   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   r   r   c                 S   s   g | ]}|  qS r
   Z	get_hatchr:   r
   r
   r   r   6   r   )r6   r7   r8   r6   c                 S   s   g | ]}|  qS r
   r1   r:   r
   r
   r   r   7   r   r2   r   r   r   r   r   fillpatchesr   r
   r
   r   test_fillcycle_basic-   s    rD   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sJ dd |jD g dksJ d S )Nr   r   r4   r5   r    )r"   r#   Dr	   r   r6   )r4   r7   c                 S   s   g | ]}|  qS r
   r9   r:   r
   r
   r   r   H   r   z)test_fillcycle_ignore.<locals>.<listcomp>c                 S   s   g | ]}t j|qS r
   r<   r?   r
   r
   r   r   I   r   )r   r   r   r   c                 S   s   g | ]}|  qS r
   r@   r:   r
   r
   r   r   J   r   )r6   r7   r7   r8   rA   )r   r   tr
   r
   r   test_fillcycle_ignore:   s    rG   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sxJ d S )N	linewidth   r   r	      皙?lwc                 S   s   g | ]}|  qS r
   Zget_linewidthr   r
   r
   r   r   U   r   z0test_property_collision_plot.<locals>.<listcomp>)rK   rK   rK   rI   r   )r   r   r   r   r   r   r   r   rF   r   r
   r
   r   test_property_collision_plotM   s    rP   c                  C   s   t  \} }|jg ddd td}tddD ]}|j||dd q0||| ||| d	d
 |jD dd
 dD ksJ dd
 |jD g dksJ d S )N)rI      r         Zbgcmy)rH   Z	facecolorr	   rJ   r   rK   rL   c                 S   s   g | ]}|  qS r
   r9   r:   r
   r
   r   r   `   r   z0test_property_collision_fill.<locals>.<listcomp>c                 S   s   g | ]}t j|qS r
   r<   r?   r
   r
   r   r   a   r   c                 S   s   g | ]}|  qS r
   rN   r:   r
   r
   r   r   b   r   )rK   rK   rK   rR   rS   )r   r   r   r   rB   rC   rO   r
   r
   r   test_property_collision_fillX   s    rT   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 )NrH   rJ   rI   r   ZrgywkbcmrM   )rJ   rI   )rJ   r   r   )r   rJ   r   )r   r   rJ   dashes   )   rQ   rJ   rQ   kwr.   r/   )rM   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   )rH   )r   r   nextZ
_get_linesZprop_cyclerr   )r   r   propgotr
   r
   r   test_cycle_resety   s    
r`   c                  C   sx  t  \} }tttf |d W d    n1 s:0    Y  tttf |ddg W d    n1 sv0    Y  tttf |dd W d    n1 s0    Y  tttf |dd W d    n1 s0    Y  tttf  |dddh W d    n1 s*0    Y  tttf |jddd W d    n1 sh0    Y  tttf  |dddg W d    n1 s0    Y  tttf  |jddgd	 W d    n1 s0    Y  tttf$ |tddgd	 W d    n1 s,0    Y  tt" |td
dd W d    n1 sj0    Y  d S )NrJ   rI   r   ZfishrH   r   )rH   r   foobar)ra   ZrgbZcmy)r   r   )r   r   pytestZraises	TypeError
ValueErrorr   r   rZ   r
   r
   r   test_invalid_input_forms   s*    (,**0.004re   )Z
matplotlibr=   Zmatplotlib.pyplotZpyplotr   Znumpyr(   rb   r   r   r'   r+   r3   rD   rG   rP   rT   r[   r`   re   r
   r
   r
   r   <module>   s   

