Ë
    ¤ehú%  ã                   óp   — d Z ddlZddlZddlZddlZddlmZmZm	Z	 ddl
mZ  G d„ d«      Z G d„ d«      Zy)zP
Tests for statistical reductions of 2nd moment or higher: var, skew, kurt, ...
é    N)Ú	DataFrameÚSeriesÚ
date_rangec                   ó¤  — e Zd Zej                  j                  deej                  ej                  g«      d„ «       Z
ej                  j                  deej                  ej                  g«      ej                  j                  dg d¢«      d„ «       «       Zej                  j                  deej                  ej                  g«      d„ «       Zy)ÚTestDatetimeLikeStatReductionsÚboxc                 óð  — |}t        dd|¬«      }|j                  g d¢«      }|j                  } ||«      }|j                  «       t	        j
                  d|¬«      k(  sJ ‚|j                  d¬«      t	        j
                  d|¬«      k(  sJ ‚t        j                  |d	<    ||«      }|j                  «       t	        j
                  d
|¬«      k(  sJ ‚|j                  d¬«      t        j                  u sJ ‚y )Nú
2001-01-01é   )ÚperiodsÚtz©é   é   é   é
   é	   é   é   é   r   é   é   z
2001-01-06)r   F©Úskipnaéþÿÿÿz2001-01-06 07:12:00)r   ÚtakeÚ_dataÚmeanÚpdÚ	TimestampÚNaT)ÚselfÚtz_naive_fixturer   r   ÚdtiÚdtarrÚobjs          új/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/pandas/tests/reductions/test_stat_reductions.pyÚtest_dt64_meanz-TestDatetimeLikeStatReductions.test_dt64_mean   sÎ   € àˆä˜¨r°bÔ9ˆà�h‰hÒ9Ó:ˆØ—	‘	ˆá�%‹jˆØ�x‰x‹zœRŸ\™\¨,¸2Ô>Ò>Ð>Ð>Ø�x‰x˜uˆxÓ%¬¯©°lÀrÔ)JÒJÐJÐJô —F‘Fˆˆb‰	á�%‹jˆØ�x‰x‹zœRŸ\™\Ð*?ÀBÔGÒGÐGÐGØ�x‰x˜uˆxÓ%¬¯©Ñ/Ð/Ñ/ó    Úfreq)ÚsÚhÚDÚWÚBc                 ó.  — t        dd¬«      }|j                  g d¢«      }|dk(  rt        nd }d}t        j                  ||¬«      5  |j
                  j                  |«      }d d d «        |«      }t        j                  t        d¬«      5  |j                  «        d d d «       t        j                  t        d¬«      5  |j                  d	¬
