Ë
    ž…jÂ;  ã                   ó°  — d dl Z d dlZd dlmZ d dlmZ d dlZd dlmZm	Z	m
Z
mZmZ d dlmZ d dlmZ  G d„ d«      Zej&                  j)                  dg d	¢«      ej&                  j)                  d
ddej*                  g«      ej&                  j)                  ddgddgg«      d„ «       «       «       Zej&                  j)                  deeeeg«      d„ «       Zy)é    N)Úlib)ÚPandas4Warning)Ú	DataFrameÚIndexÚSeriesÚ	TimestampÚ
date_range)Úget_groupby_method_argsc                   óæ  — e Zd Zej                  d„ «       Zej                  j                  dddg«      d„ «       Zej                  j                  dddg«      d„ «       Z	ej                  j                  dd	d
g«      d„ «       Z
ej                  j                  dddg«      d„ «       Zej                  j                  dddg«      d„ «       Zej                  j                  dddg«      d„ «       Zd„ Zy)ÚTestNumericOnlyc                 ó"  — t        g d¢g d¢g d¢t        g d¢d¬«      t        g d¢t        ¬«      t        t        d«      «      j	                  d«      g d	¢t        d
d¬«      t        d
dd¬«      t        j                  ddd¬«      dœ
g d¢¬«      }|S )N)é   r   é   )r   r   é   )g      @g      @ç      @)ÚaÚbÚcÚstr©ÚdtypeÚabcÚcategory)é   é   é	   Ú20130101r   )Úperiodsú
US/Eastern)r   Útzz1 sÚs)r   Úfreq)
ÚgroupÚintÚfloatÚstringÚobjectÚcategory_stringÚcategory_intÚdatetimeÚ
datetimetzÚ	timedelta)Úcolumns)r   r   r'   ÚlistÚastyper	   ÚpdÚtimedelta_range)ÚselfÚdfs     úg/root/aria/tools/markitdown-venv/lib/python3.12/site-packages/pandas/tests/groupby/test_numeric_only.pyr3   zTestNumericOnly.df   s~   € ô â"Ú Ú(Ü ¢¸Ô>Ü ¢¼Ô?Ü#)¬$¨u«+Ó#6×#=Ñ#=¸jÓ#IÚ )Ü& z¸1Ô=Ü(¨¸QÀ<ÔPÜ×/Ñ/°¸qÀsÔKñòô
ˆð4 ˆ	ó    ÚmethodÚmeanÚmedianc                 óì  — t        g d¢«      }|j                  d«      }t        ddgddgt        j                  d«      t        j                  d«      gd	d
gt        d«      t        d«      gt        dd¬«      t        dd¬«      gdœt        ddgd¬«      g d¢¬«      } t        ||«      d¬«      }t        j                  |j                  |«      |«       |j                  }| j                  ||||«       y )N©r$   r%   r)   r#   g      @r   g      @r   z1.5sÚ3sg      ø?r   z2013-01-01 12:00:00z2013-01-03 00:00:00r   )r    )r)   r%   r,   r$   r*   r+   r   r   )Úname)Úindexr-   T©Únumeric_only)r   Úgroupbyr   r0   Ú	Timedeltar   ÚgetattrÚtmÚassert_frame_equalÚreindex_liker-   Ú_check)r2   r3   r6   Úexpected_columns_numericÚgbÚexpectedÚresultÚexpected_columnss           r4   Útest_averageszTestNumericOnly.test_averages7   sõ   € ô $)Ò)IÓ#JÐ à�Z‰Z˜Ó ˆÜà!$ a Ø˜s˜Ü Ÿl™l¨6Ó2´B·L±LÀÓ4FÐGØ˜Q�xäÐ3Ó4ÜÐ3Ó4ðô
 Ð3¸ÔEÜÐ3¸ÔEðñô ˜˜A˜ WÔ-òô!
