Ë
    ž…jIU  ã                   óÞ  — d dl mZ d dlZd dlZd dlZd dlmZmZ d dl	m
c mZ d dlmZmZmZmZmZ d dlmZ d dlmZ ej0                  d„ «       Z ej0                  d ej4                  dd	d
dej6                  dddej6                  d	g
«      gd ej4                  ddddej6                  dddej6                  dg
«      gd ej4                  ddddej6                  dddej6                  dg
«      gd ej4                  ddddej6                  dddej6                  dg
«      gd ej4                  ddddej6                  dddej6                  dg
«      ggd„ ¬«      d„ «       Z ej0                  ddddd  ej:                  d! ej<                  d"«      ¬#«       ej:                  d$ ej<                  d"«      ¬#«       ej:                  d% ej<                  d"«      ¬#«      d&d'g
¬(«      d)„ «       ZdMd*„Z  G d+„ d,«      Z!ejD                  jG                  d-dgd.gfddgd/d.gfddgd.d.gfg d0¢g d1¢fg d2¢g d3¢fg d4¢g d5¢fg d6¢g d7¢fg d8¢g d9¢fg d:¢g d;¢fg	«      d<„ «       Z$ejD                  jG                  d-dgd.gfddgd/d.gfddgd/d/gfg d0¢g d1¢fg d2¢g d=¢fg d4¢g d5¢fg d6¢g d>¢fg d8¢g d?¢fg d:¢g d;¢fg	«      d@„ «       Z%ejD                  jG                  d-dgd.gfddgd/d.gfddgd.d.gfg d0¢g d1¢fg d2¢g dA¢fg d4¢g d5¢fg d6¢g dB¢fg d8¢g d9¢fg d:¢g d;¢fg	«      dC„ «       Z&ejD                  jG                  d-dgd.gfddgd/d.gfddgdDdDgfg d0¢g d1¢fg d2¢g dE¢fg d4¢g d5¢fg d6¢g dF¢fg d8¢g dG¢fg d:¢g d;¢fg	«      dH„ «       Z'ejD                  jG                  d-dgd.gfddgd/d.gfddgd/d.gfg d0¢g d1¢fg d2¢g d1¢fg d4¢g d5¢fg d6¢g dI¢fg d8¢g dJ¢fg d:¢g d;¢fg	«      dK„ «       Z(ejD                  jR                  dL„ «       Z*y)Né    )ÚchainN)ÚInfinityÚNegInfinity)ÚNAÚNaTÚSeriesÚ	TimestampÚ
date_range)ÚCategoricalDtypec                  ód   — t        ddddt        j                  dddt        j                  dg
«      S )Né   é   é   é   é   )r   ÚnpÚnan© ó    úf/root/aria/tools/markitdown-venv/lib/python3.12/site-packages/pandas/tests/series/methods/test_rank.pyÚserr      s*   € ä�1�a˜˜AœrŸv™v q¨!¨Q´·±¸Ð:Ó;Ð;r   Úaverageg      ø?g      @ç      @g      @g       @Úminr   r   é   r   é   Úmaxr   é   r   ÚfirstÚdensec                 ó   — | d   S )Nr   r   )Úxs    r   ú<lambda>r#   %   s
   € �!�A‘$€ r   )ÚparamsÚidsc                 ó   — | j                   S ©N©Úparam©Úrequests    r   Úresultsr,      s   € ð �=‰=Ðr   ÚobjectÚfloat64Úint64ÚFloat64ÚInt64úfloat64[pyarrow]Úpyarrow©Úmarksúint64[pyarrow]ústring[pyarrow]zstring[python]Ústr)r$   c                 ó   — | j                   S r'   r(   r*   s    r   Údtyper:   +   s   € ð �=‰=Ðr   c                 ób   — d}| dv rd}|S | dv r|dk(  s|rd}|S d}|S | dv r|dk(  s|rd}|S d	}|S )
Nr.   )r7   r0   )r2   r6   r   zdouble[pyarrow]úuint64[pyarrow])r0   r1   ÚUInt64r   )r:   ÚmethodÚpctÚ	exp_dtypes       r   Úexpected_dtyperA   =   s}   € Ø€IàÐ#Ñ#Øˆ	ð Ðð 
Ð8Ñ	8Ø�YÒ¡#Ø)ˆIð Ðð *ˆIð Ðð 
Ð&Ñ	&Ø�YÒ¡#Ø!ˆIð Ðð !ˆIàÐr   c                   ó  — e Zd Zd„ Zd„ Zd„ Zd„ Zd„ Zej                  j                  dg d¢«      ej                  j                  dd	d
 e«        e«       fdej                  ej                  ej                   fdeej                  ej                   f ej"                  deej                  ej                    ej&                  d«      ¬«      g«      d„ «       «       Zd„ Zej                  j                  dej.                  dgej.                  dgej0                  dgg«      d„ «       Zej                  j                  ddgdgfdgdgfdgdgfddgddgfg d¢g d¢fg d¢g d¢fg d¢g d¢fg d¢g d ¢fg«      d!„ «       Zd"„ Zd#„ Zd$„ Zd%„ Zd&„ Zy
)'ÚTestSeriesRankc                 óÐ
  — t        j                  d«      }t        j                  |d d d…<   d|d dd…<   |j	                  «       }|j                  d«      j	                  «       }t        j                  ||«       t        j                  |«      }|j                  t        j                  «      }t        |j                  |«      |j                  d¬«      }t        j                  ||<   t        j                  ||«       t        t        j                  d	«      j                  d«      «      }|j	                  «       }	|j                  t         «      j	                  «       }t        j                  |	|«       t        t        j                  d	«      «      d
