Ë
    u™ãjÒ  ã                   óh   — d Z ddlZddlZddlZddlmZmZ  ej                  d«        G d„ d«      Z
y)zTest cases for GroupBy.ploté    N)Ú_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d„ Zd„ Zd„ Zy)ÚTestDataFrameGroupByPlotsc                 ó2  — d}t        j                  t        j                  j	                  d«      j                  dd|¬«      «      }t        j                  j	                  d«      j                  ddg|¬«      }|j                  |«      j                  «        y )Né
   é   é¦   é   ©ÚsizeÚmaleÚfemale©	ÚpdÚSeriesÚnpÚrandomÚdefault_rngÚnormalÚchoiceÚgroupbyÚplot)ÚselfÚnÚweightÚgenders       úm/var/www/html/python-recuperauto-other/env/lib/python3.12/site-packages/pandas/tests/plotting/test_groupby.pyÚ,test_series_groupby_plotting_nominally_workszFTestDataFrameGroupByPlots.test_series_groupby_plotting_nominally_works   st   € ØˆÜ—‘œ2Ÿ9™9×0Ñ0°Ó3×:Ñ:¸3ÀÈÐ:ÓKÓLˆÜ—‘×&Ñ& qÓ)×0Ñ0°&¸(Ð1CÈ!Ð0ÓLˆà�‰�vÓ×#Ñ#Õ%ó    c                 ó2  — d}t        j                  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   r   Úhist©r   r   Úheightr   s       r   Ú1test_series_groupby_plotting_nominally_works_histzKTestDataFrameGroupByPlots.test_series_groupby_plotting_nominally_works_hist   st   € ØˆÜ—‘œ2Ÿ9™9×0Ñ0°Ó3×:Ñ:¸2¸rÈÐ:ÓJÓKˆÜ—‘×&Ñ& qÓ)×0Ñ0°&¸(Ð1CÈ!Ð0ÓLˆØ�‰�vÓ×#Ñ#Õ%r!   c                 ó6  — d}t        j                  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   sy   € ØˆÜ—‘œ2Ÿ9™9×0Ñ0°Ó3×:Ñ:¸2¸rÈÐ:ÓJÓKˆÜ—‘×&Ñ& qÓ)×0Ñ0°&¸(Ð1CÈ!Ð0ÓLˆà�‰�vÓ×#Ñ#¨#Ð#Õ.r!   c                 óÜ   — t        j                  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©	ç      ð?ç       @ç      @r4   r5   r6   r4   r5   r6   ©Úindexr1   r2   )r   Ú	DataFramer   r   r   Ústandard_normalr   r   ©r   Údfs     r   Ú$test_plotting_with_float_index_worksz>TestDataFrameGroupByPlots.test_plotting_with_float_index_works$   sW   € ä�\‰\â2Ü—y‘y×,Ñ,¨QÓ/×?Ñ?ÀÓBñò @ô
ˆð 	�
‰
�5Ó˜%Ñ ×%Ñ%Õ'r!   c                 óà   — t        j                  g d¢t        j                  j	                  d«      j                  d«      dœg d¢¬«      }|j                  d«      d   j                  d	„ «       y )
Nr,   r
   r/   r0   r3   r7   r1   r2   c                 ó"   — | j                  «       S )N)r   )Úxs    r   ú<lambda>zVTestDataFrameGroupByPlots.test_plotting_with_float_index_works_apply.<locals>.<lambda>9   s   € °·±³€ r!   )r   r9   r   r   r   r:   r   Úapplyr;   s     r   Ú*test_plotting_with_float_index_works_applyzDTestDataFrameGroupByPlots.test_plotting_with_float_index_works_apply0   sZ   € ä�\‰\â2Ü—y‘y×,Ñ,¨QÓ/×?Ñ?ÀÓBñò @ô
ˆð 	�
‰
�5Ó˜%Ñ ×&Ñ&Ñ'9Õ:r!   c                 ó¢   — t        j                  ddd«      }t        j                  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   r9   r$   ©r   rQ   r<   s      r   Útest_hist_single_rowz.TestDataFrameGroupByPlots.test_hist_single_row;   sN   € ä�y‰y˜˜W aÓ(ˆÜ�\‰\ E¨5 >¸QÀ¸FÈRÐQSÈHÑUÓVˆØ
ˆ6‰
�‰˜2˜g™;¨TˆÕ2r!   c                 óœ   — t        j                  ddd«      }t        j                  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_bycolA   sG   € ä�y‰y˜˜W aÓ(ˆÜ�\‰\ E 7°a°SÀ2À$ÑGÓHˆØ
ˆ6‰
�‰˜2˜g™;¨TˆÕ2r!   c                 ó¦   — t        j                  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   r9   Úlistr   r   Úscatterr;   s     r   Útest_plot_submethod_worksz3TestDataFrameGroupByPlots.test_plot_submethod_worksG   s>   € Ü�\‰\Ú!ª¼dÀ7»mÑLó
ˆð 	�
‰
�3‹×Ñ×$Ñ$ S¨#Õ.r!   c                 ó¨   — t        j                  g d¢g d¢t        d«      dœ«      }|j                  d«      d   j                  j                  «        y )NrY   r\   r]   r^   r`   r@   )r   r9   ra   r   r   Úliner;   s     r   Útest_plot_submethod_works_linez8TestDataFrameGroupByPlots.test_plot_submethod_works_lineM   s?   € Ü�\‰\Ú!ª¼dÀ7»mÑLó
ˆð 	�
‰
�3‹˜Ñ×!Ñ!×&Ñ&Õ(r!   c                 óÐ   — t        j                  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   r9   ra   r   r   ÚlenÚcollections©r   r<   Úress      r   Útest_plot_kwargsz*TestDataFrameGroupByPlots.test_plot_kwargsS   s^   € Ü�\‰\Ú!ª¼dÀ7»mÑLó
ˆð �j‰j˜‹o×"Ñ"¨	°S¸CÐ"Ó@ˆô �3�s‘8×'Ñ'Ó(¨AÒ-Ð-Ñ-r!   c                 óâ   — t        j                  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   r9   ra   r   r   rb   rj   rk   rl   s      r   Útest_plot_kwargs_scatterz2TestDataFrameGroupByPlots.test_plot_kwargs_scatter]   s`   € Ü�\‰\Ú!ª¼dÀ7»mÑLó
ˆð �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        j                  ddgz  ddgz  z   d¬«      }t        j                  t        j                  j                  d	«      j                  d
«      |ddg¬«      }|j                  d«      }|j                  d|¬«      D ]5  }t        |||¬«       t        |d   |d¬«      D ]  \  }	}
t        |	|
«       Œ Œ7 y )Nr-   ri   rq   é   Ú1Ú2Úc©Únamer
   ©é   r
   ©r8   ÚcolumnsT)ÚlegendÚcolumn©Úaxes_numÚlayoutr   )Ústrict)r   ÚIndexr9   r   r   r   r:   r   r$   r   Úzipr   )r   r~   Úexpected_axes_numÚexpected_layoutÚexpected_labelsr8   r<   ÚgÚaxesÚaxÚexpected_labels              r   Ú#test_groupby_hist_frame_with_legendz=TestDataFrameGroupByPlots.test_groupby_hist_frame_with_legendd   sÚ   € ð Ð/Ð0ˆØ Ò2 c U¨S¨E Nˆä—‘˜˜s˜e™ b¨C¨5¡jÑ0°sÔ;ˆÜ�\‰\Ü�I‰I×!Ñ! !Ó$×4Ñ4°WÓ=ØØ˜#�Jô