ˆð. %”˜˜VÓ$°$Ô7ˆÜ
×Ñ˜f×1Ñ1°(Ó;¸XÔFà#×+Ñ+Ðà�‰�B˜Ð 0Ð2JÕKr5   ÚminÚmaxc                 óJ   — t        g d¢«      }|}| j                  ||||«       y )N)r$   r%   r&   r)   r*   r+   r,   ©r   rF   ©r2   r3   r6   rK   rG   s        r4   Útest_extremazTestNumericOnly.test_extrema[   s0   € ô
 !òó

Ðð $4Ð à�‰�B˜Ð 0Ð2JÕKr5   ÚfirstÚlastc                 óJ   — t        g d¢«      }|}| j                  ||||«       y )N)	r$   r%   r&   r'   r(   r)   r*   r+   r,   rP   rQ   s        r4   Útest_first_lastzTestNumericOnly.test_first_lasto   s.   € ä ò
ó
Ðð $4Ð à�‰�B˜Ð 0Ð2JÕKr5   ÚsumÚcumsumc                 ó„   — t        g d¢«      }t        g d¢«      }|dk(  rt        g d¢«      }| j                  ||||«       y )Nr:   )r$   r%   r&   r)   r,   rX   )r$   r%   r)   r,   rP   )r2   r3   r6   rG   rK   s        r4   Útest_sum_cumsumzTestNumericOnly.test_sum_cumsum‚   sF   € ä#(Ò)IÓ#JÐ Ü ÚCó
Ðð �XÒä$Ò%RÓSÐà�‰�B˜Ð 0Ð2JÕKr5   ÚprodÚcumprodc                 óJ   — t        g d¢«      }|}| j                  ||||«       y )Nr:   rP   rQ   s        r4   Útest_prod_cumprodz!TestNumericOnly.test_prod_cumprodŽ   s(   € ä Ò!AÓBÐØ#3Ð à�‰�B˜Ð 0Ð2JÕKr5   ÚcumminÚcummaxc                 óJ   — t        g d¢«      }|}| j                  ||||«       y )N)r$   r%   r)   r*   r+   r,   rP   rQ   s        r4   Útest_cummin_cummaxz"TestNumericOnly.test_cummin_cummax•   s.   € ô !ÚSó
Ðð
 $4Ð à�‰�B˜Ð 0Ð2JÕKr5   c                 óX  — |j                  d«      }|j                  d«      rt        t        fnt        }|dv rgdj	                  dd|› d�t        j                  d|› d	�«      d