z   }|dz  }|j	                  d¬«      }	t        j                  |	|«       t        t        j                  dd«      «      }t        t        j                  dd«      «      }|j	                  d¬«      }	t        j                  |	|«       |j                  d«      }t        j                  |d<   t        t        j                  dd«      «      }t        j                  |d<   |j	                  d¬«      }	t        j                  |	|«       t        t        j                  d	«      «      d
z   }t        j                  |d<   |dz  }|j	                  d¬«      }	t        j                  |	|«       t        t        j                  t        j                  d«      «      }|}|j	                  d¬«      }	t        j                  |	|«       t        t        j                  d	«      d¬«      dz   }t        j                  |d<   |dz  }|j	                  d¬«      }	t        j                  |	|«       t#        dd	¬«      }
t        t        j                  d	«      |
d¬«      dz   }t        j                  |j$                  d<   |dz  }|j	                  d¬«      }	t        j                  |	|«       t        g d¢«      }t        g d¢«      }|j	                  «       }	t        j                  |	|«       t        dddt&        gd¬«      }t        dddt        j                  g«      }|j	                  «       }	t        j                  |	|«       t        j(                  g d¢d¬«      }t        j*                  j-                  d«      j/                  t1        |«      «      }t        ||   «      }t        |d
z   d¬«      }|j	                  «       }	t        j                  |	|«       y )Núscipy.statsr   ç      @é
   r   ÚOÚts)ÚindexÚnamer   ç      ð?ç      @T©r?   r   éd   g)\�Âõ(à?Úfloatg¥@­_)à?r   ©r:   z1/1/1990)Úperiods)g¸ÔJzî�5g0Žä.ÿ++ç#B’¡œÇ;g{®Gáz„?gIfœ¡œÇ;gš™™™™™¹?)r   r   r   r   r   ç      @z3 dayz	1 day 10mz-2 dayzm8[ns])iÎÿÿÿéÿÿÿÿg#B’¡œÇ»gÙ}ÚõÐò¾ºg¸ÔJzî�µr   gœWw'&l¡7rS   g»½×Ùß|Û=r   é(   r.   )ÚpytestÚimportorskipr   r   ÚrankÚastypeÚtmÚassert_series_equalÚisnanÚfillnaÚinfr   ÚrankdatarJ   ÚarangeÚrepeatrP   r
   Úilocr   ÚarrayÚrandomÚdefault_rngÚpermutationÚlen)ÚselfÚdatetime_seriesÚsp_statsÚranksÚoranksÚmaskÚfilledÚexpÚiseriesÚiranksÚrngÚvaluesÚrandom_orders                r   Ú	test_rankzTestSeriesRank.test_rankQ   s,  € Ü×&Ñ& }Ó5ˆä!Ÿv™vˆ™˜!˜ÑØ!$ˆ˜˜˜A˜Ñà×$Ñ$Ó&ˆØ ×'Ñ'¨Ó,×1Ñ1Ó3ˆä