ˆð
 �J‰J�s‹Oˆà—F‘F $¨v�FÓ6ò 	9ˆDÜ˜dÐ->ÀÕWÜ&)¨$¨q©'°?È4Ô&Pò 9Ñ"��NÜ$ R¨Õ8ñ9ñ	9r!   r~   Nrq   c                 ó~  — t        j                  ddgz  ddgz  z   d¬«      }t        j                  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ƒ   r9   r   r   r   r:   r   ÚpytestÚraisesÚ
ValueErrorr$   )r   r~   r8   r<   rˆ   s        r   Ú*test_groupby_hist_frame_with_legend_raiseszDTestDataFrameGroupByPlots.test_groupby_hist_frame_with_legend_raisesw   sŸ   € ô —‘˜˜s˜e™ b¨C¨5¡jÑ0°sÔ;ˆÜ�\‰\Ü�I‰I×!Ñ! !Ó$×4Ñ4°WÓ=ØØ˜#�Jô
ˆð
 �J‰J�s‹Oˆä�]‰]œ:Ð-OÔPñ 	:Ø�F‰F˜$ v°SˆFÔ9÷	:÷ 	:ñ 	:ús   ÂB3Â3B<c                 ób  — t        j                  ddgz  ddgz  z   d¬«      }t        j                  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ƒ   r9   r   r   r   r:   r   r$   r   r   )r   r8   r<   rˆ   rŠ   s        r   Ú$test_groupby_hist_series_with_legendz>TestDataFrameGroupByPlots.test_groupby_hist_series_with_legend…   sŸ   € ä—‘˜˜s˜e™ b¨C¨5¡jÑ0°sÔ;ˆÜ�\‰\Ü�I‰I×!Ñ! !Ó$×4Ñ4°WÓ=ØØ˜#�Jô