«       d d d «       t        j                  |d<   t        j                  t        d¬«      5  |j                  «        d d d «       t        j                  t        d¬«      5  |j                  d	¬
«       d d d «       y # 1 sw Y   ŒùxY w# 1 sw Y   ŒÉxY w# 1 sw Y   ŒŸxY w# 1 sw Y   ŒdxY w# 1 sw Y   y xY w)Nr
   r   ©r   r   r/   zPeriodDtype\[B\] is deprecated©ÚmatchÚ	ambiguousTr   r   )r   r   ÚFutureWarningÚtmÚassert_produces_warningr   Ú	to_periodÚpytestÚraisesÚ	TypeErrorr   r   r!   )r"   r   r*   r$   ÚwarnÚmsgÚparrr&   s           r'   Útest_period_meanz/TestDatetimeLikeStatReductions.test_period_mean'   sA  € ô ˜¨rÔ2ˆà�h‰hÒ9Ó:ˆà $¨¢�}°ˆØ/ˆÜ×'Ñ'¨°CÔ8ñ 	-Ø—9‘9×&Ñ& tÓ,ˆD÷	-á�$‹iˆÜ�]‰]œ9¨KÔ8ñ 	Ø�H‰HŒJ÷	ä�]‰]œ9¨KÔ8ñ 	"Ø�H‰H˜DˆHÔ!÷	"ô —6‘6ˆˆR‰ä�]‰]œ9¨KÔ8ñ 	Ø�H‰HŒJ÷	ä�]‰]œ9¨KÔ8ñ 	"Ø�H‰H˜DˆHÔ!÷	"ð 	"÷	-ð 	-ú÷	ð 	ú÷	"ð 	"ú÷	ð 	ú÷	"ð 	"ús<   ÁEÂE'ÃE3ÄE?Ä?FÅE$Å'E0Å3E<Å?FÆFc                 ó4  — t        j                  g d¢d«      }t        j                  |«      j	                  d«      }|j
                  } ||d¬«      }|j                  «       }t        j                  |«      j                  «       }||k(  sJ ‚t        j                  |d<   |j                  d¬«      t        j                  u sJ ‚|j                  d¬«      }||d	d  j                  «       k(  sJ ‚|j                  d
«      |dz  dz  j                  d
«      k(  sJ ‚y )N)r   r   r   iùÿÿÿr   r   éÿÿÿÿr   r   r   r   zm8[D]ÚnsF)Úcopyr   r   Tr   Úusg      &@r   )	ÚnpÚarrayr   ÚTimedeltaIndexÚas_unitr   r   r!   Úround)	r"   r   Úm8valuesÚtdiÚtdarrr&   ÚresultÚexpectedÚresult2s	            r'   Útest_td64_meanz-TestDatetimeLikeStatReductions.test_td64_meanA   só   € ä—8‘8ÒAÀ7ÓKˆÜ×Ñ Ó)×1Ñ1°$Ó7ˆà—	‘	ˆÙ�%˜eÔ$ˆà—‘“ˆÜ—8‘8˜E“?×'Ñ'Ó)ˆØ˜Ò!Ð!Ð!ä—6‘6ˆˆa‰Ø�x‰x˜uˆxÓ%¬¯©Ñ/Ð/Ð/à—(‘( $�(Ó'ˆØ˜#˜a˜b˜'Ÿ,™,›.Ò(Ð(Ð(ð �}‰}˜TÓ" v°¡}°rÑ'9×&@Ñ&@ÀÓ&FÒFÐFÑFr)   N)Ú__name__Ú
__module__Ú__qualname__r9   ÚmarkÚparametrizer   r   ÚIndexrF   r(   r?   rP   © r)   r'   r   r      s¼   „ Ø‡[�[×Ñ˜U V¨R¯X©X°r·x±xÐ$@ÓAñ0ó Bð0ð& ‡[�[×Ñ˜U V¨R¯X©X°r·x±xÐ$@ÓAØ‡[�[×Ñ˜VÒ%>Ó?ñ"ó @ó Bð"ð0 ‡[�[×Ñ˜U V¨R¯X©X°r·x±xÐ$@ÓAñGó BñGr)   r   c                   óX   — e Zd Z	 dd„Zd„ Zd„ Zd„ Zd„ Zd„ Zd„ Z	d„ Z
d	„ Zd
„ Zd„ Zd„ Zy)ÚTestSeriesStatReductionsc           	      óÖ  — t        j                  dd«      5  t        t        |«      }t        j
                  |dd |dvrHt        t        dd¬«      «      }d	|› d
�}t        j                  t        |¬«      5   ||«       d d d «       t        j                   ||«      «      sJ ‚t        j                   ||d¬«      «      sJ ‚|j                  «       }	t        j                   ||	«       ||	j                  «      «       t        j                   ||«       ||	j                  «      «       |t        j
                  z  }
|rt	        j                    ||
«      «      sJ ‚t        g d¢«      } ||«       dg}|j#                  t%        dd«      «       t        |d¬«      }t        j                  t'         ||«      «      t'         ||j                  «      «      «       |r7t        t        j(                  dd¬«      «      } ||«      } ||«      }||k(  sJ ‚|dvr>t        j                  t        d ¬«      5   |t        t+        d«      «      «       d d d «       d}t        j                  t,        |¬«      5   ||d¬«       d d d «       dt/        j0                  |«      j2                  v r
 ||d¬«       d d d «       y # 1 sw Y   �Œ=xY w# 1 sw Y   ŒzxY w# 1 sw Y   ŒVxY w# 1 sw Y   y xY w)NÚuse_bottleneckFr   é   )ÚmaxÚminr   ÚmedianÚstdz1/1/2001r   r1   zdoes not support reduction 'ú'r2   r   )r   r   r   Nr   r   l        l   è   Úint64©Údtypez1/1/2000)Úsumr^   r]   Úabcz&No axis named 1 for object type Seriesr   )ÚaxisÚnumeric_onlyT)rh   )r   Úoption_contextÚgetattrr   rE   Únanr   r9   r:   r;   ÚnotnaÚisnaÚdropnar6   Úassert_almost_equalÚvaluesÚisnanÚextendÚrangeÚfloatÚbdate_rangeÚlistÚ
