Ë
    ž…j T  ã            *       óH  — d dl m Z  d dlZd dlZd dlZd dlmZmZmZm	Z	 d dl
mZ d„ Zd„ Zej                  j!                  ddgddgg«      ej                  j!                  d	d
D � cg c]  }  ej"                  g d¢| ¬«      ‘Œ c}  ej$                  d«       ej$                  d«       ej$                  d«       ej$                  d«       ej$                  d«      g ej$                  dd¬«       ej$                  dd¬«       ej$                  dd¬«       ej$                  dd¬«       ej$                  dd¬«      g ej$                  d«       ej$                  d «      z
   ej$                  d«       ej$                  d «      z
   ej$                  d«       ej$                  d «      z
   ej$                  d«       ej$                  d «      z
   ej$                  d«       ej$                  d «      z
  g ej$                  d«      j'                  d«       ej$                  d«      j'                  d«       ej$                  d«      j'                  d«       ej$                  d«      j'                  d«       ej$                  d«      j'                  d«      ggz   d„ ¬«      ej                  j!                  ddddg d¢fdddg d¢fdddg d¢fdddg d¢fdddg d¢fdddg d¢fdddg d ¢fdddg d!¢fd"ddg d#¢fd"ddg d$¢fd"ddg d%¢fd"ddg d&¢fd'ddg d(¢fd'ddg d)¢fd'ddg d*¢fd'ddg d+¢fd,ddg d-¢fd,ddg d.¢fd,ddg d ¢fd,ddg d/¢fg«      d0„ «       «       «       Zej                  j!                  ddgddgg«      ej                  j!                  d	ej*                   ej*                   ej,                  d1ej,                  ej*                  ej*                  gg«      ej                  j!                  d2ddd3d4d4ej,                  d5ej,                  d6d6gfddd7g d8¢fddd9g d:¢fddd3d6d6ej,                  d5ej,                  d4d4gfddd7g d;¢fddd9g d<¢fddd3d1d1ej,                  d=ej,                  d>d>gfddd7g d?¢fddd9g d@¢fddd3d>d>ej,                  d=ej,                  d1d1gfddd7g dA¢fddd9g dB¢fd"dd3dCdCej,                  d=ej,                  dDdDgfd"dd7g dE¢fd"dd9g dF¢fd"dd3dDdDej,                  d=ej,                  dCdCgfd"dd7g dG¢fd"dd9g dH¢fd'dd3d1dCej,                  d=ej,                  d>dDgfd'dd7g dI¢fd'dd9g dJ¢fd'dd3d>dDej,                  d=ej,                  d1dCgfd'dd7g dK¢fd'dd9g dL¢fd,dd3d1d1ej,                  dCej,                  d=d=gfd,dd7g dM¢fd,dd9g dN¢fd,dd3d=d=ej,                  dCej,                  d1d1gfd,dd7g dO¢fd,dd9g dP¢fg«      dQ„ «       «       «       Zej                  j!                  ddgddgg«      ej                  j!                  d	dRD � cg c]<  }  ej"                  dSdSej,                  dTdSdUej,                  ej,                  g| ¬«      ‘Œ> c}  ej$                  d«       ej$                  d«      ej,                   ej$                  d«       ej$                  d«       ej$                  d«      ej,                  ej,                  g ej$                  dd¬«       ej$                  dd¬«      ej,                   ej$                  dd¬«       ej$                  dd¬«       ej$                  dd¬«      ej,                  ej,                  g ej$                  d«       ej$                  d «      z
   ej$                  d«       ej$                  d «      z
  ej,                   ej$                  d«       ej$                  d «      z
   ej$                  d«       ej$                  d «      z
   ej$                  d«       ej$                  d «      z
  ej,                  ej,                  g ej$                  d«      j'                  d«       ej$                  d«      j'                  d«      ej,                   ej$                  d«      j'                  d«       ej$                  d«      j'                  d«       ej$                  d«      j'                  d«      ej,                  ej,                  ggz   dV„ ¬«      ej                  j!                  dWg ddd3ddCdCej,                  dDdCd>ej,                  ej,                  gf‘ddd3ddXdXej,                  d1dXdYej,                  ej,                  gf‘ddd3dd>d>ej,                  d1d>dCej,                  ej,                  gf‘ddd3ddYdYej,                  dZdYdXej,                  ej,                  gf‘ddd3dd1d1ej,                  dDd1d>ej,                  ej,                  gf‘ddd3ddZdZej,                  d1dZdYej,                  ej,                  gf‘ddd3dd=d=ej,                  d1d=dCej,                  ej,                  gf‘ddd3dd[d[ej,                  dZd[dXej,                  ej,                  gf‘d"dd3dd=d=ej,                  dDd=d>ej,                  ej,                  gf‘d"dd3dd[d[ej,                  d1d[dYej,                  ej,                  gf‘d"dd3ddDdDej,                  d1dDdCej,                  ej,                  gf‘d"dd3dd1d1ej,                  dZd1dXej,                  ej,                  gf‘d'dd3dd1dCej,                  dDd=d>ej,                  ej,                  gf‘d'dd3ddZdXej,                  d1d[dYej,                  ej,                  gf‘d'dd3dd=d>ej,                  d1dDdCej,                  ej,                  gf‘d'dd3dd[dYej,                  dZd1dXej,                  ej,                  gf‘d,dd3dd1d1ej,                  d=d1dCej,                  ej,                  gf‘d,dd3dd\d\ej,                  d1d\d]ej,                  ej,                  gf‘d,dd3dd=d=ej,                  d1d=dCej,                  ej,                  gf‘d,dd3dd1d1ej,                  d\d1d]ej,                  ej,                  gf‘ddd9dg d^¢f‘ddd9dg d_¢f‘ddd9dg d`¢f‘ddd9dg da¢f‘ddd9dg db¢f‘ddd9dg dc¢f‘ddd9dg dd¢f‘ddd9dg de¢f‘d"dd9dg df¢f‘d"dd9dg dg¢f‘d"dd9dg dh¢f‘d"dd9dg di¢f‘d'dd9dg dj¢f‘d'dd9dg dk¢f‘d'dd9dg dl¢f‘d'dd9dg dm¢f‘d,dd9dg dn¢f‘d,dd9dg do¢f‘d,dd9dg dp¢f‘d,dd9dg dq¢f‘«      dr„ «       «       «       Zej                  j!                  dsdg dt¢fdg du¢fg«      dv„ «       Zej                  j!                  dwg dx¢«      ej                  j!                  dyddg«      dz„ «       «       Zej                  j!                  d{g d|¢«      ej                  j!                  d}ddg«      ej                  j!                  d	g d~¢dej,                  d€ej,                  d�gg«      d‚„ «       «       «       Zej                  j!                  d{g dƒ¢«      ej                  j!                  d}ddg«      ej                  j!                  d	g d~¢dej,                  d€ej,                  d�gd„ej,                  dSej,                  d5gg«      d…„ «       «       «       Zd†„ Zej                  j!                  d‡d„dSgd„d„gd1d1gfg dˆ¢g d‰¢g dŠ¢fg dˆ¢d„dSd„ej,                  gd‹d1d1ej,                  gfg dŒ¢d„dSej,                  gd‹d1ej,                  gfg«      d�„ «       ZdŽ„ Zej                  j!                  d�ddg«      d�„ «       Z d‘„ Z!d’„ Z"ej                  j!                  d{d7d9g«      d“„ «       Z#yc c} w c c} w )”é    )ÚdatetimeN)Ú	DataFrameÚNaTÚSeriesÚconcatc                  ó¸  — t        j                  g d¬«      } t        | «      }|j                  «       }d}|j	                  | d¬«      }t        j                  t        |¬«      5  |j                  «        d d d «       |j	                  | d¬«      }t        j                  t        |¬«      5  |j                  «        d d d «       y # 1 sw Y   ŒQxY w# 1 sw Y   y xY w)NF©Úorderedz0Cannot perform rank with non-ordered Categorical)Úobserved©Úmatch)	ÚpdÚCategoricalr   Úto_frameÚgroupbyÚpytestÚraisesÚ	TypeErrorÚrank)ÚcatÚserÚdfÚmsgÚgbÚgb2s         úg/root/aria/tools/markitdown-venv/lib/python3.12/site-packages/pandas/tests/groupby/methods/test_rank.pyÚ)test_rank_unordered_categorical_typeerrorr      s®   € ä
�.‰.˜ UÔ
+€CÜ
�‹+€CØ	�‰‹€Bà
<€Cà	�‰�S 5ˆÓ	)€BÜ	�‰”y¨Ô	,ñ Ø
�‰Œ	÷ð �*‰*�S 5ˆ*Ó
)€CÜ	�‰”y¨Ô	,ñ Ø�‰Œ
÷ð ÷	ð ú÷ð ús   Á#CÂ*CÃCÃCc                  óÂ  — t        j                  dgdz  t        ¬«      } t        j                  dgdz  t        ¬«      }t         j                  j	                  d«      j                  dddt        ¬	«      }t         j                  j	                  d«      j                  dddt        ¬	«      }t        t         j                  j	                  d«      j                  d«      | j                  |«      |j                  |«      d
œ«      }|j                  ddg«      j                  j                  «       }|j                  ddg«      D ��cg c]  \  }}|j                  j                  «       ‘Œ! }}}t        |d¬«      }|j                  |j                  «      }t!        j"                  ||«       |j                  ddg«      j                  j                  d¬«      }|j                  ddg«      D ��cg c]!  \  }}|j                  j                  d¬«      ‘Œ# }}}t        |d¬«      }|j                  |j                  «      }t!        j"                  ||«       y c c}}w c c}}w )NÚ
aaaaaaaaaaéd   ©ÚdtypeÚ
bbbbbbbbbbé‚   é   r   iô  )Úsizer"   )ÚvalueÚkey1Úkey2r(   r)   )ÚaxisT©Úpct)ÚnpÚarrayÚobjectÚrandomÚdefault_rngÚintegersÚintr   Ústandard_normalÚtaker   r'   r   r   ÚreindexÚindexÚtmÚassert_series_equal)	Úlev1Úlev2Úlab1Úlab2r   ÚresultÚkeyÚpieceÚexpecteds	            r   Útest_rank_applyrB   !   sì  € Ü�8‰8�X�J Ñ$¬FÔ3€DÜ�8‰8�X�J Ñ$¬FÔ3€DÜ�9‰9× Ñ  Ó#×,Ñ,¨Q°¸#ÄSÐ,ÓI€DÜ�9‰9× Ñ  Ó#×,Ñ,¨Q°¸#ÄSÐ,ÓI€Dä	ä—Y‘Y×*Ñ*¨1Ó-×=Ñ=¸cÓBØ—I‘I˜d“OØ—I‘I˜d“Oñ	
ó
€Bð �Z‰Z˜ Ð(Ó)×/Ñ/×4Ñ4Ó6€Fà57·Z±ZÀÈÐ@PÓ5Q×R¡z s¨E�—‘× Ñ Õ"ÐR€HÑRÜ�h QÔ'€HØ×Ñ §¡Ó-€HÜ×Ñ˜6 8Ô,à�Z‰Z˜ Ð(Ó)×/Ñ/×4Ñ4¸Ð4Ó>€Fð 68·Z±ZÀÈÐ@PÓ5Q÷Ù'1 s¨Eˆ�‰×Ñ˜TÐÕ"ð€Hñ ô �h QÔ'€HØ×Ñ §¡Ó-€HÜ×Ñ˜6 8Õ,ùó Sùós   Å$IÇ-&IÚgrpsÚquxÚquuxÚvals)Úi8Úi4Úi2Úi1Úu8Úu4Úu2Úu1Úf8Úf4Úf2)r%   r%   é   r%   é   r!   z
2018-01-02z
2018-01-08z
2018-01-06z
US/Pacific©ÚtzÚDc                 ó   — t        | d   «      S ©Nr   ©Útype©Úxs    r   ú<lambda>r]   e   ó   € ”$�q˜‘t“*€ ó    )Úidszties_method,ascending,pct,expÚaverageTF)ç       @rb   ç      @rb   ç      @)çš™™™™™Ù?re   ç      ð?re   çš™™™™™é?)rd   rd   rf   rd   rb   )rg   rg   çš™™™™™É?rg   re   Úmin)rf   rf   rc   rf   rd   )rh   rh   rf   rh   rg   )ç      @rj   rf   rj   rb   )ç333333ã?rk   rh   rk   re   Úmax)rj   rj   rc   rj   rd   )rk   rk   rf   rk   rg   )rc   rc   rf   rc   rb   )rf   rf   rh   rf   re   Úfirst)rf   rb   rc   rj   rd   )rh   re   rf   rk   rg   )rj   rd   rf   rc   rb   )rk   rg   rh   rf   re   Údense)rf   rf   rj   rf   rb   )çUUUUUUÕ?ro   rf   ro   çUUUUUUå?)rf   rf   ro   rf   rp   c                 ó¬  — t        j                  | t        |«      «      }|}t        |«      t        | «      z  }t	        |t         j
                  «      r!t        j                  ||j                  ¬«      }t        ||dœ«      }|j                  d«      j                  |||¬«      }	t        |t        | «      z  dg¬«      }