d|› d�g«      }t        j                  ||¬«      5   t        ||«      «        d d d «       n®|dv rydj	                  dt        j                  d|› d	�«      t        j                  d|› d�«      d|› d�g«      }t        j                  ||¬«      5   t        ||«      «        d d d «       n1 t        ||«      «       }t        j                  |j                  |«       |dvr‚dj	                  ddd
d|› d�t        j                  d|› d	�«      t        j                  d|› d�«      d|› d�g«      }t        j                  ||¬«      5   t        ||«      d¬«       d d d «       y  t        ||«      d¬«      }t        j                  |j                  |«       y # 1 sw Y   ŒÃxY w# 1 sw Y   ŒÏxY w# 1 sw Y   y xY w)Nr#   Úcum)rM   rN   r_   r`   rX   r\   ú|zCategorical is not orderedzCannot perform z with non-ordered Categoricalúagg function failed [how->ú,dtype->object]ú*function is not implemented for this dtypez(dtype 'str' does not support operation 'ú'©Úmatch)rW   r7   r8   r[   z-category type does not support sum operationsz,dtype->string]©rS   rT   zcategory type does not supportFr>   )r@   Ú
startswithÚNotImplementedErrorÚ	TypeErrorÚjoinÚreÚescapeÚpytestÚraisesrB   rC   Úassert_index_equalr-   )	r2   r3   r6   rK   rG   rH   Ú	exceptionÚmsgrJ   s	            r4   rF   zTestNumericOnly._check¡   sC  € Ø�Z‰Z˜Ó ˆð
 17×0AÑ0AÀ%Ô0HÔ ¤)Ñ,Ìið 	ð ÐLÑLà—(‘(à0Ø% f XÐ-JÐKÜ—I‘IÐ :¸6¸(À/ÐRÓSà@Ø>¸v¸hÀaÐHðó	ˆCô —‘˜y°Ô4ñ &Ø#”˜˜FÓ#Ô%÷&ð &àÐ8Ñ8Ø—(‘(àCÜ—I‘IÐ :¸6¸(À/ÐRÓSÜ—I‘IÐ :¸6¸(À/ÐRÓSØ>¸v¸hÀaÐHð	óˆCô —‘˜y°Ô4ñ &Ø#”˜˜FÓ#Ô%÷&ð &ð )”W˜R Ó(Ó*ˆFÜ×!Ñ! &§.¡.Ð2JÔKàÐ*Ñ*Ø—(‘(à0Ø4Ø@Ø% f XÐ-JÐKÜ—I‘IÐ :¸6¸(À/ÐRÓSÜ—I‘IÐ :¸6¸(À/ÐRÓSØ>¸v¸hÀaÐHðó
ˆCô —‘˜y°Ô4ñ 8Ø#”˜˜FÓ#°Õ7÷8ð 8ð )”W˜R Ó(°eÔ<ˆFÜ×!Ñ! &§.¡.Ð2BÕC÷C&ð &ú÷&ð &ú÷$8ð 8ús$   ÂHÄHÆ7H ÈHÈHÈ H)N)Ú__name__Ú
__module__Ú__qualname__rs   Úfixturer3   ÚmarkÚparametrizerL   rR   rV   rZ   r^   rb   rF   © r5   r4   r   r      s(  „ ð ‡^�^ñó ðð> ‡[�[×Ñ˜X¨°Ð'9Ó:ñ!Ló ;ð!LðF ‡[�[×Ñ˜X¨¨u ~Ó6ñLó 7ðLð& ‡[�[×Ñ˜X¨°Ð'8Ó9ñLó :ðLð$ ‡[�[×Ñ˜X¨¨xÐ'8Ó9ñ	Ló :ð	Lð ‡[�[×Ñ˜X¨°	Ð':Ó;ñLó <ðLð ‡[�[×Ñ˜X¨°(Ð';Ó<ñ	Ló =ð	Ló6Dr5   r   zkernel, has_arg))ÚallF)ÚanyF)ÚbfillF)ÚcorrT)ÚcorrwithT)ÚcovT)r`   T)r_   T)r\   T)rX   T)ÚdiffF)ÚffillF)rS   T)ÚidxmaxT)ÚidxminT)rT   T)rN   T)r7   T)r8   T)rM   T)ÚnthF)ÚnuniqueF)Ú
pct_changeF)r[   T)ÚquantileT)ÚsemT)ÚskewT)ÚkurtT)ÚstdT)rW   T)ÚvarTr?   TFÚkeysÚa1Úa2c                 óÜ  — t        ddgddgddgdt        gz  dœ«      }t        | |«      }|t        j                  u ri nd|i}|j                  |«      }t        || «      }|rO|du rK| dk(  r	t        }	d	}
nd }	d
}
t        j                  |	|
¬«      5   ||i |¤Ž}d d d «       dj                  vsJ ‚y | dv s| dv r+|t        j                  u r ||i |¤Ž}d|j                  v sJ ‚y |rÃ|dusJ ‚| j                  d«      rt        nt        }dj                  dddddt        j                   d| › d�«      g«      }
| dk(  rd}
n| dk(  rd}
n| dk(  rd}
t#        j$                  ||
¬«      5  | dk(  r	t        }	d	}
nd }	d
}
t        j                  |	|
¬«      5   ||i |¤Ž d d d «       d d d «       y |s?|t        j                  ur-t#        j$                  t        d¬«      5   ||i |¤Ž d d d «       y | dv sJ ‚|t        j                  u sJ ‚t#        j$                  t        d¬«      5   ||i |¤Ž d d d «       y # 1 sw Y   �Œ›xY w# 1 sw Y   Œ§xY w# 1 sw Y   y xY w# 1 sw Y   y xY w# 1 sw Y   y xY w) Nr   r   é   é   )r“   r”   Úa3r   r?   Trƒ   z'DataFrameGroupBy.corrwith is deprecatedÚ rj   r   rl   )r€   r   r�   r†   r‰   rŠ   rd   re   znot allowed for this dtypez+cannot be performed against 'object' dtypesz!must be a string or a real numberzunsupported operand typerh   rf   rg   rŒ   ú4dtype 'object' does not support operation 'quantile'rˆ   z8'<' not supported between instances of 'type' and 'type'r‡   z8'>' not supported between instances of 'type' and 'type'ú1got an unexpected keyword argument 'numeric_only')r…   r‹   )r   r'   r
   r   Ú