ValueErrorÚinspectÚgetfullargspecÚargs)r"   ÚnameÚ	alternateÚstring_series_Úcheck_objectsÚcheck_allnaÚfÚdsr=   ÚnonaÚallnar+   ÚitemsÚresÚexps                  r'   Ú_check_stat_opz'TestSeriesStatReductions._check_stat_op\   sz  € ô ×ÑÐ/°Ó7ñ 8	5Üœ Ó%ˆAô $&§6¡6ˆN˜1˜RÐ ð ÐBÑBÜœJ z¸2Ô>Ó?�Ø4°T°F¸!Ð<�Ü—]‘]¤9°CÔ8ñ Ù�b”E÷ô —8‘8™A˜nÓ-Ô.Ð.Ð.Ü—7‘7™1˜^°EÔ:Ô;Ð;Ð;ð "×(Ñ(Ó*ˆDÜ×"Ñ"¡1 T£7©I°d·k±kÓ,BÔCÜ×"Ñ"¡1 ^Ó#4±iÀÇÁÓ6LÔMà"¤R§V¡VÑ+ˆEáÜ—x‘x¡ %£Ô)Ð)Ð)ô Ò)Ó*ˆAÙˆaŒDð �CˆEØ�L‰Lœ˜u lÓ3Ô4Ü�u GÔ,ˆAÜ×"Ñ"¤5©¨1«£;´±iÀÇÁÓ6IÓ0JÔKñ Üœ2Ÿ>™>¨*¸bÔAÓB�Ù˜“d�Ù “l�Ø˜c’zÐ!�zð Ð0Ñ0Ü—]‘]¤9°DÔ9ñ +Ù”fœT %›[Ó)Ô*÷+ð ;ˆCÜ—‘œz°Ô5ñ *Ù�. qÕ)÷*ð ¤×!7Ñ!7¸Ó!:×!?Ñ!?Ñ?á�.¨tÕ4÷q8	5ð 8	5÷ñ ú÷H+ð +ú÷
*ð *ú÷g8	5ð 8	5úsU   —AKÁ6	J:Á?F4KÈ3KÉ%KÉ3KÉ>3KÊ:K	Ê?KËK	ËKËK	ËKËK(c                 ó”   — t        t        d«      t        j                  d¬«      }| j	                  dt        j
                  |d¬«       y )Né   Úseries©rd   r{   re   F)r   )r   rs   rE   Úfloat64r‡   re   ©r"   Ústring_seriess     r'   Útest_sumz!TestSeriesStatReductions.test_sum™   s3   € Üœu R›y´·
±
ÀÔJˆØ×Ñ˜E¤2§6¡6¨=ÀeÐÕLr)   c                 ó�   — t        t        d«      t        j                  d¬«      }| j	                  dt        j
                  |«       y )Nr‰   rŠ   r‹   r   )r   rs   rE   rŒ   r‡   r   r�   s     r'   Ú	test_meanz"TestSeriesStatReductions.test_mean�   ó.   € Üœu R›y´·
±
ÀÔJˆØ×Ñ˜F¤B§G¡G¨]Õ;r)   c                 ó\  — t        t        d«      t        j                  d¬«      }| j	                  dt        j
                  |«       t        t        j                  dt        ¬«      t        d«      ¬«      }t        j                  t        j
                  |«      |j                  «       «       y )Nr‰   rŠ   r‹   r_   r   rc   )Úindex)
r   rs   rE   rŒ   r‡   r_   ÚonesÚintr6   ro   )r"   rŽ   Úint_tss      r'   Útest_medianz$TestSeriesStatReductions.test_median¡   sj   € Üœu R›y´·
±
ÀÔJˆØ×Ñ˜H¤b§i¡i°Ô?ô œŸ™ ¬#Ô.´e¸B³iÔ@ˆÜ
×ÑœrŸy™y¨Ó0°&·-±-³/ÕBr)   c                 ó�   — t        t        d«      t        j                  d¬«      }| j	                  dt        j
                  |«       y )Nr‰   rŠ   r‹   Úprod)r   rs   rE   rŒ   r‡   rš   r�   s     r'   Ú	test_prodz"TestSeriesStatReductions.test_prod©   r’   r)   c                 ó”   — t        t        d«      t        j                  d¬«      }| j	                  dt        j
                  |d¬«       y )Nr‰   rŠ   r‹   r^   T©r~   )r   rs   rE   rŒ   r‡   r^   r�   s     r'   Útest_minz!TestSeriesStatReductions.test_min­   ó3   € Üœu R›y´·