t        j                  |	|
«       y )Nr!   ©r?   Úvalr?   )ÚmethodÚ	ascendingr,   rs   ©Úcolumns©r-   ÚrepeatÚlenÚlistÚ
isinstanceÚndarrayr.   r"   r   r   r   r8   Úassert_frame_equal)rC   rF   Úties_methodru   r,   Úexpr?   Ú	orig_valsr   r>   Úexp_dfs              r   Útest_rank_argsrƒ   @   s¨   € ôB �)‰)�Dœ#˜d›)Ó
$€Cà€IÜ�‹:œ˜D›	Ñ!€DÜ�)œRŸZ™ZÔ(Ü�x‰x˜ I§O¡OÔ4ˆä	˜3 tÑ,Ó	-€BØ�Z‰Z˜Ó×#Ñ#¨;À)ÐQTÐ#ÓU€Fä�sœS ›Y‘°°Ô8€FÜ×Ñ˜& &Õ)r_   rf   z#ties_method,ascending,na_option,expÚkeepç      ø?é   ç      @Útop)ç      @r‰   r…   rc   r…   ç      @rŠ   Úbottom)r…   r…   rŠ   rj   rŠ   r‡   r‡   )rŠ   rŠ   r…   rc   r…   r‰   r‰   )r‡   r‡   rŠ   rj   rŠ   r…   r…   rj   rd   )rj   rj   rf   rc   rf   ç      @rŒ   )rf   rf   rŒ   rj   rŒ   rd   rd   )rŒ   rŒ   rf   rc   rf   rj   rj   )rd   rd   rŒ   rj   rŒ   rf   rf   rb   rc   )rd   rd   rb   rc   rb   ç      @r�   )rb   rb   r�   rj   r�   rc   rc   )r�   r�   rb   rc   rb   rd   rd   )rc   rc   r�   rj   r�   rb   rb   )rj   rd   rf   rc   rb   rŒ   r�   )rf   rb   rŒ   rj   r�   rd   rc   )rŒ   r�   rf   rc   rb   rj   rd   )rd   rc   rŒ   rj   r�   rf   rb   )rb   rb   rf   rj   rf   rd   rd   )rf   rf   rd   rb   rd   rj   rj   )rd   rd   rf   rj   rf   rb   rb   )rj   rj   rd   rb   rd   rf   rf   c                 ó   — t        j                  | t        |«      «      }|t        | «      z  }t        ||dœ«      }|j	                  d«      j                  |||¬«      }t        |t        | «      z  dg¬«      }	t        j                  ||	«       y )Nrr   r?   )rt   ru   Ú	na_optionrs   rv   )r-   ry   rz   r   r   r   r8   r~   )
rC   rF   r   ru   r�   r€   r?   r   r>   r‚   s
             r   Útest_infs_n_nansr�   �   sƒ   € ôR �)‰)�Dœ#˜d›)Ó