no_defaultr@   rB   r   rC   Úassert_produces_warningr-   rm   rn   ro   rp   rq   rr   rs   rt   )ÚkernelÚhas_argr?   r’   r3   ÚargsÚkwargsrH   r6   Úwarnrw   rJ   rv   s                r4   Útest_numeric_onlyr£   Ú   s½  € ôT 
˜1˜a˜&¨¨A¨°q¸!°fÀ1ÌÀxÁ<ÑPÓ	Q€Bä" 6¨2Ó.€DØ¤3§>¡>Ñ1‰R¸ÈÐ7U€Fà	�‰�DÓ	€BÜ�R˜Ó €FÙ�< 4Ñ'à�ZÒÜ!ˆDØ;‰CàˆDØˆCÜ×'Ñ'¨°CÔ8ñ 	-Ù˜TÐ, VÑ,ˆF÷	-à˜&Ÿ.™.Ñ(Ð(Ñ(ð 	Ð#Ñ#ð ÐHÑHØ¤§¡Ñ.ñ ˜Ð( Ñ(ˆØ�f—n‘nÑ$Ð$Ñ$Ù	Ø 4Ñ'Ð'Ð'ð
 ,2×+<Ñ+<¸UÔ+CÕ'Ìˆ	à�h‰hà,Ø=Ø3Ø*Ø<Ü—	‘	Ð6°v°h¸oÐNÓOðó	
ˆð �ZÒØH‰CØ�xÒØL‰CØ�xÒØLˆCÜ�]‰]˜9¨CÔ0ñ 	(Ø˜Ò#Ü%�Ø?‘à�Ø�Ü×+Ñ+¨D¸Ô<ñ (Ù˜Ð' Ò'÷(÷	(ð 	(ñ ˜¬S¯^©^Ñ;Ü�]‰]ÜÐPô
ñ 	$ñ �DÐ#˜FÒ#÷	$ð 	$ð
 Ð/Ñ/Ð/Ð/ØœsŸ~™~Ñ-Ð-Ð-ä�]‰]œ9Ð,GÔHñ 	$Ù�DÐ#˜FÒ#÷	$ð 	$÷q	-ñ 	-ú÷Z(ð (ú÷	(ð 	(ú÷	$ð 	$ú÷	$ð 	$úsH   Â	H1Å%*I
Æ	H>ÆI
Ç	IÈ	I"È1H;È>I	ÉI
É
IÉIÉ"I+r   c                 ó"  — g d¢}t        g d¢| ¬«      }|j                  |«      }|dk(  rt        ||«      rJ ‚y t        ||«      }t        g d¢«      }|j                  |«      }t        ||«      }	t	        ||«      }
d}d}||v r<| t
        u r4|dk(  rd}nd	}t        j                  t        |¬
«      5   ||
Ž  d d d «       nA| t
        u r9 ||
Ž } |	|
Ž }||v r|j                  t
        «      }t        j                  ||«       d}||vr/d}t        j                  t        |¬
«      5   ||
ddiŽ d d d «       y | t
        u r@dj                  g d¢«      }t        j                  t        |¬