ˆð
 �J‰J�s‹Oˆà�C‘&—+‘+ T�+Ó*ò 	1ˆBÜ˜b¨1°VÕ<Ü   c¨3 ZÕ0ñ	1r!   c                 óR  — t        j                  ddgz  ddgz  z   d¬«      }t        j                  t        j                  j                  d«      j                  d«      |dd	g¬
«      }|j                  d«      d   j                  d¬«      }t        |j                  d   ddg«       y )Nrs   rt   ru   rv   rw   r
   ry   ri   rq   r{   Tr™   r   )r   rƒ   r9   r   r   r   r:   r   r   r   Úiloc)r   r8   r<   r‰   s       r   Ú$test_groupby_plot_series_with_legendz>TestDataFrameGroupByPlots.test_groupby_plot_series_with_legend“   s’   € ô —‘˜˜s˜e™ b¨C¨5¡jÑ0°sÔ;ˆÜ�\‰\Ü�I‰I×!Ñ! !Ó$×4Ñ4°WÓ=ØØ˜#�Jô
ˆð �z‰z˜#‹˜sÑ#×(Ñ(°Ð(Ó5ˆÜ˜TŸY™Y q™\¨C°¨:Õ6r!   c                 ó|  — t        j                  ddgz  ddgz  z   d¬«      }t        j                  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   r8   r<   rˆ   s       r   Ú+test_groupby_hist_series_with_legend_raiseszETestDataFrameGroupByPlots.test_groupby_hist_series_with_legend_raises    s›   € ä—‘˜˜s˜e™ b¨C¨5¡jÑ0°sÔ;ˆÜ�\‰\Ü�I‰I×!Ñ! !Ó$×4Ñ4°WÓ=ØØ˜#�Jô
ˆð
 �J‰J�s‹Oˆä�]‰]œ:Ð-OÔPñ 	+Ø�F‰F˜$ cˆFÔ*÷	+÷ 	+ñ 	+ús   ÂB2Â2B;c                 óê   — t        j                  g d¢g d¢t        d«      dœ«      }|j                  d«      j                  j                  ddd¬	«      }t        |d
   d
g«       t        |d   dg«       y )NrY   r\   r]   r^   r`   r@   r_   T©r@   r_   r}   ri   rq   )r   r9   ra   r   r   rb   r   rl   s      r   Ú&test_plot_kwargs_scatter_legend_labelsz@TestDataFrameGroupByPlots.test_plot_kwargs_scatter_legend_labels­   sr   € ä�\‰\â$Ú$Ü˜'“]ñó
ˆð �j‰j˜‹o×"Ñ"×*Ñ*ØØØð +ó 
ˆô 	˜S ™X¨ uÔ-Ü˜S ™X¨ uÕ-r!   c                 óþ   — t        j                  g d¢g d¢t        d«      dœ«      }|j                  d«      j                  j                  ddd¬	«      }|d
   j                  «       �J ‚|d   j                  «       �J ‚y )NrY   r\   r]   r^   r`   r@   r_   Fr¡   ri   rq   )r   r9   ra   r   r   rb   Ú
get_legendrl   s      r   Ú"test_plot_kwargs_scatter_no_legendz<TestDataFrameGroupByPlots.test_plot_kwargs_scatter_no_legendÀ   s„   € ä�\‰\â$Ú$Ü˜'“]ñó
ˆð �j‰j˜‹o×"Ñ"×*Ñ*ØØØð +ó 
ˆð �3‰x×"Ñ"Ó$Ð,Ð,Ð,Ø�3‰x×"Ñ"Ó$Ð,Ð,Ñ,r!   )Ú__name__Ú
__module__Ú__qualname__r    r'   r*   r=   rC   rU   rW   rc   rf   rn   rp   r”   ÚmarkÚparametrizerŒ   r—   rš   r�   rŸ   r¢   r¥   © r!   r   r   r      s§   „ ò&ò&ò/ò
(ò	;ò3ò3ò/ò)ò.ò.ð ‡[�[×ÑÐ8¸9ÀhÐ:OÓPñ9ó Qð9ð$ ‡[�[×Ñ˜X¨¨c {Ó3ñ:ó 4ð:ò1ò7ò+ò.ó&-r!   r   )Ú__doc__Únumpyr   r”   Úpandasr   Úpandas.tests.plotting.commonr   r   Úimportorskipr   r«   r!   r   ú<module>r±      s4   ðÙ !ã Û ã ÷ð
 €× Ñ �LÔ !÷B-ò B-r!   