$€CØ”#�d“)Ñ€DÜ	˜3 tÑ,Ó	-€BØ�Z‰Z˜Ó×#Ñ#Ø i¸9ð $ó €Fô �sœS ›Y‘°°Ô8€FÜ×Ñ˜& &Õ)r_   )rO   rP   rQ   r%   rR   rS   c                 ó   — t        | d   «      S rX   rY   r[   s    r   r]   r]   ó   r^   r_   z'ties_method,ascending,na_option,pct,expre   rg   rh   rk   ro   rp   )rb   rb   r�   rc   rb   rd   r�   r�   )ç      Ð?r’   ç      ì?ç      ä?r’   ç      à?r“   r“   )rd   rd   r�   rf   rd   rb   r�   r�   )r•   r•   r“   ç      À?r•   r’   r“   r“   )rf   rf   rŒ   rc   rf   rd   rŒ   rŒ   )r–   r–   ç      è?r”   r–   r•   r—   r—   )rj   rj   rŒ   rf   rj   rb   rŒ   rŒ   )ç      Ø?r˜   r—   r–   r˜   r’   r—   r—   )rj   rj   ç       @rc   rj   rd   r™   r™   )r˜   r˜   rf   r”   r˜   r•   rf   rf   )rc   rc   r™   rf   rc   rb   r™   r™   )r”   r”   rf   r–   r”   r’   rf   rf   )rf   rb   rŒ   rc   rj   rd   r�   r™   )r–   r’   r—   r”   r˜   r•   r“   rf   )rj   rd   rŒ   rf   rc   rb   r�   r™   )r˜   r•   r—   r–   r”   r’   r“   rf   )rf   rf   rd   rj   rf   rb   rd   rd   )r’   r’   rf   r—   r’   r•   rf   rf   )rj   rj   rd   rf   rj   rb   rd   rd   )r—   r—   rf   r’   r—   r•   rf   rf   c                 ó®  — t        j                  | t        |«      «      }|}t        |«      t        | «      z  }t	        |t         j
                  «      r!t        j                  ||j                  ¬«      }t        ||dœ«      }	|	j                  d«      j                  ||||¬«      }