«      5   ||
ddiŽ d d d «       y | t        k(  r4|dk(  r/d}t        j                  t        |¬
«      5   ||
ddiŽ d d d «       y  ||
ddiŽ} ||
ddiŽ}t        j                  ||«       y # 1 sw Y   ŒêxY w# 1 sw Y   y xY w# 1 sw Y   y xY w# 1 sw Y   y xY w)N)r   r   r   )r   r   r   r   rƒ   )r‚   r„   r`   r_   r\   rX   rŒ   )rS   rT   r‰   r�   r†   ÚshiftrW   r…   r‹   r‘   r7   r8   rM   rN   r[   rŽ   r�   rŒ   rš   z!is not supported for object dtyperj   )rS   rT   rN   r7   r8   rM   r[   rŒ   r�   rŽ   r�   r�   rW   r‘   r`   r_   r\   rX   r›   r?   Tre   )z@SeriesGroupBy.sem called with numeric_only=True and dtype objectz=Series.skew does not allow numeric_only=True with non-numericz7cum(sum|prod|min|max) is not supported for object dtypezCCannot use numeric_only=True with SeriesGroupBy\..* and non-numericz#Cannot use quantile with bool dtypeF)r   r@   ÚhasattrrB   r
   r'   rs   rt   ro   r/   rC   Úassert_series_equalrp   Úbool)r   Úgroupby_funcÚrequestÚgrouperÚserrH   r6   Úexpected_serÚexpected_gbÚexpected_methodr    Úfails_on_numeric_objectÚ
obj_resultrw   rJ   rI   Úhas_numeric_onlys                    r4   Ú"test_deprecate_numeric_only_seriesr³   O  s4  € ò €Gä
’ %Ô
(€CØ	�‰�WÓ	€Bà�zÒ!ä˜2˜|Ô,Ð,Ð,Øä�R˜Ó&€Fäš)Ó$€LØ×&Ñ& wÓ/€KÜ˜k¨<Ó8€Oä" <°Ó5€DðÐð€Jð, Ð.Ñ.°5¼F±?Ø˜:Ò%ØH‰Cà5ˆCÜ�]‰]œ9¨CÔ0ñ 	Ù�D‰M÷	ð 	à	”&‰Ù˜�ˆÙ" DÐ)ˆØ˜:Ñ%Ø—‘¤vÓ.ˆHÜ
×Ñ˜v xÔ0ðÐð( Ð+Ñ+ØAˆÜ�]‰]œ9¨CÔ0ñ 	-Ù�DÐ, tÒ,÷	-ð 	-à	”&‰Ø�h‰hòó
ˆô �]‰]œ9¨CÔ0ñ 	-Ù�DÐ, tÒ,÷	-ð 	-à	”$Š˜<¨:Ò5Ø3ˆÜ�]‰]œ9¨CÔ0ñ 	.á�DÐ- uÒ-÷	.ð 	.ñ ˜Ð1¨DÑ1ˆÙ˜4Ð4¨eÑ4ˆÜ
×Ñ˜v xÕ0÷i	ð 	ú÷>	-ð 	-ú÷	-ð 	-ú÷	.ð 	.ús0   Â0G!Ä#	G-Å+	G9Æ(	HÇ!G*Ç-G6Ç9HÈH)rq   rs   Úpandas._libsr   Úpandas.errorsr   Úpandasr0   r   r   r   r   r	   Úpandas._testingÚ_testingrC   Úpandas.tests.groupbyr
   r   r|   r}   rœ   r£   r¨   r$   r%   r'   r³   r~   r5   r4   ú<module>rº      sá   ðÛ 	ã å Ý (ã ÷õ õ Ý 8÷CDñ CDðL ‡�×ÑØò ó#ðH ‡�×Ñ˜¨$°°s·~±~Ð)FÓGØ‡�×Ñ˜ 4 &¨4°¨,Ð!7Ó8ñL$ó 9ó HóI#ðLL$ð^ ‡�×Ñ˜ 4¨¨e°VÐ"<Ó=ñm1ó >ñm1r5   