ó
šxŠ\c           @   s   d  Z  d d l Z d d l j j Z d d l m Z m	 Z	 d d l
 m Z d d l j j Z e j d e f d „  ƒ  Yƒ Z d S(   s    Test cases for GroupBy.plot iÿÿÿÿN(   t	   DataFramet   Series(   t   TestPlotBaset   TestDataFrameGroupByPlotsc           B   s5   e  Z d  „  Z d „  Z d „  Z d „  Z d „  Z RS(   c         C   sİ   d } t  t j j d d d | ƒƒ } t  t j j d d d | ƒƒ } t j d ƒ # t j j d d g d | ƒ} Wd  QX| j | ƒ j ƒ  t j	 ƒ  | j | ƒ j
 ƒ  t j	 ƒ  | j | ƒ j d	 d
 ƒ t j	 ƒ  d  S(   Ni
   i¦   i   t   sizei<   i*   t   malet   femalet   alphag      à?(   R   t   npt   randomt   normalt   tmt
   RNGContextt   choicet   groupbyt   plott   closet   hist(   t   selft   nt   weightt   heightt   gender(    (    sA   lib/python2.7/site-packages/pandas/tests/plotting/test_groupby.pyt,   test_series_groupby_plotting_nominally_works   s    !!$

c         C   s®   t  i d d d d d d d d d g	 d 6t j j d ƒ d 6d d d	 d
 d d	 d
 d d	 d
 g	 ƒ} | j d ƒ d j ƒ  t j ƒ  | j d ƒ d j d „  ƒ t j ƒ  d  S(   Ni   i   i   t   defi	   t   valt   indexg      ğ?g       @g      @c         S   s
   |  j  ƒ  S(   N(   R   (   t   x(    (    sA   lib/python2.7/site-packages/pandas/tests/plotting/test_groupby.pyt   <lambda>)   s    (	   R    R   R	   t   randnR   R   R   R   t   apply(   R   t   df(    (    sA   lib/python2.7/site-packages/pandas/tests/plotting/test_groupby.pyt$   test_plotting_with_float_index_works!   s    ($
c         C   s²   t  j d d d ƒ } t i d d g d 6d d g d 6d	 d
 g d 6ƒ } | d j d | d d | ƒ t i d g d 6d g d 6d	 g d 6ƒ } | d j d | d d | ƒ d  S(   NiP   id   i   i   t   AAAt   BBBt   Namet   ByColiU   iY   t   Markt   byt   binsif   (   R   t   arangeR    R   (   R   R'   R   (    (    sA   lib/python2.7/site-packages/pandas/tests/plotting/test_groupby.pyt   test_hist_single_row,   s    *c         C   s“   t  i d d d d d g d 6d d d d d g d 6t d ƒ d	 6ƒ } | j d	 ƒ j j d d ƒ t j ƒ  | j d	 ƒ d j j ƒ  t j ƒ  d  S(
   Ni   i   i   i   i   R   t   yt   ababat   z(   R    t   listR   R   t   scatterR   R   t   line(   R   R   (    (    sA   lib/python2.7/site-packages/pandas/tests/plotting/test_groupby.pyt   test_plot_submethod_works6   s    
c         C   sÒ   t  i d d d d d g d 6d d d d d g d 6t d ƒ d	 6ƒ } | j d	 ƒ j d
 d d d d d ƒ } t | d j ƒ d k s‹ t ‚ | j d	 ƒ j j d d d d ƒ } t | d j ƒ d k sÎ t ‚ d  S(   Ni   i   i   i   i   R   R*   R+   R,   t   kindR.   t   a(   R    R-   R   R   t   lent   collectionst   AssertionErrorR.   (   R   R   t   res(    (    sA   lib/python2.7/site-packages/pandas/tests/plotting/test_groupby.pyt   test_plot_kwargs?   s    '$(   t   __name__t
   __module__R   R    R)   R0   R7   (    (    (    sA   lib/python2.7/site-packages/pandas/tests/plotting/test_groupby.pyR      s
   			
		(   t   __doc__t   numpyR   t   pandas.util._test_decoratorst   utilt   _test_decoratorst   tdt   pandasR    R   t   pandas.tests.plotting.commonR   t   pandas.util.testingt   testingR   t   skip_if_no_mplR   (    (    (    sA   lib/python2.7/site-packages/pandas/tests/plotting/test_groupby.pyt   <module>   s   	