t        |t        | «      z  dg¬«      }t        j                  |
|«       y )Nr!   rr   r?   ©rt   ru   r�   r,   rs   rv   rx   )rC   rF   r   ru   r�   r,   r€   r?   r�   r   r>   r‚   s               r   Útest_rank_args_missingrœ   Â   s°   € ôb �)‰)�Dœ#˜d›)Ó
$€Cà€IÜ�‹:œ˜D›	Ñ!€DÜ�)œRŸZ™ZÔ(Ü�x‰x˜ I§O¡OÔ4ˆä	˜3 tÑ,Ó	-€BØ�Z‰Z˜Ó×#Ñ#Ø i¸9È#ð $ó €Fô �sœS ›Y‘°°Ô8€FÜ×Ñ˜& &Õ)r_   zpct,exp)rj   rj   rj   rj   rj   )rk   rk   rk   rk   rk   c                 ó¼   — t        g d¢dgdz  dœ«      }|j                  d«      j                  | ¬«      }t        |dz  dg¬	«      }t        j                  ||«       y )
N)
Úarž   rž   rž   rž   ÚbrŸ   rŸ   rŸ   rŸ   é   é
   rr   r?   r+   r%   rs   rv   ©r   r   r   r8   r~   )r,   r€   r   r>   r‚   s        r   Útest_rank_resets_each_groupr£   Ã  s]   € ô 
ÚBÈAÈ3ÐQSÉ8ÑTó
€Bð �Z‰Z˜Ó×#Ñ#¨Ð#Ó,€FÜ�s˜Q‘w¨¨Ô0€FÜ×Ñ˜& &Õ)r_   r"   )Úint64Úint32Úuint64Úuint32Úfloat64Úfloat32Úupperc                 ó�  — |r+| d   j                  «       | dd  z   } | j                  dd«      } t        dgdz  dgdz  dœ«      }|d   j                  | «      |d<   |d   j                  | k(  sJ ‚|j                  d	«      j                  «       }t        g d
¢dg¬«      }|r|j                  d«      }t        j                  ||«       y )Nr   r    ÚUiÚUIrž   é   rr   rs   r?   )ç      @r¯   r¯   r¯   rv   ÚFloat64)	rª   Úreplacer   Úastyper"   r   r   r8   r~   )r"   rª   r   r>   r‚   s        r   Útest_rank_avg_even_valsr³   Ï  sÂ   € ñ
 à�a‘—‘Ó  5¨¨ 9Ñ,ˆØ—‘˜d DÓ)ˆÜ	˜C˜5 1™9¨a¨S°1©WÑ5Ó	6€BØ�5‘	× Ñ  Ó'€B€u�IØˆe‰9�?‰?˜eÒ#Ð#Ð#à�Z‰Z˜Ó×#Ñ#Ó%€FÜÒ+°e°WÔ=€FÙØ—‘˜yÓ)ˆÜ×Ñ˜& &Õ)r_   r�   )r„   rˆ   r‹   r,   )Úbarr´   Úfoor´   Úbazr´   rµ   r¶   c                 óÂ  — t        dgdz  |dœ«      }|d   j                  «       }|j                  d«      }|j                  | |||¬«      }|j	                  «       r4t        dgdz  dt
        j                  dt
        j                  d	gdœ«      }	nt        dgdz  g d
