§
    tŠtjq  ã                   óx   — d Z ddlZddlZddlmZmZmZ ddlm	Z	m
Z
  ej        d¦  «          G d„ d¦  «        ZdS )zTest cases for GroupBy.ploté    N)Ú	DataFrameÚIndexÚSeries)Ú_check_axes_shapeÚ_check_legend_labelsÚ
matplotlibc                   óô   — e Zd Zd„ Zd„ Zd„ Zd„ Zd„ Zd„ Zd„ Z	d„ Z
d	„ Zd
„ Zd„ Zej                             dddg¦  «        d„ ¦   «         Zej                             dddg¦  «        d„ ¦   «         Zd„ Zd„ ZdS )ÚTestDataFrameGroupByPlotsc                 óH  — d}t          t          j                             d¦  «                             dd|¬¦  «        ¦  «        }t          j                             d¦  «                             ddg|¬¦  «        }|                     |¦  «                             ¦   «          d S )Né
   é   é¦   é   ©ÚsizeÚmaleÚfemale©r   ÚnpÚrandomÚdefault_rngÚnormalÚchoiceÚgroupbyÚplot)ÚselfÚnÚweightÚgenders       ú`/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/pandas/tests/plotting/test_groupby.pyÚ,test_series_groupby_plotting_nominally_workszFTestDataFrameGroupByPlots.test_series_groupby_plotting_nominally_works   sŠ   € ØˆÝ�œ	×-Ò-¨aÑ0Ô0×7Ò7¸¸RÀaÐ7ÑHÔHÑIÔIˆÝ”×&Ò& qÑ)Ô)×0Ò0°&¸(Ð1CÈ!Ð0ÑLÔLˆà�Š�vÑÔ×#Ò#Ñ%Ô%Ð%Ð%Ð%ó    c                 óH  — d}t          t          j                             d¦  «                             dd|¬¦  «        ¦  «        }t          j                             d¦  «                             ddg|¬¦  «        }|                     |¦  «                             ¦   «          d S )Nr   r   é<   r   r   r   )r   r   r   r   r   r   r   Úhist©r   r   Úheightr   s       r    Ú1test_series_groupby_plotting_nominally_works_histzKTestDataFrameGroupByPlots.test_series_groupby_plotting_nominally_works_hist   sŠ   € ØˆÝ�œ	×-Ò-¨aÑ0Ô0×7Ò7¸¸BÀQÐ7ÑGÔGÑHÔHˆÝ”×&Ò& qÑ)Ô)×0Ò0°&¸(Ð1CÈ!Ð0ÑLÔLˆØ�Š�vÑÔ×#Ò#Ñ%Ô%Ð%Ð%Ð%r"   c                 óL  — d}t          t          j                             d¦  «                             dd|¬¦  «        ¦  «        }t          j                             d¦  «                             ddg|¬¦  «        }|                     |¦  «                             d¬¦  «         d S )	Nr   r   r$   r   r   r   g      à?)Úalphar   r&   s       r    Ú2test_series_groupby_plotting_nominally_works_alphazLTestDataFrameGroupByPlots.test_series_groupby_plotting_nominally_works_alpha!   s�   € ØˆÝ�œ	×-Ò-¨aÑ0Ô0×7Ò7¸¸BÀQÐ7ÑGÔGÑHÔHˆÝ”×&Ò& qÑ)Ô)×0Ò0°&¸(Ð1CÈ!Ð0ÑLÔLˆà�Š�vÑÔ×#Ò#¨#Ð#Ñ.Ô.Ð.Ð.Ð.r"   c                 óð   — t          g d¢t          j                             d¦  «                             d¦  «        dœg d¢¬¦  «        }|                     d¦  «        d                              ¦   «          d S )	N©	é   r.   r.   r   r   r   é   r/   r/   r   é	   ©ÚdefÚval©	ç      ð?ç       @ç      @r5   r6   r7   r5   r6   r7   ©Úindexr2   r3   )r   r   r   r   Ústandard_normalr   r   ©r   Údfs     r    Ú$test_plotting_with_float_index_worksz>TestDataFrameGroupByPlots.test_plotting_with_float_index_works(   s�   € åà2Ð2Ð2Ý”y×,Ò,¨QÑ/Ô/×?Ò?ÀÑBÔBðð ð @Ð?Ð?ð
ñ 
ô 
ˆð 	�
Š
�5ÑÔ˜%Ô ×%Ò%Ñ'Ô'Ð'Ð'Ð'r"   c                 óô   — t          g d¢t          j                             d¦  «                             d¦  «        dœg d¢¬¦  «        }|                     d¦  «        d                              d	„ ¦  «         d S )
Nr-   r   r0   r1   r4   r8   r2   r3   c                 ó*   — |                       ¦   «         S )N)r   )Úxs    r    ú<lambda>zVTestDataFrameGroupByPlots.test_plotting_with_float_index_works_apply.<locals>.<lambda>=   s   € °·²±´€ r"   )r   r   r   r   r:   r   Úapplyr;   s     r    Ú*test_plotting_with_float_index_works_applyzDTestDataFrameGroupByPlots.test_plotting_with_float_index_works_apply4   s‡   € åà2Ð2Ð2Ý”y×,Ò,¨QÑ/Ô/×?Ò?ÀÑBÔBðð ð @Ð?Ð?ð
ñ 
ô 
ˆð 	�
Š
�5ÑÔ˜%Ô ×&Ò&Ð'9Ð'9Ñ:Ô:Ð:Ð:Ð:r"   c                 óª   — t          j        ddd¦  «        }t          ddgddgddgd	œ¦  «        }|d
                              |d         |¬¦  «         d S )NéP   éf   r.   ÚAAAÚBBBr   éU   éY   ©ÚNameÚByColÚMarkrN   rM   ©ÚbyÚbins©r   Úaranger   r%   ©r   rQ   r<   s      r    Útest_hist_single_rowz.TestDataFrameGroupByPlots.test_hist_single_row?   s\   € åŒy˜˜W aÑ(Ô(ˆÝ ¨ ¸!¸Q¸È"ÈbÈÐRÐRÑSÔSˆØ
ˆ6Œ
�Š˜2˜gœ;¨TˆÑ2Ô2Ð2Ð2Ð2r"   c                 ó¤   — t          j        ddd¦  «        }t          dgdgdgdœ¦  «        }|d                              |d         |¬	¦  «         d S )
NrE   rF   r.   rG   rI   rK   rN   rM   rO   rR   rT   s      r    Ú!test_hist_single_row_single_bycolz;TestDataFrameGroupByPlots.test_hist_single_row_single_bycolE   sV   € åŒy˜˜W aÑ(Ô(ˆÝ  °1°#À¸tÐDÐDÑEÔEˆØ
ˆ6Œ
�Š˜2˜gœ;¨TˆÑ2Ô2Ð2Ð2Ð2r"   c                 óª   — t          g d¢g d¢t          d¦  «        dœ¦  «        }|                     d¦  «        j                             dd¦  «         d S )N©r.   r   r/   é   é   ©r.   r   r/   r   r.   Úababa©r@   ÚyÚzr`   r@   r_   )r   Úlistr   r   Úscatterr;   s     r    Útest_plot_submethod_worksz3TestDataFrameGroupByPlots.test_plot_submethod_worksK   sP   € Ý˜_˜_˜_°?°?°?ÍÈgÉÌÐWÐWÑXÔXˆØ
�
Š
�3‰ŒÔ×$Ò$ S¨#Ñ.Ô.Ð.Ð.Ð.r"   c                 ó²   — t          g d¢g d¢t          d¦  «        dœ¦  «        }|                     d¦  «        d         j                             ¦   «          d S )NrY   r\   r]   r^   r`   r@   )r   ra   r   r   Úliner;   s     r    Útest_plot_submethod_works_linez8TestDataFrameGroupByPlots.test_plot_submethod_works_lineO   sQ   € Ý˜_˜_˜_°?°?°?ÍÈgÉÌÐWÐWÑXÔXˆØ
�
Š
�3‰Œ˜ÔÔ!×&Ò&Ñ(Ô(Ð(Ð(Ð(r"   c                 óä   — t          g d¢g d¢t          d¦  «        dœ¦  «        }|                     d¦  «                             ddd¬	¦  «        }t	          |d
         j        ¦  «        dk    sJ ‚d S )NrY   r\   r]   r^   r`   rb   r@   r_   )Úkindr@   r_   Úar.   )r   ra   r   r   ÚlenÚcollections©r   r<   Úress      r    Útest_plot_kwargsz*TestDataFrameGroupByPlots.test_plot_kwargsS   ss   € Ý˜_˜_˜_°?°?°?ÍÈgÉÌÐWÐWÑXÔXˆà�jŠj˜‰oŒo×"Ò"¨	°S¸CÐ"Ñ@Ô@ˆõ �3�s”8Ô'Ñ(Ô(¨AÒ-Ð-Ð-Ð-Ð-Ð-r"   c                 óì   — t          g d¢g d¢t          d¦  «        dœ¦  «        }|                     d¦  «        j                             dd¬¦  «        }t          |d	         j        ¦  «        d
k    sJ ‚d S )NrY   r\   r]   r^   r`   r@   r_   )r@   r_   ri   r.   )r   ra   r   r   rb   rj   rk   rl   s      r    Útest_plot_kwargs_scatterz2TestDataFrameGroupByPlots.test_plot_kwargs_scatter[   sr   € Ý˜_˜_˜_°?°?°?ÍÈgÉÌÐWÐWÑXÔXˆØ�jŠj˜‰oŒoÔ"×*Ò*¨S°CÐ*Ñ8Ô8ˆÝ�3�s”8Ô'Ñ(Ô(¨AÒ-Ð-Ð-Ð-Ð-Ð-r"   zcolumn, expected_axes_num)Nr   )Úbr.   c                 ó¼  — d|f}|pdgdgg}t          ddgz  ddgz  z   d¬¦  «        }t          t          j                             d	¦  «                             d
¦  «        |ddg¬¦  «        }|                     d¦  «        }|                     d|¬¦  «        D ]B}t          |||¬¦  «         t          |d         |d¬¦  «        D ]\  }	}
t          |	|
¦  «         ŒŒCd S )Nr.   ri   rq   é   Ú1Ú2Úc©Únamer   ©é   r   ©r9   ÚcolumnsT)ÚlegendÚcolumn©Úaxes_numÚlayoutr   )Ústrict)r   r   r   r   r   r:   r   r%   r   Úzipr   )r   r~   Úexpected_axes_numÚexpected_layoutÚexpected_labelsr9   r<   ÚgÚaxesÚaxÚexpected_labels              r    Ú#test_groupby_hist_frame_with_legendz=TestDataFrameGroupByPlots.test_groupby_hist_frame_with_legend`   s  € ð Ð/Ð0ˆØ Ð2 c U¨S¨E Nˆå�b˜C˜5‘j 2¨¨¡:Ñ-°CÐ8Ñ8Ô8ˆÝÝŒI×!Ò! !Ñ$Ô$×4Ò4°WÑ=Ô=ØØ˜#�Jð
ñ 
ô 
ˆð
 �JŠJ�s‰OŒOˆà—F’F $¨v�FÑ6Ô6ð 	9ð 	9ˆDÝ˜dÐ->ÀÐWÑWÔWÐWÝ&)¨$¨q¬'°?È4Ð&PÑ&PÔ&Pð 9ð 9Ñ"��NÝ$ R¨Ñ8Ô8Ð8Ð8ð9ð	9ð 	9r"   r~   Nrq   c                 óˆ  — t          ddgz  ddgz  z   d¬¦  «        }t          t          j                             d¦  «                             d¦  «        |dd	g¬