±
ÀÔJˆØ×Ñ˜E¤2§6¡6¨=ÈÐÕMr)   c                 ó”   — t        t        d«      t        j                  d¬«      }| j	                  dt        j
                  |d¬«       y )Nr‰   rŠ   r‹   r]   Tr�   )r   rs   rE   rŒ   r‡   r]   r�   s     r'   Útest_maxz!TestSeriesStatReductions.test_max±   rŸ   r)   c                 ó  — t        t        d«      t        j                  d¬«      }t        t        j                  dt        j                  ¬«      t        dd¬«      d¬	«      }d
„ }| j                  d||«       d„ }| j                  d||«       |j                  d¬«      }t        j                  |j                  d¬«      }t        j                  ||«       |j                  d¬«      }t        j                  |j                  d¬«      }t        j                  ||«       |j                  dg   }|j                  d¬«      }t        j                  |«      sJ ‚|j                  d¬«      }t        j                  |«      sJ ‚y )Nr‰   rŠ   r‹   r   rc   ú
2020-01-01r1   Úts©r”   r{   c                 ó0   — t        j                  | d¬«      S ©Nr   ©Úddof)rE   r`   ©Úxs    r'   ú<lambda>z7TestSeriesStatReductions.test_var_std.<locals>.<lambda>½   ó   € œŸ™˜q qÔ)€ r)   r`   c                 ó0   — t        j                  | d¬«      S r§   )rE   Úvarrª   s    r'   r¬   z7TestSeriesStatReductions.test_var_std.<locals>.<lambda>À   r­   r)   r¯   r   r¨   r   r   )r   rs   rE   rŒ   Úaranger   r‡   r`   rp   r6   ro   r¯   Úilocr   rm   ©r"   rŽ   Údatetime_seriesÚaltrM   rN   r+   s          r'   Útest_var_stdz%TestSeriesStatReductions.test_var_stdµ   s5  € Üœu R›y´·
±
ÀÔJˆÜ Ü�I‰I�b¤§
¡
Ô+Ü˜\°2Ô6Øô
ˆñ *ˆØ×Ñ˜E 3¨Ô6á)ˆØ×Ñ˜E 3¨Ô6à ×$Ñ$¨!Ð$Ó,ˆÜ—6‘6˜/×0Ñ0°qÔ9ˆÜ
×Ñ˜v xÔ0à ×$Ñ$¨!Ð$Ó,ˆÜ—6‘6˜/×0Ñ0°qÔ9ˆÜ
×Ñ˜v xÔ0ð × Ñ  ! Ñ%ˆØ—‘˜A�“ˆÜ�w‰w�vŒÐˆà—‘˜A�“ˆÜ�w‰w�vŒÐ‰r)   c                 óH  — t        t        d«      t        j                  d¬«      }t        t        j                  dt        j                  ¬«      t        dd¬«      d¬	«      }d
„ }| j                  d||«       |j                  d¬«      }t        j                  |j                  d¬«      t        j                  t        |j                  «      «      z  }t        j                  ||«       |j                  dg   }|j                  d¬«      }t        j                   |«      sJ ‚y )Nr‰   rŠ   r‹   r   rc   r£   r1   r¤   r¥   c                 ón   — t        j                  | d¬«      t        j                  t        | «      «      z  S r§   )rE   r`   ÚsqrtÚlenrª   s    r'   r¬   z3TestSeriesStatReductions.test_sem.<locals>.<lambda>Û   s"   € œŸ™˜q qÔ)¬B¯G©G´C¸³F«OÑ;€ r)   Úsemr   r¨   r   r   )r   rs   rE   rŒ   r°   r   r‡   rº   r`   rp   r¸   r¹   r6   ro   r±   r   rm   r²   s          r'   Útest_semz!TestSeriesStatReductions.test_semÓ   sæ   € Üœu R›y´·
±
ÀÔJˆÜ Ü�I‰I�b¤§
¡
Ô+Ü˜\°2Ô6Øô
ˆñ <ˆØ×Ñ˜E 3¨Ô6à ×$Ñ$¨!Ð$Ó,ˆÜ—6‘6˜/×0Ñ0°qÔ9¼B¿G¹GÜ�×&Ñ&Ó'ó=
ñ 
ˆô 	×Ñ˜v xÔ0ð × Ñ  ! Ñ%ˆØ—‘˜A�“ˆÜ�w‰w�vŒÐ‰r)   c                 óÖ  ‡— t        j                  d«      Št        t        d«      t        j
                  d¬«      }ˆfd„}| j                  d||«       d}t        d|dz   «      D �]  }t        t	        j                  |«      «      }t        t	        j                  ||f«      «      }||k  rYt	        j                  |j                  «       «      sJ ‚t	        j                  |j                  «       «      j                  «       rŒžJ ‚d	|j                  «       k(  sJ ‚t        |j                  «       t        j
                  «      sJ ‚|j                  «       d	k(  j                  «       r�ŒJ ‚ y )
Núscipy.statsr‰   rŠ   r‹   c                 ó*   •— ‰j                  | d¬«      S ©NF)Úbias)Úskew©r«   Úsp_statss    €r'   r¬   z4TestSeriesStatReductions.test_skew.<locals>.<lambda>î   s   ø€ ˜Ÿ™ a¨e˜Ó4€ r)   rÁ   r   r   r   )r9   Úimportorskipr   rs   rE   rŒ   r‡   r•   r   rq   rÁ   ÚallÚ
isinstance)r"   rŽ   r´   Úmin_NÚir+   ÚdfrÃ   s          @r'   Ú	test_skewz"TestSeriesStatReductions.test_skewé   s  ø€ Ü×&Ñ& }Ó5ˆäœu R›y´·
±
ÀÔJˆã4ˆØ×Ñ˜F C¨Ô7ð ˆÜ�q˜% !™)Ó$ó 		.ˆAÜ”r—w‘w˜q“zÓ"ˆAÜœ2Ÿ7™7 A q 6›?Ó+ˆBØ�5ŠyÜ—x‘x §¡£Ô)Ð)Ð)Ü—x‘x §¡£	Ó*×.Ñ.Õ0Ð0Ð0à˜AŸF™F›H’}Ð$�}Ü! !§&¡&£(¬B¯J©JÔ7Ð7Ð7ØŸ™›	 Q™×+Ñ+Ö-Ð-Ð-ñ		.r)   c                 óª   ‡— t        j                  d«      Št        t        d«      t        j
                  d¬«      }ˆfd„}| j                  d||«       y )Nr½   r‰   rŠ   r‹   c                 ó*   •— ‰j                  | d¬«      S r¿   )ÚkurtosisrÂ   s    €r'   r¬   z4TestSeriesStatReductions.test_kurt.<locals>.<lambda>  s   ø€ ˜×)Ñ)¨!°%Ð)Ó8€ r)   Úkurt)r9   rÄ   r   rs   rE   rŒ   r‡   )r"   rŽ   r´   rÃ   s      @r'   Ú	test_kurtz"TestSeriesStatReductions.test_kurtÿ   s@   ø€ Ü×&Ñ& }Ó5ˆäœu R›y´·
