Ë
    ¤ehg  ã                   ót   — 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y)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                  j!                  dddg«      d„ «       Zej                  j!                  dddg«      d„ «       Zd„ Zd„ Zy)ÚTestDataFrameGroupByPlotsc                 ó  — d}t        t        j                  j                  d«      j	                  dd|¬«      «      }t        j                  j                  d«      j                  ddg|¬«      }|j                  |«      j                  «        y )Né
   é   é¦   é   ©ÚsizeÚmaleÚfemale©r   ÚnpÚrandomÚdefault_rngÚnormalÚchoiceÚgroupbyÚplot)ÚselfÚnÚweightÚgenders       ú`/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/pandas/tests/plotting/test_groupby.pyÚ,test_series_groupby_plotting_nominally_workszFTestDataFrameGroupByPlots.test_series_groupby_plotting_nominally_works   sp   € ØˆÜœŸ	™	×-Ñ-¨aÓ0×7Ñ7¸¸RÀaÐ7ÓHÓIˆÜ—‘×&Ñ& qÓ)×0Ñ0°&¸(Ð1CÈ!Ð0ÓLˆà�‰�vÓ×#Ñ#Õ%ó    c                 ó  — d}t        t        j                  j                  d«      j	                  dd|¬«      «      }t        j                  j                  d«      j                  ddg|¬«      }|j                  |«      j                  «        y )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   sp   € ØˆÜœŸ	™	×-Ñ-¨aÓ0×7Ñ7¸¸BÀQÐ7ÓGÓHˆÜ—‘×&Ñ& qÓ)×0Ñ0°&¸(Ð1CÈ!Ð0ÓLˆØ�‰�vÓ×#Ñ#Õ%r"   c                 ó"  — d}t        t        j                  j                  d«      j	                  dd|¬«      «      }t        j                  j                  d«      j                  ddg|¬«      }|j                  |«      j                  d¬«       y )	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"   su   € ØˆÜœŸ	™	×-Ñ-¨aÓ0×7Ñ7¸¸BÀQÐ7ÓGÓHˆÜ—‘×&Ñ& qÓ)×0Ñ0°&¸(Ð1CÈ!Ð0ÓLˆà�‰�vÓ×#Ñ#¨#Ð#Õ.r"   c                 óÈ   — t        g d¢t        j                  j                  d«      j	                  d«      dœg d¢¬«      }|j                  d«      d   j                  «        y )	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)   sS   € äâ2Ü—y‘y×,Ñ,¨QÓ/×?Ñ?ÀÓBñò @ô
ˆð 	�
‰
�5Ó˜%Ñ ×%Ñ%Õ'r"   c                 óÌ   — t        g d¢t        j                  j                  d«      j	                  d«      dœg d¢¬«      }|j                  d«      d   j                  d	„ «       y )
Nr-   r   r0   r1   r4   r8   r2   r3   c                 ó"   — | j                  «       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_apply5   sV   € äâ2Ü—y‘y×,Ñ,¨QÓ/×?Ñ?ÀÓBñò @ô
ˆð 	�
‰
�5Ó˜%Ñ ×&Ñ&Ñ'9Õ:r"   c                 óŽ   — t        j                  ddd«      }t        ddgddgddgd	œ«      }|d
   j                  |d   |¬«       y )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@   sI   € ä�y‰y˜˜W aÓ(ˆÜ ¨ ¸!¸Q¸È"ÈbÈÑRÓSˆØ
ˆ6‰
�‰˜2˜g™;¨TˆÕ2r"   c                 óˆ   — t        j                  ddd«      }t        dgdgdgdœ«      }|d   j                  |d   |¬	«       y )
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_bycolF   sC   € ä�y‰y˜˜W aÓ(ˆÜ  °1°#À¸tÑDÓEˆØ
ˆ6‰
�‰˜2˜g™;¨TˆÕ2r"   c                 ó’   — t        g d¢g d¢t        d«      dœ«      }|j                  d«      j                  j	                  dd«       y )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_worksL   s5   € Üš_²?ÌÈgËÑWÓXˆØ
�
‰
�3‹×Ñ×$Ñ$ S¨#Õ.r"   c                 ó”   — t        g d¢g d¢t        d«      dœ«      }|j                  d«      d   j                  j	                  «        y )NrY   r\   r]   r^   r`   r@   )r   ra   r   r   Úliner;   s     r    Útest_plot_submethod_works_linez8TestDataFrameGroupByPlots.test_plot_submethod_works_lineP   s6   € Üš_²?ÌÈgËÑWÓXˆØ
�
‰
�3‹˜Ñ×!Ñ!×&Ñ&Õ(r"   c                 ó¼   — t        g d¢g d¢t        d«      dœ«      }|j                  d«      j                  ddd¬	«      }t	        |d
   j
                  «      dk(  sJ ‚y )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_kwargsT   sU   € Üš_²?ÌÈgËÑWÓXˆà�j‰j˜‹o×"Ñ"¨	°S¸CÐ"Ó@ˆô �3�s‘8×'Ñ'Ó(¨AÒ-Ð-Ñ-r"   c                 óÎ   — t        g d¢g d¢t        d«      dœ«      }|j                  d«      j                  j	                  dd¬«      }t        |d	   j                  «      d