¦  «        }|                     d¦  «        }t          j        t          d¬¦  «        5  | 
                    d|d¬¦  «         d d d ¦  «         d S # 1 swxY w Y   d S )Nrs   rt   ru   rv   rw   r   ry   ri   rq   r{   ú Cannot use both legend and label©ÚmatchTÚd)r}   r~   Úlabel©r   r   r   r   r   r:   r   ÚpytestÚraisesÚ
ValueErrorr%   )r   r~   r9   r<   r‡   s        r    Ú*test_groupby_hist_frame_with_legend_raiseszDTestDataFrameGroupByPlots.test_groupby_hist_frame_with_legend_raisess   s	  € õ �b˜C˜5‘j 2¨¨¡:Ñ-°CÐ8Ñ8Ô8ˆÝÝŒI×!Ò! !Ñ$Ô$×4Ò4°WÑ=Ô=ØØ˜#�Jð
ñ 
ô 
ˆð
 �JŠJ�s‰OŒOˆåŒ]�:Ð-OÐPÑPÔPð 	:ð 	:Ø�FŠF˜$ v°SˆFÑ9Ô9Ð9ð	:ð 	:ð 	:ñ 	:ô 	:ð 	:ð 	:ð 	:ð 	:ð 	:ð 	:ð 	:øøøð 	:ð 	:ð 	:ð 	:ð 	:ð 	:s   ÂB7Â7B;Â>B;c                 óv  — t          ddgz  ddgz  z   d¬¦  «        }t          t          j                             d¦  «                             d¦  «        |dd	g¬