±
ÀÔJˆã8ˆØ×Ñ˜F C¨Õ7r)   c                 ó0  — d}t        d|dz   «      D �]  }t        t        j                  |«      «      }t	        t        j                  ||f«      «      }||k  rYt        j
                  |j                  «       «      sJ ‚t        j
                  |j                  «       «      j                  «       rŒžJ ‚d|j                  «       k(  sJ ‚t        |j                  «       t        j                  «      sJ ‚|j                  «       dk(  j                  «       r�ŒJ ‚ y )Nr   r   r   )
rs   r   rE   r•   r   rq   rÎ   rÅ   rÆ   rŒ   )r"   rÇ   rÈ   r+   rÉ   s        r'   Útest_kurt_cornerz)TestSeriesStatReductions.test_kurt_corner  sÓ   € ð ˆÜ�q˜% !™)Ó$ó 		.ˆAÜ”r—w‘w˜q“zÓ"ˆAÜœ2Ÿ7™7 A q 6›?Ó+ˆBØ�5ŠyÜ—x‘x §¡£Ô)Ð)Ð)Ü—x‘x §¡£	Ó*×.Ñ.Õ0Ð0Ð0à˜AŸF™F›H’}Ð$�}Ü! !§&¡&£(¬B¯J©JÔ7Ð7Ð7ØŸ™›	 Q™×+Ñ+Ö-Ð-Ð-ñ		.r)   N)FF)rQ   rR   rS   r‡   r�   r‘   r˜   r›   rž   r¡   rµ   r»   rÊ   rÏ   rÑ   rW   r)   r'   rY   rY   W   sJ   „ ð QVó;5òzMò<òCò<òNòNòò<ò,.ò,8ó.r)   rY   )Ú__doc__rx   ÚnumpyrE   r9   Úpandasr   r   r   r   Úpandas._testingÚ_testingr6   r   rY   rW   r)   r'   ú<module>r×      sB   ðñó ã Û ã ÷ñ õ
 ÷BGñ BG÷J}.ò }.r)   