
VW3                 @   s	  d  d l  Z  d  d l m Z m Z d  d l j Z d  d l Z d  d l	 m
 Z
 d  d l m Z e d d g d d d	 d
 g  d d    Z e d d d g d d d	 d
 g  d d    Z e d d g d d d	 d
 g  d d    Z e d d g d d d	 d
 g  d d    Z e d d g d d d	 d
 g  d d    Z e d d g d d d	 d
 g  d d    Z e d d g d d d	 d
 g  d d    Z e d d     Z e d! d"    Z e d# d$    Z e d% k rd  d l Z e j d& d' d( g d) d*  d S)+    N)image_comparisoncleanup)assert_raises)cyclerbaseline_imagesZcolor_cycle_basicZremove_textT
extensionspngc              C   s  t  j   }  |  j d  } | j t d d d d g   t j d  } d | d } | j | | d	 d
 d d d | d } | j | | d	 d d d d | d } | j | | d	 d d d d | d } | j | | d	 d d d | j d d  d  S)No   colorrgy
   g      ?   labelZredlw   g?   Zgreeng?Zyellowg333333?   Zred2locz
upper left)	pltfigureadd_subplotset_prop_cycler   nparangeplotlegend)figaxxsys r"   W/s/python-3.5.2/amd64_ubu14/lib/python3.5/site-packages/matplotlib/tests/test_cycles.pytest_colorcycle_basic   s    r$   Zmarker_cyclec           	   C   sd  t  j   }  |  j d  } | j t d d d d g  t d d d d	 g   t j d
  } d | d } | j | | d d d d d d d | d } | j | | d d d d d d d | d } | j | | d d d d d d d | d } | j | | d d d d d d | j d d  t  j   }  |  j d  } | j d t j	 d d d g  d t
 d d d	 g   t j d
  } d | d } | j | | d d d d d d d | d } | j | | d d d d d d d | d } | j | | d d d d d d d | d } | j | | d d d d d d | j d d  d  S)Nr	   r
   r   r   r   marker.*xr   g      ?r   r   zred dotr   r   ms   g?r   z
green starg?zyellow xg333333?r   zred2 dotr   z
upper left)r   r   r   r   r   r   r   r   r   arrayiter)r   r   r    r!   r"   r"   r#   test_marker_cycle   s8    """"!""""r-   Zlineprop_cycle_basicc              C   s  t  j   }  |  j d  } | j t d d d d g   t j d  } d | d } | j | | d	 d
 d d d | d } | j | | d	 d d d d | d } | j | | d	 d d d d | d } | j | | d	 d d d | j d d  d  S)Nr	   	linestyle-z--:r   g      ?r   r   Zsolidr   r   g?r   Zdashedg?Zdottedg333333?r   Zsolid2r   z
upper left)	r   r   r   r   r   r   r   r   r   )r   r   r    r!   r"   r"   r#   test_linestylecycle_basic@   s    r1   Zfill_cycle_basicc              C   sA  t  j   }  |  j d  } | j t d d d d g  t d d d d	 g  t d
 d d d g   t j d  } d | d d } | j | | d d d d d | d d } | j | | d d d d d | d d } | j | | d d d d d | d d } | j | | d d d d | j d d  d  S) Nr	   r
   r   r   r   hatchxxOz|-r.   r/   z--r0   r   g      ?g      ?r   r   zred, xx	linewidthr   g?zgreen, circleg?r   zyellow, crossg333333?r   zred2, xxr   z
upper left)	r   r   r   r   r   r   r   fillr   )r   r   r    r!   r"   r"   r#   test_fillcycle_basicR   s    +r7   Zfill_cycle_ignorec              C   s8  t  j   }  |  j d  } | j t d d d d g  t d d d d	 g  t d
 d d d g   t j d  } d | d d } | j | | d d d d d d | d d } | j | | d d d d d | d d } | j | | d d d | d d } | j | | d d | j d d  d  S)Nr	   r
   r   r   r   r2   r3   r4   z|-r%   r&   r'   Dr   g      ?g      ?r   r   zred, xxg?r   zred, circleg?r   zgreen, circleg333333?r   zyellow, crossr   z
upper left)	r   r   r   r   r   r   r   r6   r   )r   r   r    r!   r"   r"   r#   test_fillcycle_ignoref   s    +r9   Zproperty_collision_plotc              C   s   t  j   \ }  } | j d d d g  xC t d d  D]2 } | j t j d  | t j d  d d q8 W| j t j d  d t j d   | j t j d  d t j d   d  S)	Nr5   r   r      r   r   g?r   )r   subplotsr   ranger   r   r   )r   r   cr"   r"   r#   test_property_collision_plot~   s    0&r>   Zproperty_collision_fillc              C   s   t  j   \ }  } t j d  } d | d d } | j d d d d d d	 g d
 d  x1 t d d  D]  } | j | | | d d qh W| j | d |  | j | d |  d  S)Nr   g      ?g      ?r   r5   r   r   r      Z	facecolorZbgcmyr:   r   g?)r   r;   r   r   r   r<   r6   )r   r   r    r!   r=   r"   r"   r#   test_property_collision_fill   s    %r@   c           	   C   sp  t  j   \ }  } | j d   | j t d d d g   | j d d  | j d d  | j d d d g  | j d t d d g   | j d t j d d g   | j d t j d d d g d d d g d d d g g   | j d d d g d d d	 g d
 d d g  | j d t j d d g  d t j d d	 g  d
 t j d d g   d slt  d  S)Nr5   r:   r   r
   Zrgywkbcmr   r   kwZlsr/   z--T)r:   r   )r   r;   r   r   r,   r   r+   AssertionError)r   r   r"   r"   r#   test_valid_input_forms   s     .rD   c              C   sP  t  j   \ }  } t | j j  } | j d d d d g  | t | j j  k s[ t  | j d   t | j j  } | | k s t d | | f   t  j   \ }  } t j    t | j j  } | j	 d d d d	 g  | t | j j  k st  | j	 d   t | j j  } | | k sEt d | | f   Wd  QRXd  S)
Nr5   r   	   r   zexpected %s, got %sr=   mr   rA   )
r   r;   nextZ
_get_linesZprop_cyclerr   rC   warningscatch_warningsZset_color_cycle)r   r   propZgotr"   r"   r#   test_cycle_reset   s    "rK   c              C   s   t  j   \ }  } t t t f | j d  t t t f | j d d g  t t t f | j d d  t t t f | j d d  t t t f | j d d d d d i  t t t f | j d d d d d  S)	Nr:   r   r
   Zfishr5   12r   )r   r;   r   	TypeError
ValueErrorr   )r   r   r"   r"   r#   test_invalid_input_forms   s    rP   __main__argvz-sz--with-doctestexitF)rH   Zmatplotlib.testing.decoratorsr   r   Zmatplotlib.pyplotZpyplotr   numpyr   
nose.toolsr   r   r$   r-   r1   r7   r9   r>   r@   rD   rK   rP   __name__nose	runmoduler"   r"   r"   r#   <module>   s2   "