¦  «        }|                     d¦  «        }|d                              d¬¦  «        D ]&}t          |dd¬¦  «         t          |ddg¦  «         Œ'd S )Nrs   rt   ru   rv   rw   r   ry   ri   rq   r{   T)r}   r.   )r.   r.   r   )
r   r   r   r   r   r:   r   r%   r   r   )r   r9   r<   r‡   r‰   s        r    Ú$test_groupby_hist_series_with_legendz>TestDataFrameGroupByPlots.test_groupby_hist_series_with_legend�   sÌ   € å�b˜C˜5‘j 2¨¨¡:Ñ-°CÐ8Ñ8Ô8ˆÝÝŒI×!Ò! !Ñ$Ô$×4Ò4°WÑ=Ô=ØØ˜#�Jð
ñ 
ô 
ˆð
 �JŠJ�s‰OŒOˆà�C”&—+’+ T�+Ñ*Ô*ð 	1ð 	1ˆBÝ˜b¨1°VÐ<Ñ<Ô<Ð<Ý   c¨3 ZÑ0Ô0Ð0Ð0ð	1ð 	1r"   c                 ó†  — t          ddgz  ddgz  z   d¬¦  «        }t          t          j                             d¦  «                             d¦  «        |dd	g¬
¦  «        }|                     d¦  «        }t          j        t          d¬¦  «        5  | 
                    dd¬¦  «         d d d ¦  «         d S # 1 swxY w Y   d S )Nrs   rt   ru   rv   rw   r   ry   ri   rq   r{   r�   rŽ   Tr�   )r}   r‘   r’   )r   r9   r<   r‡   s       r    Ú+test_groupby_hist_series_with_legend_raiseszETestDataFrameGroupByPlots.test_groupby_hist_series_with_legend_raises�   s  € å�b˜C˜5‘j 2¨¨¡:Ñ-°CÐ8Ñ8Ô8ˆÝÝŒI×!Ò! !Ñ$Ô$×4Ò4°WÑ=Ô=ØØ˜#�Jð