×Ñ˜u fÔ-ä�x‰x˜Ó(ˆØ ×'Ñ'¬¯©Ó/ˆô �X×&Ñ& vÓ.°f·l±lÈÔNˆÜ—F‘FˆˆD‰	ä
×Ñ˜u cÔ*äœŸ™ 1›×,Ñ,¨QÓ/Ó0ˆà—‘“ˆØ�n‰nœUÓ#×(Ñ(Ó*ˆÜ
×Ñ˜v sÔ+ÜœŸ™ 1›Ó&¨Ñ,ˆØ˜‰mˆØ—‘ $�Ó'ˆä
×Ñ˜v sÔ+äœŸ™ 1 cÓ*Ó+ˆÜ”R—Y‘Y˜u cÓ*Ó+ˆØ—‘ $�Ó'ˆÜ
×Ñ˜v sÔ+ð —.‘. Ó)ˆÜ—V‘Vˆ�‰
Ü”R—Y‘Y˜{¨CÓ0Ó1ˆÜ—‘ˆˆA‰Ø—‘ $�Ó'ˆÜ
×Ñ˜v sÔ+äœŸ™ 1›Ó&¨Ñ,ˆÜ—V‘Vˆ�‰
Ø˜‰mˆØ—‘ $�Ó'ˆÜ
×Ñ˜v sÔ+äœŸ™¤2§6¡6¨3Ó/Ó0ˆØˆØ—‘ $�Ó'ˆÜ
×Ñ˜v sÔ+ô œŸ™ 1›¨WÔ5¸Ñ9ˆÜ—V‘Vˆ�‰
Ø˜‰mˆØ—‘ $�Ó'ˆÜ
×Ñ˜v sÔ+ä˜¨QÔ/ˆäœŸ™ 1› s°'Ô:¸QÑ>ˆÜŸ&™&ˆ�‰�Q‰Ø˜‰mˆØ—‘ $�Ó'ˆÜ
×Ñ˜v sÔ+äÒJÓKˆÜÒ)Ó*ˆØ—‘“ˆÜ
×Ñ˜v sÔ+ô ˜' ;°¼#Ð>ÀhÔOˆÜ�a˜˜AœrŸv™vÐ&Ó'ˆØ—‘“ˆÜ
×Ñ˜v sÔ+ä—‘ÚLØô
ˆô —y‘y×,Ñ,¨QÓ/×;Ñ;¼CÀ»KÓHˆÜ˜ Ñ-Ó.ˆÜ�\ CÑ'¨yÔ9ˆØ—‘“ˆÜ
×Ñ˜v sÕ+r   c           	      ó–  — t        g d¢«      }t        g d¢«      }t        g d¢«      j                  t        g d¢d¬«      «      }t        j                  |j                  «       |«       t        j                  |j                  d¬«      |«       t        g d¢«      j                  t        g d¢d¬«      «      }t        g d¢«      }|j                  «       }t        j                  ||«       t        g d	¢«      j                  t        g d	¢d«      «      }t        g d¢«      }|j                  «       }	t        j                  |	|«       t        d
dddddt        j                  g«      j                  t        g d¢d«      «      }
t        g d¢«      }t        g d¢«      }t        ddddddt        j                  g«      }t        j                  |
j                  d¬«      |«       t        j                  |
j                  d¬«      |«       t        j                  |
j                  d¬«      |«       t        g d¢«      }t        g d¢«      }t        ddddddt        j                  g«      }t        j                  |
j                  dd¬«      |«       t        j                  |
j                  dd¬«      |«       t        j                  |
j                  dd¬«      |«       d }t        j                  t        |¬!«      5  |
j                  d"d¬«       d d d «       t        j                  t        |¬!«      5  |
j                  dd¬«       d d d «       t        d
dddt        j                  g«      j                  t        g d#¢d«      «      }
t        g d$¢«      }t        g d%¢«      }t        d&d'd(dt        j                  g«      }t        j                  |
j                  dd¬)«      |«       t        j                  |
j                  dd¬)«      |«       t        j                  |
j                  dd¬)«      |«       y # 1 sw Y   �Œ(xY w# 1 sw Y   ŒýxY w)*N)rL   ç       @ç      @rF   rM   rT   )rT   rM   rF   ry   rx   rL   )r   ÚsecondÚthirdÚfourthÚfifthÚsixthT)Ú
categoriesÚorderedF©Ú	ascending)rx   rF   rT   ry   rL   rM   )r   r   r   r   r   r   r   rz   r{   r|   r}   r~   )r   rz   r{   r|   r}   r~   Úseventh)rx   ry   rF   rM   rT   r   rL   )rL   rx   ry   rF   rM   rT   r   rL   rx   ry   rF   rM   rT   Útop©Ú	na_optionÚbottomÚkeep)r   rT   rM   rF   ry   rx   rL   )rT   rM   rF   ry   rx   rL   r   )r†   r‚   z3na_option must be one of 'keep', 'top', or 'bottom'©ÚmatchÚbad)r   rz   r{   r|   )çš™™™™™Ù?ç333333ã?çš™™™™™é?rL   çš™™™™™É?©r�   rŒ   r�   rŽ   rL   ç      Ð?ç      à?ç      è?)r†   r?   )r   rZ   r   r[   r\   rY   r   r   rW   ÚraisesÚ
ValueError)ri   rp   Úexp_descr€   Ú	unorderedÚexp_unorderedÚresÚ
unordered1Úexp_unordered1Úres1Úna_serÚexp_topÚexp_botÚexp_keepÚmsgs                  r   Útest_rank_categoricalz$TestSeriesRank.test_rank_categorical¬   s�  € ô Ò3Ó4ˆÜÒ8Ó9ˆÜÚDó
ç
‰&ÜÚSØôó
ð 	ô 	×Ñ˜wŸ|™|›~¨sÔ3Ü
×Ñ˜wŸ|™|°e˜|Ó<¸hÔGô ÚDó
ç
‰&ÜÚSØôó
ð 	ô Ò=Ó>ˆØ�n‰nÓˆÜ
×Ñ˜s MÔ2äÒ.Ó/×6Ñ6ÜÒ/°Ó7ó
ˆ
ô  Ò >Ó?ˆØ�‰Ó ˆÜ
×Ñ˜t ^Ô4ô Ø�h ¨°7¸GÄRÇVÁVÐLó
ç
‰&ÜÚSØóó
ð 	ô Ò<Ó=ˆÜÒ<Ó=ˆÜ˜3  S¨#¨s°C¼¿¹Ð@ÓAˆä
×Ñ˜vŸ{™{°U˜{Ó;¸WÔEÜ
×Ñ˜vŸ{™{°X˜{Ó>ÀÔHÜ
×Ñ˜vŸ{™{°V˜{Ó<¸hÔGô Ò<Ó=ˆÜÒ<Ó=ˆÜ˜3  S¨#¨s°C¼¿¹Ð@ÓAˆä
×Ñ˜vŸ{™{°UÀe˜{ÓLÈgÔVÜ
×ÑØ�K‰K (°eˆKÓ<¸gô	
ô 	×Ñ˜vŸ{™{°VÀu˜{ÓMÈxÔXð Dˆä�]‰]œ:¨SÔ1ñ 	:Ø�K‰K %°5ˆKÔ9÷	:ô �]‰]œ:¨SÔ1ñ 	9Ø�K‰K $°%ˆKÔ8÷	9ô ˜ (¨G°X¼r¿v¹vÐFÓG×NÑNÜÒCÀTÓJó
ˆô Ò2Ó3ˆÜÒ2Ó3ˆÜ˜4  d¨C´·±Ð8Ó9ˆä
×Ñ˜vŸ{™{°UÀ˜{ÓEÀwÔOÜ
×Ñ˜vŸ{™{°XÀ4˜{ÓHÈ'ÔRÜ
×Ñ˜vŸ{™{°VÀ˜{ÓFÈÕQ÷#	:ñ 	:ú÷	9ð 	9ús   Ë8P2Ì/P?Ð2P<Ð?Qc                 ó�   — t        g d¢d¬«      }|j                  d¬«      }t        g d¢d¬«      }t        j                  ||«       y )N)Nr   Nr   r   r   r   r   r1   rQ   rˆ   r…   )Nç      @NrM   rM   r¤   rM   rL   r0   )r   rY   r[   r\   )ri   rp   ÚresultÚexpecteds       r   Útest_rank_nullable_integerz)TestSeriesRank.test_rank_nullable_integer  s<   € äÒ3¸7ÔCˆØ—‘ F�Ó+ˆäÒDÈIÔVˆä
×Ñ˜v xÕ0r   c                 óÈ   — t        ddg«      }|j                  d¬«       d}t        j                  t        |¬«      5  |j                  d«       d d d «       y # 1 sw Y   y xY w)Nr   r   r   ©r>   z,No axis named average for object type Seriesr‰   )r   rY   rW   r”   r•   )ri   Úsr¡   s      r   Útest_rank_signaturez"TestSeriesRank.test_rank_signature  sQ   € Ü�A�q�6‹NˆØ	�‰�iˆÔ Ø<ˆÜ�]‰]œ:¨SÔ1ñ 	Ø�F‰F�9Ô÷	÷ 	ñ 	ús   ½AÁA!c                 ó|  — |\  }}|dk(  s|dv r|r|s|dk(  rt        j                  d«       |€|n|j                  |«      }|dv r~|s|t        |t	        j
                  |j                  t        j                  t        j                  ¬«      «      <   t	        j
                  |«      }|j                  t        «      }t        ||<   |j                  |¬«      }	|dk(  r5|s3t	        j
                  |«      }|j                  t        «      }t        ||<   t        |t        ||«      ¬	«      }
t        j                  |	|
«       y )
Nr/   )r6   r<   r1   r8   zint64/str does not support NaN©r2   r0   )r:   Úna_valuer©   r7   rQ   )rW   ÚskiprZ   r   r   r]   Úto_numpyr.   r   r-   rY   r   rA   r[   r\   )ri   r   r,   r:   Úusing_infer_stringÚusing_nan_is_nar>   rp   rn   r¥   r¦   s              r   Útest_rank_tie_methodsz$TestSeriesRank.test_rank_tie_methods  s  € ð ‰ˆ�à�WÒð ÐGÑGÙ'á&¨5°Eª>ä�K‰KÐ8Ô9à�]‰c¨¯
©
°5Ó(9ˆØÐ3Ñ3¹OäMOˆC”—‘˜Ÿ™¬B¯J©JÄÇÁ˜ÓHÓIÑJÜ—8‘8˜C“=ˆDØ—*‘*œVÓ$ˆCÜˆC�‰Ià—‘ �Ó(ˆàÐ%Ò%©oÜ—8‘8˜C“=ˆDØ—*‘*œVÓ$ˆCÜˆC�‰Iä˜#¤^°E¸6Ó%BÔCˆÜ
×Ñ˜v xÕ0r   r†   )r„   r‡   rˆ   z!dtype, na_value, pos_inf, neg_infr-   Nr.   r0   r2   r3   r4   c	                 ó¬  — t        j                  d«       |dk(  r|dk(  rd}	nd}	n|dk(  r|dk(  rd}	nd}	nd}	d}
|g|
z  |g|
z  z   |g|
z  z   }t        ||¬	«      }g d
¢g d¢g d¢fg d¢g d¢g d¢fg d¢g d¢g d¢fg d¢g d¢g d¢fg d¢g d
¢g d¢fdœ}||   }|dk(  r|d   |d   |d   g}nK|dk(  r|d   |d   |d   g}n7|dv r|s|d   t        g|
z  |d   g}n|d   t        j
                  g|
z  |d   g}|r|n|d d d…   }t        t        j                  |«      «      }|j                  |||¬«      }t        ||	¬	«      }t        j                  ||«       y )NÚscipyr2   r   r<   r0   r=   r.   r   rQ   )r   r   r   )r   r   r   )r   r   r   )r   r   r   )r   r   r   )r   r   r   )r   r   r   )r   r   r   )é	   r¶   r¶   ©r   r   r   )r   r   r   )r   r   r¶   )r   r   r   r   r    r„   r   r   r   r‡   r­   rU   )r>   r†   r‚   )rW   rX   r   r   r   r   Úlistr   Úfrom_iterablerY   r[   r\   )ri   Úrank_methodr†   r‚   r:   r®   Úpos_infÚneg_infr²   r@   ÚchunkÚin_arrrq   Ú	exp_ranksrl   Úorderr¦   r¥   Úexp_sers                      r   Ú"test_rank_tie_methods_on_infs_nansz1TestSeriesRank.test_rank_tie_methods_on_infs_nans5  s¨  € ô6 	×Ñ˜GÔ$ØÐ&Ò&Ø˜iÒ'Ø.‘	à-‘	Ø�iÒà˜iÒ'Ø%‘	à$‘	à!ˆIàˆØ�˜UÑ" h Z°%Ñ%7Ñ7¸7¸)ÀeÑ:KÑKˆÜ˜ uÔ-ˆâ!¢9ªiÐ8Úšyª)Ð4Úšyª)Ð4Ú¢ªIÐ6Ú¢ªIÐ6ñ
ˆ	ð ˜+Ñ&ˆØ˜ÒØ˜1‘X˜u Q™x¨¨q©Ð2‰EØ˜(Ò"Ø˜1‘X˜u Q™x¨¨q©Ð2‰EØÐ5Ñ5¹oØ˜1‘X¤˜t e™|¨U°1©XÐ6‰Eà˜1‘X¤§¡˜x¨%Ñ/°°q±Ð:ˆEÙ%‘5¨5±°2°©;ˆÜœ×+Ñ+¨HÓ5Ó6ˆØ—‘Ø¨)¸yð ó 
ˆô ˜¨Ô3ˆÜ
×Ñ˜v wÕ/r   c                 ó
  — t        dt        j                  t        j                  t        j                   dg«      }|j	                  d¬«      }t        dt        j                  dddgd¬	«      }t        j                  ||«       y )