k(  sJ ‚y )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\   sW   € Üš_²?ÌÈgËÑWÓXˆØ�j‰j˜‹o×"Ñ"×*Ñ*¨S°CÐ*Ó8ˆÜ�3�s‘8×'Ñ'Ó(¨AÒ-Ð-Ñ-r"   zcolumn, expected_axes_num)Nr   )Úbr.   c                 ó|  — d|f}|xs dgdgg}t        ddgz  ddgz  z   d¬«      }t        t        j                  j	                  d	«      j                  d
«      |ddg¬«      }|j                  d«      }|j                  d|¬«      D ]3  }t        |||¬«       t        |d   |«      D ]  \  }	}
t        |	|
«       Œ Œ5 y )Nr.   ri   rq   é   Ú1Ú2Úc©Únamer   ©é   r   ©r9   ÚcolumnsT)ÚlegendÚcolumn©Úaxes_numÚlayoutr   )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_legenda   sÐ   € ð Ð/Ð0ˆØ Ò2 c U¨S¨E Nˆä�b˜C˜5‘j 2¨¨¡:Ñ-°CÔ8ˆÜÜ�I‰I×!Ñ! !Ó$×4Ñ4°WÓ=ØØ˜#�Jô
ˆð
 �J‰J�s‹Oˆà—F‘F $¨v�FÓ6ò 	9ˆDÜ˜dÐ->ÀÕWÜ&)¨$¨q©'°?Ó&Cò 9Ñ"��NÜ$ R¨Õ8ñ9ñ	9r"   r~   Nrq   c                 óV  — t        ddgz  ddgz  z   d¬«      }t        t        j                  j	                  d«      j                  d«      |dd	g¬
«      }|j                  d«      }t        j                  t        d¬«      5  |j                  d|d¬«       d d d «       y # 1 sw Y   y xY w)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_raisest   s—   € ô �b˜C˜5‘j 2¨¨¡:Ñ-°CÔ8ˆÜÜ�I‰I×!Ñ! !Ó$×4Ñ4°WÓ=ØØ˜#�Jô
ˆð
 �J‰J�s‹Oˆä�]‰]œ:Ð-OÔPñ 	:Ø�F‰F˜$ v°SˆFÔ9÷	:÷ 	:ñ 	:ús   ÂBÂB(c                 ó:  — t        ddgz  ddgz  z   d¬«      }t        t        j                  j	                  d«      j                  d«      |dd	g¬
«      }|j                  d«      }|d   j                  d¬«      D ]  }t        |dd¬«       t        |ddg«       Œ  y )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ˆÜÜ�I‰I×!Ñ! !Ó$×4Ñ4°WÓ=ØØ˜#�Jô
ˆð
 �J‰J�s‹Oˆà�C‘&—+‘+ T�+Ó*ò 	1ˆBÜ˜b¨1°VÕ<Ü   c¨3 ZÕ0ñ	1r"   c                 óT  — t        ddgz  ddgz  z   d¬«      }t        t        j                  j	                  d«      j                  d«      |dd	g¬
«      }|j                  d«      }t        j                  t        d¬«      5  |j                  dd¬«       d d d «       y # 1 sw Y   y xY w)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ˆÜÜ�I‰I×!Ñ! !Ó$×4Ñ4°WÓ=ØØ˜#�Jô
ˆð
 �J‰J�s‹Oˆä�]‰]œ:Ð-OÔPñ 	+Ø�F‰F˜$ cˆFÔ*÷	+÷ 	+ñ 	+ús   Â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ò/ò)ò.ò.ð
 ‡[�[×ÑÐ8¸9ÀhÐ:OÓPñ9ó Qð9ð$ ‡[�[×Ñ˜X¨¨c {Ó3ñ:ó 4ð:ò1ó+r"   r
   )Ú__doc__Únumpyr   r’   Úpandasr   r   r   Úpandas.tests.plotting.commonr   r   Úimportorskipr
   rŸ   r"   r    ú<module>r¥      s=   ðÙ #ó Û ÷ñ ÷
ð
 €× Ñ �LÔ !÷G+ò G+r"   