ñ 
ô 
ˆð
 �JŠJ�s‰OŒOˆåŒ]�:Ð-OÐPÑPÔPð 	+ð 	+Ø�FŠF˜$ cˆFÑ*Ô*Ð*ð	+ð 	+ð 	+ñ 	+ô 	+ð 	+ð 	+ð 	+ð 	+ð 	+ð 	+ð 	+øøøð 	+ð 	+ð 	+ð 	+ð 	+ð 	+s   ÂB6Â6B:Â=B:)Ú__name__Ú
__module__Ú__qualname__r!   r(   r+   r=   rC   rU   rW   rc   rf   rn   rp   r“   ÚmarkÚparametrizer‹   r–   r˜   rš   © r"   r    r
   r
      s>  € € € € € ð&ð &ð &ð&ð &ð &ð/ð /ð /ð
(ð 
(ð 
(ð	;ð 	;ð 	;ð3ð 3ð 3ð3ð 3ð 3ð/ð /ð /ð)ð )ð )ð.ð .ð .ð.ð .ð .ð
 „[×ÒÐ8¸9ÀhÐ:OÑPÔPð9ð 9ñ QÔPð9ð$ „[×Ò˜X¨¨c {Ñ3Ô3ð:ð :ñ 4Ô3ð:ð1ð 1ð 1ð+ð +ð +ð +ð +r"   r
   )Ú__doc__Únumpyr   r“   Úpandasr   r   r   Úpandas.tests.plotting.commonr   r   Úimportorskipr
   r    r"   r    ú<module>r¦      sË   ðØ !Ð !à Ð Ð Ð Ø €€€ðð ð ð ð ð ð ð ð ð ð
ð ð ð ð ð ð ð ð
 €Ô �LÑ !Ô !Ð !ðG+ð G+ð G+ð G+ð G+ñ G+ô G+ð G+ð G+ð G+r"   