Nr   é   Fr�   r   r   r   r.   rQ   )r   r   r   r_   rY   r[   r\   )ri   rq   r¥   rp   s       r   Útest_rank_desc_mix_nans_infsz+TestSeriesRank.test_rank_desc_mix_nans_infsz  sb   € ô ˜!œRŸV™V¤R§V¡V¬b¯f©f¨W°bÐ9Ó:ˆØ—‘¨�Ó.ˆÜ�aœŸ™  A qÐ)°Ô;ˆÜ
×Ñ˜v sÕ+r   z	op, valuer   g    €„.Ag�íµ ÷Æ°>c                 ó®  — t        j                  d«      }t        j                  j	                  d«      j                  d«      }t        j                  t        ddd«      D �cg c]  }||d  ‘Œ	 c}«      }t        j                  j	                  d«      j                  |«       t        t        |«      «      D �cg c]  }t        t        d«      |z   «      ‘Œ }} |||«      }t        ||¬«      }	|	j                  |¬«      }
|j                  ||dk7  r|nd	«      }t        ||¬«      j                  d
«      }t!        j"                  |
|«       y c c}w c c}w )NrE   r   r¶   r   Úa)rJ   r©   r   Úordinalr.   )rW   rX   r   re   rf   Ústandard_normalÚconcatenateÚrangeÚshufflerh   ÚchrÚordr   rY   r`   rZ   r[   r\   )ri   rº   ÚopÚvaluerk   ÚxsÚirJ   ÚvalsrI   r¥   Úsprankr¦   s                r   Útest_rank_methods_seriesz'TestSeriesRank.test_rank_methods_series‚  s  € ô ×&Ñ& }Ó5ˆä�Y‰Y×"Ñ" 1Ó%×5Ñ5°aÓ8ˆÜ�^‰^¬U°1°a¸«^Ö<¨˜R  šVÒ<Ó=ˆÜ
�	‰	×Ñ˜aÓ ×(Ñ(¨Ô,ä,1´#°b³'«NÖ; q””S˜“X ‘\Õ"Ð;ˆÐ;Ù�"�e‹}ˆÜ�D Ô&ˆØ—‘ �Ó,ˆØ×"Ñ"Ø °Ò!7‘+¸Yó
ˆô ˜&¨Ô.×5Ñ5°iÓ@ˆÜ
×Ñ˜v xÕ0ùò =ùò <s   Á"EÂ9Eúser, expr   r   r·   ©r   r   r   )r   r   r   ©r   r   r   r   r   )r   r   r   r   r   ©éûÿÿÿéüÿÿÿéýÿÿÿéþÿÿÿrU   )r   r   r   r   r   c                 ó  — |d   dk  r|j                  d«      r|d d d…   }t        |«      j                  |«      }|j                  d¬«      }t        |«      j                  t	        |d«      «      }t        j                  ||«       y )Nr   r8   rU   r    r©   ©Ú
startswithr   rZ   rY   rA   r[   r\   )ri   r:   r   rp   rª   r¥   r¦   s          r   Útest_rank_dense_methodz%TestSeriesRank.test_rank_dense_method›  sx   € ð ˆq‰6�AŠ:˜%×*Ñ*¨5Ô1Ø‘d˜�d‘)ˆCÜ�3‹K×Ñ˜uÓ%ˆØ—‘˜w�Ó'ˆÜ˜#“;×%Ñ%¤n°U¸GÓ&DÓEˆÜ
×Ñ˜v xÕ0r   c           	      ó  — |\  }}|dk(  s|dv r|r|s%|dk(  r |j                  «       j                  |«      }n|j                  |«      }|j                  d¬«      }	|j                  d«      r@|j                  d«      j	                  «       |j                  d«      z
  j                  «       }
n!|j	                  «       |z
  j                  «       }
|dk(  r&|s$|
j                  t        j                  t        «      }
t        j                  |	|
j                  t        |d«      «      «       |j                  d«      rB|j                  d«      j	                  «       |j                  d«      z
  j                  |¬	«      }
n#|j	                  «       |z
  j                  |¬	«      }
|j                  |d¬
«      }|dk(  r&|s$|
j                  t        j                  t        «      }
t        j                  ||
j                  t        ||«      «      «       y )Nr/   )r6   r1   r0   r8   Fr�   r.   r7   r   r©   )r>   r‚   )ÚdropnarZ   rY   rà   r   Úreplacer   r   r   r[   r\   rA   )ri   r   r,   r:   r±   r²   r>   Ú_rª   r™   r¦   Úres2s               r   Útest_rank_descendingz#TestSeriesRank.test_rank_descending°  s«  € ð ‰	ˆ�à�WÒØÐ?Ñ?ÉÙ&¨5°Eª>à—
‘
“×#Ñ# EÓ*‰Aà—
‘
˜5Ó!ˆAà�f‰f˜uˆfÓ%ˆØ×Ñ˜EÔ"ØŸ™ Ó+×/Ñ/Ó1°A·H±H¸YÓ4GÑG×MÑMÓO‰HàŸ™› !™×)Ñ)Ó+ˆHØÐ%Ò%©oØ×'Ñ'¬¯©´Ó3ˆHÜ
×Ñ˜s H§O¡O´NÀ5È)Ó4TÓ$UÔVà×Ñ˜EÔ"ØŸ™ Ó+×/Ñ/Ó1°A·H±H¸YÓ4GÑG×MÑMØð Nó ‰Hð Ÿ™› !™×)Ñ)°Ð)Ó8ˆHØ�v‰v˜V¨uˆvÓ5ˆØÐ%Ò%©oØ×'Ñ'¬¯©´Ó3ˆHÜ
×Ñ˜t X§_¡_´^ÀEÈ6Ó5RÓ%SÕTr   c                 óð   — |\  }}|j                  «       j                  d«      }|j                  |¬«      }t        |«      j                  «       }|j                  |_        t        j                  ||«       y )NÚi8r©   )rã   rZ   rY   r   rJ   r[   r\   )ri   r   r,   r>   rp   rª   r¥   r¦   s           r   Útest_rank_intzTestSeriesRank.test_rank_intÑ  s^   € Ø‰ˆ�Ø�J‰J‹L×Ñ Ó%ˆà—‘˜v�Ó&ˆÜ˜#“;×%Ñ%Ó'ˆØŸ™ˆŒÜ
×Ñ˜v xÕ0r   c                 ó  — t        t        j                  gdz  «      j                  t        «      j                  d¬«       t        t        j                  gdz  «      j                  t        «      j                  d¬«       y )Né    Tr�   F)r   r   r   rZ   r-   rY   )ri   s    r   Útest_rank_object_bugz#TestSeriesRank.test_rank_object_bugÚ  sZ   € ô 	”—‘ˆx˜"‰}Ó×$Ñ$¤VÓ,×1Ñ1¸DÐ1ÔAÜ”—‘ˆx˜"‰}Ó×$Ñ$¤VÓ,×1Ñ1¸EÐ1ÕBr   c                 ó¨   — t        t        d«      t        g«      }|j                  «       }|j	                  «        |}t        j                  ||«       y )Nz2017-01-05 10:20:27.569000)r   r	   r   ÚcopyrY   r[   r\   )ri   rª   r¦   r¥   s       r   Útest_rank_modify_inplacez'TestSeriesRank.test_rank_modify_inplaceá  sB   € ô ”IÐ:Ó;¼SÐAÓBˆØ—6‘6“8ˆà	�‰ŒØˆÜ
×Ñ˜v xÕ0r   c                 ó¬   — t        dddt        dgd¬«      }|j                  d¬«      }t        d	d
dt        dgd¬«      }t        j                  ||«       y )Ng¶*"¨¾Fg¦3Ø!äÇ»g{èQÁí ½:g1AÙ;Æ-:r0   rQ   r   r©   r   r   r   r   r=   )r   r   rY   r[   r\   )ri   r   r¥   r¦   s       r   Útest_rank_ea_small_valuesz(TestSeriesRank.test_rank_ea_small_valuesë  sV   € äØ˜=¨-¼¸]ÐKØô
ˆð —‘ �Ó'ˆÜ˜1˜a ¤B¨Ð*°(Ô;ˆÜ
×Ñ˜v xÕ0r   ) Ú__name__Ú
__module__Ú__qualname__rv   r¢   r§   r«   r³   rW   ÚmarkÚparametrizer   r   r   r   r_   r   r)   ÚtdÚ
skip_if_norÂ   rÅ   ÚoperatorÚaddÚmulrÕ   rá   rç   rê   rí   rð   rò   r   r   r   rC   rC   P   sÆ  „ òY,òvVRòp1òò1ðB ‡[�[×Ñ˜[Ò*CÓDØ‡[�[×ÑØ+à�t™X›Z©«Ð7Ø˜Ÿ™ §¡¨¯©¨Ð0Ø˜˜BŸF™F R§V¡V GÐ,ØˆF�L‰LØ"ØØ—‘Ø—‘�Ø#�b—m‘m IÓ.ôð		
óñ30óó Eð 30òj,ð ‡[�[×ÑØà�\‰\˜1ÐØ�\‰\˜3ÐØ�\‰\˜4Ð ð	
óñ1óð1ð" ‡[�[×ÑØàˆS�1�#ˆJØˆS�1�#ˆJØˆS�1�#ˆJØ�ˆV�a˜�VÐÚš	Ð"Úš	Ð"ÚšoÐ.Ú!¢?Ð3ð		
óñ1óð1òUòB1òCò1ó1r   rC   rÖ   rL   r’   r·   )çUUUUUUÕ?çUUUUUUå?rL   )r   r   r   )r’   rL   rL   r×   )rL   rþ   rý   rØ   )rý   rý   rL   rL   rþ   )r   r   r   r   r   r   )rý   rý   rþ   rþ   rL   rL   rÙ   r�   c                 ó  — |d   dk  r| j                  d«      r|d d d…   }t        |«      j                  | «      }|j                  dd¬«      }t        |«      j                  t	        | dd¬«      «      }t        j                  ||«       y )Nr   r8   rU   r    T©r>   r?   rN   rß   ©r:   r   rp   rª   r¥   r¦   s         r   Útest_rank_dense_pctr  ù  ó|   € ð ˆ1�v�‚z�e×&Ñ& uÔ-Ø‘$�B�$‰iˆÜˆs‹×Ñ˜5Ó!€AØ�V‰V˜7¨ˆVÓ-€FÜ�c‹{×!Ñ!¤.°¸ÀTÔ"JÓK€HÜ×Ñ˜6 8Õ,r   )rý   rþ   rþ   )r�   r�   rŽ   rŽ   r�   )çUUUUUUÅ?r  r’   r’   ç«ªªªªªê?r  c                 ó  — |d   dk  r| j                  d«      r|d d d…   }t        |«      j                  | «      }|j                  dd¬«      }t        |«      j                  t	        | dd¬«      «      }t        j                  ||«       y )Nr   r8   rU   r   Tr   rN   rß   r  s         r   Útest_rank_min_pctr    ó|   € ð ˆ1�v�‚z�e×&Ñ& uÔ-Ø‘$�B�$‰iˆÜˆs‹×Ñ˜5Ó!€AØ�V‰V˜5 dˆVÓ+€FÜ�c‹{×!Ñ!¤.°¸À4Ô"HÓI€HÜ×Ñ˜6 8Õ,r   )rý   rL   rL   )rŒ   rŒ   rL   rL   r�   c                 ó  — |d   dk  r| j                  d«      r|d d d…   }t        |«      j                  | «      }|j                  dd¬«      }t        |«      j                  t	        | dd¬«      «      }t        j                  ||«       y )Nr   r8   rU   r   Tr   rN   rß   r  s         r   Útest_rank_max_pctr
  '  r  r   r“   )rý   r  r  )ç333333Ó?r  çÍÌÌÌÌÌì?r  r�   )r‘   r‘   ç«ªªªªªâ?r  çUUUUUUí?r  c                 ó  — |d   dk  r| j                  d«      r|d d d…   }t        |«      j                  | «      }|j                  dd¬«      }t        |«      j                  t	        | dd¬«      «      }t        j                  ||«       y )Nr   r8   rU   r   Tr   rN   rß   r  s         r   Útest_rank_average_pctr  >  s|   € ð ˆ1�v�‚z�e×&Ñ& uÔ-Ø‘$�B�$‰iˆÜˆs‹×Ñ˜5Ó!€AØ�V‰V˜9¨$ˆVÓ/€FÜ�c‹{×!Ñ!¤.°¸	ÀtÔ"LÓM€HÜ×Ñ˜6 8Õ,r   )r�   rŒ   rŽ   rL   r�   )r  rý   r’   rþ   r  rL   c                 ó  — |d   dk  r| j                  d«      r|d d d…   }t        |«      j                  | «      }|j                  dd¬«      }t        |«      j                  t	        | dd¬«      «      }t        j                  ||«       y )Nr   r8   rU   r   Tr   rN   rß   r  s         r   Útest_rank_first_pctr  U  r  r   c                  óŽ   — t        t        j                  d«      «      } | j                  d¬«      j	                  «       }|dk(  sJ ‚y )Ni  TrN   r   )r   r   ra   rY   r   )rª   r¥   s     r   Útest_pct_max_many_rowsr  l  s<   € ô 	Œr�y‰y˜Ó#Ó$€AØ�V‰V˜ˆVÓ×!Ñ!Ó#€FØ�QŠ;Ð‰;r   )F)+Ú	itertoolsr   rú   Únumpyr   rW   Úpandas._libs.algosr   r   Úpandas.util._test_decoratorsÚutilÚ_test_decoratorsrø   Úpandasr   r   r   r	   r
   Úpandas._testingÚ_testingr[   Úpandas.api.typesr   Úfixturer   rd   r   r,   r)   rù   r:   rA   rC   rö   r÷   r  r  r
  r  r  Ú
single_cpur  r   r   r   ú<module>r!     sã  ðÝ Û ã Û ÷÷ *Ð )÷õ õ Ý -ð ‡�ñ<ó ð<ð €‡�à	�H�B—H‘H˜c 3¨¨S°"·&±&¸#¸sÀCÈÏÉÐQTÐUÓVÐWØ	��—‘˜!˜Q  1 b§f¡f¨a°°A°r·v±v¸qÐAÓBÐCØ	��—‘˜!˜Q  1 b§f¡f¨a°°A°r·v±v¸qÐAÓBÐCØ	�(�"—(‘(˜A˜q ! Q¨¯©°°1°a¸¿¹ÀÐCÓDÐEØ	�(�"—(‘(˜A˜q ! Q¨¯©°°1°a¸¿¹ÀÐCÓDÐEðñ 	ô	ñó	ðð €‡�àØØØØØˆ�‰Ð'¨}¨r¯}©}¸YÓ/GÔHØˆ�‰Ð%¨]¨R¯]©]¸9Ó-EÔFØˆ�‰Ð&¨m¨b¯m©m¸IÓ.FÔGØØðôñóðó÷&c1ñ c1ðR ‡�×ÑØà
ˆˆsˆeˆØ
ˆQˆ�'˜7Ð#Ð$Ø
ˆQˆ�#�s�ÐÚ	Ò/Ð0Ú	Ò/Ð0Ú	Ò/Ð0Ú	ÒGÐHÚ	ÒSÐTÚ	ÒLÐMð
óñ-óð-ð ‡�×ÑØà
ˆˆsˆeˆØ
ˆQˆ�'˜7Ð#Ð$Ø
ˆQˆ�'˜7Ð#Ð$Ú	Ò/Ð0Ú	Ò/Ð0Ú	Ò/Ð0Ú	ÒGÐHÚ	ÒSÐTÚ	ÒLÐMð
óñ-óð-ð ‡�×ÑØà
ˆˆsˆeˆØ
ˆQˆ�'˜7Ð#Ð$Ø
ˆQˆ�#�s�ÐÚ	Ò/Ð0Ú	Ò/Ð0Ú	Ò/Ð0Ú	ÒGÐHÚ	ÒSÐTÚ	ÒLÐMð
óñ-óð-ð ‡�×ÑØà
ˆˆsˆeˆØ
ˆQˆ�'˜7Ð#Ð$Ø
ˆQˆ�'˜7Ð#Ð$Ú	Ò/Ð0Ú	Ò/Ð0Ú	Ò/Ð0Ú	ÒGÐHÚ	ÒSÐTÚ	ÒLÐMð
óñ-óð-ð ‡�×ÑØà
ˆˆsˆeˆØ
ˆQˆ�'˜7Ð#Ð$Ø
ˆQˆ�'˜9Ð%Ð&Ú	Ò/Ð0Ú	Ò/Ð0Ú	Ò/Ð0Ú	ÒGÐHÚ	ÒSÐTÚ	ÒLÐMð
óñ-óð-ð ‡�×Ññó ñr   