U
    sVc                     @   sl   d Z ddlZddlZddlm  mZ ddlm	Z	m
Z
mZ ddlmZ ddlmZ ejG dd deZdS )z Test cases for GroupBy.plot     N)	DataFrameIndexSeries)TestPlotBasec                   @   sx   e Zd Zdd Zdd Zdd Zdd Zd	d
 Zej	
dddgdd Zej	
dddgdd Zdd Zdd ZdS )TestDataFrameGroupByPlotsc              	   C   s   d}t tjjdd|d}t tjjdd|d}td tjjddg|d}W 5 Q R X ||  t	  ||
  t	  ||jd	d
 t	  d S )N
         )size<   *   ZmaleZfemaleg      ?)alpha)r   nprandomnormaltmZ
RNGContextchoicegroupbyplotclosehist)selfnZweightheightZgender r   F/tmp/pip-unpacked-wheel-xj8nt62q/pandas/tests/plotting/test_groupby.py,test_series_groupby_plotting_nominally_works   s    zFTestDataFrameGroupByPlots.test_series_groupby_plotting_nominally_worksc                 C   s|   t dddddddddg	tjdddddddddddg	d	}|d
d   t  |d
d dd  t  d S )N         	   )defvalg      ?g       @g      @)indexr!   r"   c                 S   s   |   S )N)r   )xr   r   r   <lambda>,       zPTestDataFrameGroupByPlots.test_plotting_with_float_index_works.<locals>.<lambda>)	r   r   r   randnr   r   r   r   applyr   dfr   r   r   $test_plotting_with_float_index_works#   s    "z>TestDataFrameGroupByPlots.test_plotting_with_float_index_worksc                 C   sp   t ddd}tddgddgddgd	}|d
 j|d |d tdgdgdgd	}|d
 j|d |d d S )NP   f   r   ZAAAZBBBr   U   Y   )NameByColMarkr2   r1   )Zbybins)r   Zaranger   r   )r   r3   r*   r   r   r   test_hist_single_row/   s
    z.TestDataFrameGroupByPlots.test_hist_single_rowc                 C   sd   t dddddgdddddgtdd}|djd	d
 t  |dd	 j  t  d S )Nr   r   r         ababar$   yzr:   r$   r9   )r   listr   r   scatterr   r   liner)   r   r   r   test_plot_submethod_works7   s
    (z3TestDataFrameGroupByPlots.test_plot_submethod_worksc                 C   s   t dddddgdddddgtdd}|djd	d
dd}t|d jdksTt|djjd
dd}t|d jdkstd S )Nr   r   r   r5   r6   r7   r8   r:   r<   r$   r9   )kindr$   r9   a)r$   r9   )r   r;   r   r   lencollectionsAssertionErrorr<   )r   r*   resr   r   r   test_plot_kwargs>   s
    (z*TestDataFrameGroupByPlots.test_plot_kwargszcolumn, expected_axes_num)Nr   )br   c                 C   s   d|f}|pdgdgg}t ddg ddg  dd}ttjd	d
|ddgd}|d}|jd|dD ]8}| j|||d t|d |D ]\}	}
| 	|	|
 qqhd S )Nr   r@   rF      12cname   r   r#   columnsT)legendcolumnZaxes_numZlayoutr   )
r   r   r   r   r'   r   r   _check_axes_shapezip_check_legend_labels)r   rQ   Zexpected_axes_numZexpected_layoutZexpected_labelsr#   r*   gZaxesaxZexpected_labelr   r   r   #test_groupby_hist_frame_with_legendJ   s    
  z=TestDataFrameGroupByPlots.test_groupby_hist_frame_with_legendrQ   NrF   c              	   C   sp   t ddg ddg  dd}ttjdd|dd	gd
}|d}tjtdd |j	d|dd W 5 Q R X d S )NrG   rH   rI   rJ   rK   rM   r   r@   rF   rN    Cannot use both legend and labelmatchTd)rP   rQ   label
r   r   r   r   r'   r   pytestZraises
ValueErrorr   )r   rQ   r#   r*   rV   r   r   r   *test_groupby_hist_frame_with_legend_raises[   s
    
zDTestDataFrameGroupByPlots.test_groupby_hist_frame_with_legend_raisesc                 C   s|   t ddg ddg  dd}ttjdd|dd	gd
}|d}|d jddD ]$}| j|ddd | |ddg qRd S )NrG   rH   rI   rJ   rK   rM   r   r@   rF   rN   T)rP   r   )r   r   rR   )	r   r   r   r   r'   r   r   rS   rU   )r   r#   r*   rV   rW   r   r   r   $test_groupby_hist_series_with_legende   s    
z>TestDataFrameGroupByPlots.test_groupby_hist_series_with_legendc              	   C   sn   t ddg ddg  dd}ttjdd|dd	gd
}|d}tjtdd |j	ddd W 5 Q R X d S )NrG   rH   rI   rJ   rK   rM   r   r@   rF   rN   rY   rZ   Tr\   )rP   r]   r^   )r   r#   r*   rV   r   r   r   +test_groupby_hist_series_with_legend_raiseso   s
    
zETestDataFrameGroupByPlots.test_groupby_hist_series_with_legend_raises)__name__
__module____qualname__r   r+   r4   r>   rE   r_   markZparametrizerX   ra   rb   rc   r   r   r   r   r      s   

	
r   )__doc__Znumpyr   r_   Zpandas.util._test_decoratorsutilZ_test_decoratorstdZpandasr   r   r   Zpandas._testingZ_testingr   Zpandas.tests.plotting.commonr   Zskip_if_no_mplr   r   r   r   r   <module>   s   