¢dœ«      }	|	j                  d«      }
|
j                  | |||¬«      }t        j                  ||«       y )Nrµ   é   rr   rs   r?   r›   r   r%   r    )r   r   r%   r   r    )	r   Úisnar   r   Úanyr-   Únanr8   r~   )Úrank_methodru   r�   r,   rF   r   Úmaskr   ÚresÚdf2r   Úalts               r   Útest_rank_object_dtyperÁ   ã  sÒ   € ô 
˜E˜7 Q™;¨tÑ4Ó	5€BØˆe‰9�>‰>Ó€Dà	�‰�EÓ	€BØ
�'‰'˜°	ÀYÐTWˆ'Ó
X€Cð ‡x�x„zÜ  ¨!¡°Q¼¿¹ÀÄ2Ç6Á6È1Ð4MÑNÓO‰ä  ¨!¡²OÑDÓEˆà
�+‰+�eÓ
€CØ
�(‰(Ø i¸9È#ð ó €Cô ×Ñ˜#˜sÕ#r_   )TÚbadr    r    c                 óÔ   — t        dgdz  |dœ«      }d}t        j                  t        |¬«      5  |j	                  d«      j                  | |||¬«       d d d «       y # 1 sw Y   y xY w)Nrµ   r¸   rr   z3na_option must be one of 'keep', 'top', or 'bottom'r   r?   r›   )r   r   r   Ú
ValueErrorr   r   )r¼   ru   r�   r,   rF   r   r   s          r   Útest_rank_naoption_raisesrÅ   ý  sf   € ô 
˜E˜7 Q™;¨tÑ4Ó	5€BØ
?€Cä	�‰”z¨Ô	-ñ 
Ø
�
‰
�5Ó×ÑØ¨)¸yÈcð 	ô 	
÷
÷ 
ñ 
ús   °%AÁA'c                  ó²  — d} t        g d¢dt        j                  dgdœ«      }|j                  | «      j                  j                  d¬«      }t        dt        j                  dgd	¬
«      }t        j                  ||«       |j                  | «      j                  d¬«      }t        d	dt        j                  dgi«      }t        j                  ||«       y )NÚA)r   r    r   rf   rb   ©rÇ   ÚBTr+   r•   rÉ   ©Úname)
r   r-   r»   r   rÉ   r   r   r8   r9   r~   )Úcolumnr   r>   rA   s       r   Útest_rank_empty_grouprÍ     s°   € à€FÜ	š¨#¬r¯v©v°sÐ);Ñ<Ó	=€Bà�Z‰Z˜Ó×!Ñ!×&Ñ&¨4Ð&Ó0€FÜ�sœBŸF™F CÐ(¨sÔ3€HÜ×Ñ˜6 8Ô,à�Z‰Z˜Ó×$Ñ$¨Ð$Ó.€FÜ˜# ¤R§V¡V¨SÐ1Ð2Ó3€HÜ×Ñ˜& (Õ+r_   z"input_key,input_value,output_value)r    r    r%   r%   )r    r%   r    r%   )r•   rf   r•   rf   r•   ©r    r    r%   c                 óª   — t        | |dœ«      }|j                  d«      j                  dd¬«      }t        d|i«      }t        j                  ||«       y )NrÈ   rÇ   rn   T©rt   r,   rÉ   r¢   )Ú	input_keyÚinput_valueÚoutput_valuer   r>   rA   s         r   Útest_rank_zero_divrÔ     sO   € ô 
˜¨Ñ5Ó	6€Bà�Z‰Z˜‹_×!Ñ!¨°dÐ!Ó;€FÜ˜#˜|Ð,Ó-€HÜ×Ñ˜& (Õ+r_   c            
      ó"  — t        g d¢t        j                  t        j                  «      j                  t        j                  t        j                  «      j
                  t        j                  t        j                  «      j                  gt        t        ddd«      t        gdœ«      } | j                  d«      j                  «       }t        g d¢t        j                  dt        j                  gdœ«      }t        j                  ||«       y )	NrÎ   iÑ  r    )ÚgrpÚint_colÚdatetimelikerÖ   )rf   rb   rf   rf   )r×   rØ   )r   r-   Úiinfor¤   ri   rl   r   r   r   r   r»   r8   r~   ©r   r>   rA   s      r   Útest_rank_min_intrÛ   1  s¼   € ä	âä—‘œŸ™Ó"×&Ñ&Ü—‘œŸ™Ó"×&Ñ&Ü—‘œŸ™Ó"×&Ñ&ðô
 !¤(¨4°°AÓ"6¼Ð<ñ	
ó

€Bð �Z‰Z˜Ó×#Ñ#Ó%€FÜÚ#´b·f±f¸cÄ2Ç6Á6Ð5JÑKó€Hô ×Ñ˜& (Õ+r_   Úuse_nanc                 óX  — | rt         j                  nd}t        ddgddgd|gd|ggddg¬«      }|j                  dg«      d   j	                  dd	¬
«      }| r.t        ddt         j                  t         j                  gd¬«      }nt        g d¢d¬«      }t        j                  ||«       y )Nr†   éÿÿÿÿr    r%   Úgrouprs   rv   rn   TrÐ   r•   rÊ   )ro   rp   r    r    )r-   r»   r   r   r   r   r8   r9   )rÜ   Ú
fill_valuer   r>   rA   s        r   Ú.test_rank_pct_equal_values_on_group_transitionrá   G  s³   € ñ #”—’¨€JÜ	à�ˆGØ�ˆGØ�
ˆOØ�Ðð		
ð ˜%Ð ô
€Bð �Z‰Z˜˜	Ó" 5Ñ)×.Ñ.ØØð /ó €Fñ Ü˜3 ¤2§6¡6¬2¯6©6Ð2¸Ô?‰äÒ.°UÔ;ˆä×Ñ˜6 8Õ,r_   c                  óˆ  — t        dddt        j                  gddœt        j                  dd¬«      gdz  ¬	«      } | j                  | j                  d
g«      j                  j                  dd¬«      }t        dddt        j                  gt        j                  dd¬«      gdz  d¬«      }t        j                  ||«       y )Nrf   rb   rj   )rÇ   r'   Ú20170101z
US/EasternrT   r®   )r7   rÇ   T)ru   r,   r'   )r7   rË   )r   r-   r»   r   Ú	Timestampr   r7   r'   r   r   r8   r9   rÚ   s      r   Útest_non_unique_indexrå   `  sª   € ä	Ø�C˜œbŸf™fÐ%°Ñ4Ü�|‰|˜J¨<Ô8Ð9¸AÑ=ô
€Bð �Z‰Z˜Ÿ™ 3˜Ó(×.Ñ.×3Ñ3¸dÈÐ3ÓM€FÜØ	ˆc�3œŸ™ÐÜ�|‰|˜J¨<Ô8Ð9¸AÑ=Øô€Hô
 ×Ñ˜6 8Õ,r_   c                  ó¤  — t        j                  dddt        j                  ddgd¬«      } t        j                  dddt        j                  d	d
gd¬«      }t	        g d¢| |dœ«      }|j                  d«      }|j                  «       }|j                  t        «      j                  d«      j                  «       }t        j                  ||«       y )Nrž   rŸ   ÚcTr	   r    r%   r†   r®   r¸   )r   r    r   r    r   r    )Úcol1Úcol2Úcol3rè   )r   r   r-   r»   r   r   r   r²   r/   r8   r~   )r   Úcat2r   r   r¾   rA   s         r   Útest_rank_categoricalrì   o  s¡   € Ü
�.‰.˜#˜s C¬¯©°°cÐ:ÀDÔ
I€CÜ�>‰>˜1˜a ¤B§F¡F¨A¨qÐ1¸4Ô@€Dä	Ò.¸ÀTÑJÓ	K€Bà	�‰�FÓ	€Bà
�'‰'‹)€Cà�y‰yœÓ ×(Ñ(¨Ó0×5Ñ5Ó7€HÜ×Ñ˜#˜xÕ(r_   c                 óÐ   — t        dd gid¬«      }|j                  dd¬«      d   j                  d| ¬«      }t        dgd|j                  ¬	«      }t        j                  ||«       y )
Nr\   r°   r!   F)Údropnari   )rt   r�   rf   )r"   rË   )r   r   r   r   rË   r8   r9   )r�   r   r>   rA   s       r   Útest_groupby_op_with_nullablesrï   }  sa   € ô 
�C˜$˜�=¨	Ô	2€BØ�Z‰Z˜ EˆZÓ*¨3Ñ/×4Ñ4¸EÈYÐ4ÓW€FÜ�s�e 9°6·;±;Ô?€HÜ×Ñ˜6 8Õ,r_   )$r   Únumpyr-   r   Úpandasr   r   r   r   r   Úpandas._testingÚ_testingr8   r   rB   ÚmarkÚparametrizer.   rä   Ú	to_periodrƒ   Úinfr»   r�   rœ   r£   r³   rÁ   rÅ   rÍ   rÔ   rÛ   rá   rå   rì   rï   r!   s   0r   ú<module>rø      s—  ðÝ ã Û ã ÷ó õ òò"-ð> ‡�×Ñ˜ 5 '¨E°6¨?Ð!;Ó<Ø‡�×ÑØ
ð Xöàð 	ˆ�‰’¨Ö.òð ˆB�L‰L˜Ó&ØˆB�L‰L˜Ó&ØˆB�L‰L˜Ó&ØˆB�L‰L˜Ó&ØˆB�L‰L˜Ó&ð	
ð ˆB�L‰L˜¨,Ô7ØˆB�L‰L˜¨,Ô7ØˆB�L‰L˜¨,Ô7ØˆB�L‰L˜¨,Ô7ØˆB�L‰L˜¨,Ô7ð	
ð ˆB�L‰L˜Ó&¨¨¯©°a«Ñ8ØˆB�L‰L˜Ó&¨¨¯©°a«Ñ8ØˆB�L‰L˜Ó&¨¨¯©°a«Ñ8ØˆB�L‰L˜Ó&¨¨¯©°a«Ñ8ØˆB�L‰L˜Ó&¨¨¯©°a«Ñ8ð	
ð ˆB�L‰L˜Ó&×0Ñ0°Ó5ØˆB�L‰L˜Ó&×0Ñ0°Ó5ØˆB�L‰L˜Ó&×0Ñ0°Ó5ØˆB�L‰L˜Ó&×0Ñ0°Ó5ØˆB�L‰L˜Ó&×0Ñ0°Ó5ð	
ð-ñ	!ñD 	ðI ó %ðL ‡�×ÑØ#à	�D˜%Ò!:Ð;Ø	�D˜$Ò 9Ð:Ø	�E˜5Ò";Ð<Ø	�E˜4Ò!:Ð;Ø	��eÒ6Ð7Ø	��dÒ5Ð6Ø	��uÒ7Ð8Ø	��tÒ6Ð7Ø	��eÒ6Ð7Ø	��dÒ5Ð6Ø	��uÒ7Ð8Ø	��tÒ6Ð7Ø	�$˜Ò8Ð9Ø	�$˜Ò7Ð8Ø	�%˜Ò 9Ð:Ø	�%˜Ò8Ð9Ø	�$˜Ò8Ð9Ø	�$˜ÒUÐVØ	�%˜Ò 9Ð:Ø	�%˜ÒVÐWð)óñ2*ó3óM%ó =ð@*ð ‡�×Ñ˜ 5 '¨E°6¨?Ð!;Ó<Ø‡�×ÑØ
ˆr�v‰vˆg˜Ÿ™�w §¡¨¨R¯V©V°R·V±V¸R¿V¹VÐDÐEóð ‡�×ÑØ)à	�D˜& 3¨¨R¯V©V°Q¸¿¹ÀÀSÐ"IÐJØ	�D˜%Ò!DÐEØ	�D˜(Ò$GÐHØ	�E˜6 C¨¨b¯f©f°a¸¿¹ÀÀcÐ#JÐKØ	�E˜5Ò"EÐFØ	�E˜8Ò%HÐIØ	��f˜s C¨¯©°°b·f±f¸cÀ3ÐGÐHØ	��eÒ@ÐAØ	��hÒ CÐDØ	��v  S¨"¯&©&°#°r·v±v¸sÀCÐHÐIØ	��uÒAÐBØ	��xÒ!DÐEØ	��f˜s C¨¯©°°b·f±f¸cÀ3ÐGÐHØ	��eÒ@ÐAØ	��hÒ CÐDØ	��v  S¨"¯&©&°#°r·v±v¸sÀCÐHÐIØ	��uÒAÐBØ	��xÒ!DÐEØ	�$˜  c¨2¯6©6°3¸¿¹ÀÀSÐ IÐJØ	�$˜ÒBÐCØ	�$˜Ò"EÐFØ	�%˜ # s¨B¯F©F°C¸¿¹ÀÀcÐ!JÐKØ	�%˜Ò CÐDØ	�%˜Ò#FÐGØ	�$˜  c¨2¯6©6°3¸¿¹ÀÀSÐ IÐJØ	�$˜ÒBÐCØ	�$˜Ò"EÐFØ	�%˜ # s¨B¯F©F°C¸¿¹ÀÀcÐ!JÐKØ	�%˜Ò CÐDØ	�%˜Ò#FÐGð=ó"ñF	*óG"óó =ðN	*ð ‡�×Ñ˜ 5 '¨E°6¨?Ð!;Ó<Ø‡�×ÑØ
ð (öàð 	ˆ�‰�!�Q˜Ÿ™  1 a¨¯©°·±Ð8ÀÖFòð ˆB�L‰L˜Ó&ØˆB�L‰L˜Ó&Ø�F‰FØˆB�L‰L˜Ó&ØˆB�L‰L˜Ó&ØˆB�L‰L˜Ó&Ø�F‰FØ�F‰Fð		
ð ˆB�L‰L˜¨,Ô7ØˆB�L‰L˜¨,Ô7Ø�F‰FØˆB�L‰L˜¨,Ô7ØˆB�L‰L˜¨,Ô7ØˆB�L‰L˜¨,Ô7Ø�F‰FØ�F‰Fð		
ð ˆB�L‰L˜Ó&¨¨¯©°a«Ñ8ØˆB�L‰L˜Ó&¨¨¯©°a«Ñ8Ø�F‰FØˆB�L‰L˜Ó&¨¨¯©°a«Ñ8ØˆB�L‰L˜Ó&¨¨¯©°a«Ñ8ØˆB�L‰L˜Ó&¨¨¯©°a«Ñ8Ø�F‰FØ�F‰Fð		
ð ˆB�L‰L˜Ó&×0Ñ0°Ó5ØˆB�L‰L˜Ó&×0Ñ0°Ó5Ø�F‰FØˆB�L‰L˜Ó&×0Ñ0°Ó5ØˆB�L‰L˜Ó&×0Ñ0°Ó5ØˆB�L‰L˜Ó&×0Ñ0°Ó5Ø�F‰FØ�F‰Fð		
ð?)ñ	-ñ\ 	ða ó 1ðd ‡�×ÑØ-ðyàØØØØ�#�r—v‘v˜s C¨¨b¯f©f°b·f±fÐ=ð	
ðyð ØØØØ�#�r—v‘v˜s C¨¨b¯f©f°b·f±fÐ=ð	
ðyð  ØØØØ�#�r—v‘v˜s C¨¨b¯f©f°b·f±fÐ=ð	
ðyð. ØØØØ�#�r—v‘v˜s C¨¨b¯f©f°b·f±fÐ=ð	
ð-yð: 
��f˜e c¨3°·±¸¸SÀ#ÀrÇvÁvÈrÏvÉvÐ%VÐWð;yð< 
��f˜d S¨#¨r¯v©v°s¸CÀÀbÇfÁfÈbÏfÉfÐ$UÐVð=yð@ ØØØØ�#�r—v‘v˜s C¨¨b¯f©f°b·f±fÐ=ð	
ð?yðL 
��v˜t c¨3°·±¸¸SÀ#ÀrÇvÁvÈrÏvÉvÐ%VÐWðMyðN 
��f˜e c¨3°·±¸¸SÀ#ÀrÇvÁvÈrÏvÉvÐ%VÐWðOyðP 
��f˜d S¨#¨r¯v©v°s¸CÀÀbÇfÁfÈbÏfÉfÐ$UÐVðQyðT ØØØØ�#�r—v‘v˜s C¨¨b¯f©f°b·f±fÐ=ð	
ðSyð` 
��v˜t c¨3°·±¸¸SÀ#ÀrÇvÁvÈrÏvÉvÐ%VÐWðayðd ØØØØ�#�r—v‘v˜s C¨¨b¯f©f°b·f±fÐ=ð	
ðcyðr ØØØØ�#�r—v‘v˜s C¨¨b¯f©f°b·f±fÐ=ð	
ðqyð@ ØØØØ�#�r—v‘v˜s C¨¨b¯f©f°b·f±fÐ=ð	
ðyðN ØØØØ�#�r—v‘v˜s C¨¨b¯f©f°b·f±fÐ=ð	
ðMyð\ ØØØØ�#�r—v‘v˜s C¨¨b¯f©f°b·f±fÐ=ð	
ð[yðj ØØØàØØ—‘ØØØØ—‘Ø—‘ð	ð	
ðiyðJ ØØØØ�#�r—v‘v˜s C¨¨b¯f©f°b·f±fÐ=ð	
ðIyðX ØØØàØØ—‘ØØØØ—‘Ø—‘ð	ð	
ðWyðv 
�D˜( EÒ+SÐTðwyðz ØØØÚ?ð	
ðyyðF 
�E˜8 UÒ,TÐUðGyðJ ØØØÚ=ð	
ðIyðV 
��h Ò'OÐPðWyðZ ØØØÚ?ð	
ðYyðf 
��x Ò(PÐQðgyðj ØØØÚ@ð	
ðiyðv 
��h Ò'OÐPðwyðx 
��h Ò&VÐWðyyðz 
��x Ò(PÐQð{yð~ ØØØÚ=ð	
ð}yðJ 
�$˜ %Ò)QÐRðKyðN ØØØÚ>ð	
ðMyðZ 
�%˜ 5Ò*RÐSð[yð^ ØØØÚ>ð	
ð]yðj 
�$˜ %Ò)QÐRðkyðl 
�$˜ $Ò(TÐUðmyðn 
�%˜ 5Ò*RÐSðoyðp 
�%˜ 4Ò)UÐVðqyó|ñz*ó{|óe1ó =ð`*ð" ‡�×ÑØ�Ò1Ð2°TÒ;TÐ4UÐVóñ*óð*ð ‡�×ÑØÒIóð ‡�×Ñ˜ 4¨ -Ó0ñ*ó 1óð*ð  ‡�×Ñ˜Ò&?Ó@Ø‡�×Ñ˜  u Ó.Ø‡�×ÑØ
Ò0°5¸"¿&¹&À%ÈÏÉÐQVÐ2WÐXóñ$óó /ó Að
$ð* ‡�×Ñ˜Ò&6Ó7Ø‡�×Ñ˜  u Ó.Ø‡�×ÑØ
â+Ø	�—‘˜˜rŸv™v uÐ-Ø	
ˆB�F‰F�A�r—v‘v˜qÐ!ðóñ
óó /ó 8ð
ò,ð ‡�×ÑØ(à
ˆQˆ�!�Q�˜#˜s˜Ð$Ú	’|Ò%9Ð:Ú	˜˜1˜a §¡Ð(¨3°°S¸"¿&¹&Ð*AÐBÚ	�Q˜˜2Ÿ6™6�N S¨#¨r¯v©vÐ$6Ð7ð	óñ,óð,ò,ð, ‡�×Ñ˜ T¨5 MÓ2ñ-ó 3ð-ò0-ò)ð ‡�×Ñ˜ u¨hÐ&7Ó8ñ-ó 9ñ-ùòwùòDs   Á%~ØA~