Ë
    ¤ehâ?  ã                   óL  — d dl Zd dlZd dlZd dlmZmZ d dlmZ	 ej                  j                  dg d¢«      ej                  j                  dg d¢g d¢fg d¢g d	¢fg d¢g d	¢fg d
¢g d¢fdej                  dej                  dgdej                  dej                  dgfej                  dej                  dej                  gej                  dej                  dej                  gf ej                  ddd¬«       ej                  ddd¬«      ddd…   f ej                  ddd¬«      j                  d«       ej                  ddd¬«      ddd…   j                  d«      fej                  gdz  ej                  gdz  fg	«      ej                  j                  dg d¢«      d„ «       «       «       Zd„ Zd„ Zd„ Zd„ Zej                  j                  ddd g«      ej                  j                  d!d gd d"gg«      ej                  j                  dd#d$gg«      d%„ «       «       «       Zd&„ Zd'„ Zd(„ Zej                  j                  d)dej                  dej                  g ed*«      ddgd+dgfdej                  ddg ed*«      ddgd+d,gfd-d.d.ej                  g ed*«      d-d.gd d/gfd gd0gd gd1gfg g  ej2                  g d2¬3«       ej2                  g d2¬3«      fg«      d4„ «       Zej                  j                  d5 ej2                  g d6¢d7z  d8¬3«       ej2                  g d9¢d7z  d:¬3«      g«      ej                  j                  dd#g d;¢g«      d<„ «       «       Zej                  j                  dd#g d;¢g«      ej                  j                  d=d>d?g«      d@„ «       «       ZdA„ ZdB„ Zej                  j                  dCg dD¢«      dE„ «       Zej                  j                  ddFdGg«      dH„ «       Z ej                  j                  dIdJdKg«      dL„ «       Z!dM„ Z"dN„ Z#dO„ Z$dP„ Z%dQ„ Z&y)Ré    N)Ú	DataFrameÚIndexÚinterpolation)ÚlinearÚlowerÚhigherÚnearestÚmidpointza_vals,b_vals)é   é   é   é   é   )r   r   r   r   r   )r   r   r   r   ©r   r   r   r   )ç      ð?ç       @ç      @ç      @ç      @)r   r   r   r   r   r   r   r   r   r   z1/1/18ÚDr   )ÚfreqÚperiodséÿÿÿÿÚsÚq)r   ç      Ð?ç      à?ç      è?r   c                 ól  — | dk(  rK|dk(  rFt        |t        «      r6|g d¢k(  r/|j                  t        j                  j                  d¬«      «       t        j                  t        j                  |«      t        j                  |«      g«      }t        j                  |«      j                  || ¬«      }t        j                  |«      j                  || ¬«      }t        dgt        |«      z  dgt        |«      z  z   |d	œ«      }t        ||gd
gt        ddgd¬«      ¬«      }	|j                  j                  dk(  rA|	j                  j                   d   j                  dk(  r|	j#                  |j                  «      }	|j%                  d«      j                  || ¬«      }
t'        j(                  |
|	«       y )Nr	   r   r   zBUnclear numpy expectation for nearest result with equidistant data)Úreason)r   ÚaÚb©ÚkeyÚvalr%   r$   ©Úname)ÚcolumnsÚindexÚMr   )Ú
isinstanceÚlistÚapplymarkerÚpytestÚmarkÚxfailÚpdÚconcatÚSeriesÚquantiler   Úlenr   ÚdtypeÚkindÚdtypesÚvaluesÚastypeÚgroupbyÚtmÚassert_frame_equal)r   Úa_valsÚb_valsr   ÚrequestÚall_valsÚ
a_expectedÚ
b_expectedÚdfÚexpectedÚresults              úh/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/pandas/tests/groupby/methods/test_quantile.pyÚtest_quantilerH      sr  € ð> 	˜Ò"Ø�ŠHÜ�vœtÔ$Ø’lÒ"à×ÑÜ�K‰K×Ñð/ð ó ô	
ô �y‰yœ"Ÿ)™) FÓ+¬R¯Y©Y°vÓ->Ð?Ó@€Hä—‘˜6Ó"×+Ñ+¨A¸]Ð+ÓK€JÜ—‘˜6Ó"×+Ñ+¨A¸]Ð+ÓK€Jä	˜C˜5¤3 v£;Ñ.°#°¼¸V»Ñ1DÑDÈXÑVÓ	W€BäØ	�ZÐ ¨5¨'¼ÀÀS¸zÐPUÔ9Vô€Hð ‡~�~×Ñ˜cÒ! h§o¡o×&<Ñ&<¸QÑ&?×&DÑ&DÈÒ&Kð —?‘? 8§>¡>Ó2ˆØ�Z‰Z˜Ó×'Ñ'¨¸Ð'ÓG€Fä×Ñ˜& (Õ+ó    c                  óœ  — t        dg d¢i«      } t        j                  g d¢t        j                  ¬«      }| j	                  |«      j                  dg«      }t        j                  j                  ddgdgg«      }t        dddgi|¬	«      }t        j                  ||«       t        g d
¢g d¢dœ«      } t        j                  j                  ddgddgg«      }t        j                  g d¢t        j                  ¬«      }| j	                  |«      j                  ddg«      }t        g d¢g d¢dœ|¬	«      }t        j                  ||«       y )NÚA©r   r   r   r   r   )r   r   r   r   r   ©r6   r   r   r   ç      @©r)   )r   r   r   r   )r   r   é   é   ©rK   ÚBr   )r   r   r   r   )r   r   g      @g      @)g      @g      @g      @g      @©r   ÚnpÚarrayÚint64r;   r4   r1   Ú
MultiIndexÚfrom_productr<   r=   )rD   r$   rF   r)   rE   s        rG   Útest_quantile_arrayrZ   I   s
  € ä	�CšÐ)Ó	*€BÜ
�(‰(’?¬"¯(©(Ô
3€CØ�Z‰Z˜‹_×%Ñ% t fÓ-€Fä�M‰M×&Ñ&¨¨A¨°°Ð'7Ó8€EÜ˜#  d˜|Ð,°EÔ:€HÜ×Ñ˜& (Ô+ä	šªLÑ9Ó	:€BÜ�M‰M×&Ñ&¨¨A¨°°t°Ð'=Ó>€Eä
�(‰(’<¤r§x¡xÔ
0€CØ�Z‰Z˜‹_×%Ñ% t¨T lÓ3€FÜÚ&Ò-EÑFÈeô€Hô ×Ñ˜& (Õ+rI   c                  ó–  — t         j                  j                  d«      j                  dddt         j                  ¬«      } t        | t        d«      ¬«      }|j                  d«      j                  d	d
g«      }t        g d¢g d¢dœt        j                  j                  g d¢d	d
ggdd g¬«      ¬«      }t        j                  ||«       y )Nr   r   r   )é
   r   )Úsizer6   ÚABC©r(   rK   ç333333Ó?çffffffæ?)
r   r   gffffff@gš™™™™™@r`   ra   gš™™™™™	@r   r`   ra   )
r   r   gffffffþ?gÌÌÌÌÌÌ@r`   ra   çÍÌÌÌÌÌ@r   ç333333ó?gffffff@)rS   ÚCrL   ©ÚnamesrO   )rU   ÚrandomÚdefault_rngÚintegersrW   r   r,   r;   r4   r1   rX   rY   r<   r=   )ÚarrrD   rF   rE   s       rG   Útest_quantile_array2rk   ^   s¬   € ä
�)‰)×
Ñ
 Ó
"×
+Ñ
+¨A¨q°wÄbÇhÁhÐ
+Ó
O€CÜ	�3¤ U£Ô	,€BØ�Z‰Z˜‹_×%Ñ% s¨C jÓ1€FÜâCÚRñ	
ô �m‰m×(Ñ(Ú˜s C˜jÐ)°#°t°ð )ó 
ô€Hô ×Ñ˜& (Õ+rI   c                  ó:  — t        g d¢g d¢dœ«      } t        j                  g d¢t        j                  ¬«      }| j	                  |d¬«      j                  g d¢«      }t        g d	¢g d
¢dœt        j                  j                  ddgg d¢g«      ¬«      }t        j                  ||«       | j	                  |d¬«      j                  ddg«      }t        g d¢g d¢dœt        j                  j                  ddgddgg«      ¬«      }t        j                  ||«       y )N©r   r   r   ©r   r   r   rR   )r   r   r   rM   F)Úsort)r   r   r   )r   r   ç      ø?r   r   r   )ç      @r   ç      @r   r   r   r   r   rO   r   r   )rp   r   r   r   )rr   rq   r   r   rT   )rD   r$   rF   rE   s       rG   Útest_quantile_array_no_sortrs   o   sé   € Ü	šªÑ3Ó	4€BÜ
�(‰(’9¤B§H¡HÔ
-€CØ�Z‰Z˜ %ˆZÓ(×1Ñ1Ò2CÓD€FÜÚ,Ò3QÑRÜ�m‰m×(Ñ(¨1¨a¨&Ò2CÐ)DÓEô€Hô ×Ñ˜& (Ô+à�Z‰Z˜ %ˆZÓ(×1Ñ1°4¸°,Ó?€FÜÚ"Ò)=Ñ>Ü�m‰m×(Ñ(¨1¨a¨&°4¸°,Ð)?Ó@ô€Hô ×Ñ˜& (Õ+rI   c                  ó  — t        g d¢g d¢g d¢g d¢dœ«      } | j                  ddg«      j                  dd	g«      }t        j                  j                  g d
¢g d¢¬«      }t        g d¢g d¢dœ|¬«      }t        j                  ||«       y )Nrm   rn   )r!   r!   r!   )r!   r!   r"   )rK   rS   ÚcÚdru   rv   r   r   ))r!   r!   r   )r!   r!   r   )r!   r"   r   )r!   r"   r   )ru   rv   Nre   )r   r   r   r   )g      
@g      @r   r   rR   rO   )r   r;   r4   r1   rX   Úfrom_tuplesr<   r=   )rD   rF   r)   rE   s       rG   Ú#test_quantile_array_multiple_levelsrx   �   sƒ   € Ü	ÚšiªoÂOÑTó
€Bð �Z‰Z˜˜c˜
Ó#×,Ñ,¨d°D¨\Ó:€FÜ�M‰M×%Ñ%ÚPÚð &ó €Eô Ú$Ò+AÑBÈ%ô€Hô ×Ñ˜& (Õ+rI   Ú
frame_size)r   r   )éd   r\   r;   r   r   g333333ã?c           
      ót  — | \  }}t        t        j                  t        |«      D �cg c]  }||dz  gz  ‘Œ c}«      t        |«      ¬«      }t        j                  t        |d«      «      gt        |«      z  |gz   }t        t        |d«      «      D ��cg c]  }|D ]  }|‘Œ Œ c}}gt        |«      z  t        t        t        |«      «      «      t        |d«      z  gz   }	t        j                  ||	|d gz   ¬«      }
t        t        |d«      «      D ��cg c]$  }|D ]  }t        |«      g|t        |«      z
  z  ‘Œ Œ& }}}t        |«      D �cg c]	  }||vsŒ|‘Œ }}t        ||
|¬«      }|j                  |«      j                  |«      }t        j                  ||«       y c c}w c c}}w c c}}w c c}w )Nr   r_   )ÚlevelsÚcodesrf   ©r)   r(   )r   rU   rV   ÚrangeÚarangeÚminr5   r,   r1   rX   Úfloatr;   r4   r<   r=   )ry   r;   r   ÚnrowÚncolÚ_rD   Ú
idx_levelsÚxÚ	idx_codesÚexpected_indexÚexpected_valuesÚexpected_columnsrE   rF   s                  rG   Ú6test_groupby_quantile_with_arraylike_q_and_int_columnsrŒ   �   s   € ð
 �J€Dˆ$Ü	”2—8‘8´U¸4³[ÖA°˜T Q¨¡U G›^ÒAÓBÌEÐRVËKÔ	X€Bä—)‘)œC  a›LÓ)Ð*¬S°«\Ñ9¸Q¸CÑ?€JÜ"¤3 t¨Q£<Ó0×<˜¸!Ò<°Q’!Ð<�!Ó<Ð=ÄÀGÃÑLÜŒU”3�q“6‹]Óœc $¨›lÑ*ðPñ €Iô —]‘]Ø °'¸T¸FÑ2Bô€Nô 5:¼#¸dÀA»,Ó4G÷Ø/0ÐQRòØLMŒˆq‹ˆ
�dœS ›\Ñ)Ó*ðØ*ð€Oñ ô $)¨£;ÖC˜a°!¸7Ò2BšÐCÐÐCÜØ˜~Ð7Gô€Hð �Z‰Z˜Ó ×)Ñ)¨!Ó,€Fä×Ñ˜& (Õ+ùò% Bùó =ùóùò Ds   §F$ÂF)Ä)F/Å	F5ÅF5c                  óÒ   — t        ddgddgddggddg¬«      } t        j                  t        d¬	«      5  | j	                  d«      j                  «        d d d «       y # 1 sw Y   y xY w)
NÚfoor!   r"   ru   r$   r%   r_   z+cannot be performed against 'object' dtypes©Úmatch)r   r.   ÚraisesÚ	TypeErrorr;   r4   )rD   s    rG   Útest_quantile_raisesr“   «   s^   € Ü	�U˜C�L 5¨# ,°¸°Ð=ÈÈuÀ~Ô	V€Bä	�‰”yÐ(UÔ	Vñ %Ø
�
‰
�5Ó×"Ñ"Ô$÷%÷ %ñ %ús   ´ AÁA&c                  ó`  — t        g d¢t        d«      dœ«      } | j                  g d¢«      }t        j                  t
        d¬«      5  |j                  d«       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   y xY w)	N)r   r   r   r   r   r   rP   ©r!   r"   zGot '50.0' insteadr�   é2   zGot '-1.0' insteadr   )r   r   r;   r.   r‘   Ú
ValueErrorr4   )rD   Úgs     rG   Ú$test_quantile_out_of_bounds_q_raisesr™   ²   sŠ   € ä	Ò+´%¸³(Ñ;Ó	<€BØ
�
‰
Ò%Ó&€AÜ	�‰”zÐ)=Ô	>ñ Ø	�
‰
�2Œ÷ô 
�‰”zÐ)=Ô	>ñ Ø	�
‰
�2Œ÷ð ÷ð ú÷ð ús   ÁBÁ=B$ÂB!Â$B-c                  óà   — t        j                  dt         j                  dg«      } t        | t	        d«      dœ«      }|j                  d«      }t	        d«      D ]  }|j                  «        Œ y )Nr   r   r#   r$   rz   )rU   rV   Únanr   r   r;   r4   )ÚdatarD   Úgrpr…   s       rG   Ú/test_quantile_missing_group_values_no_segfaultsrž   ½   sZ   € ä�8‰8�Sœ"Ÿ&™& #Ð&Ó'€DÜ	˜4¬¨a«Ñ1Ó	2€Bð �*‰*�UÓ
€CÜ�3‹Zò ˆØ�‰�ñrI   z$key, val, expected_key, expected_valr   ç        rN   r!   r"   rp   é*   g      E@Úfloat64rM   c                 ó  — t        | |dœ«      }t        |t        |d¬«      dg¬«      }|j                  d«      }|j                  d«      }t	        j
                  ||«       |j                  «       }t	        j
                  ||«       y )Nr#   r$   r&   r%   r~   r   )r   r   r;   r4   r<   r=   )r$   r%   Úexpected_keyÚexpected_valrD   rE   r�   rF   s           rG   Ú2test_quantile_missing_group_values_correct_resultsr¥   È   sv   € ô 
˜3 sÑ+Ó	,€BäØœE ,°UÔ;ÀeÀWô€Hð �*‰*�UÓ
€Cà�\‰\˜#Ó€FÜ×Ñ˜& (Ô+à�\‰\‹^€FÜ×Ñ˜& (Õ+rI   r9   )r   r   Nr   ÚInt64)TFNÚboolean©rŸ   r   r   c                 ó|  — t        dgdz  dgdz  z   | dœ«      }|j                  d«      d   j                  |«      }t        |t        «      r,t
        j                  j                  ddg|fdd g¬«      }g d¢}nt        ddgd¬	«      }d
g}t        j                  |dz  |dd¬«      }t        j                  ||«       y )Nr‡   r   Úyr•   r!   r"   re   r¨   r&   r   r   ÚFloat64©r)   r'   r6   )r   r;   r4   r+   r,   r1   rX   rY   r   r3   r<   Úassert_series_equal)r9   r   rD   rF   ÚidxÚtrue_quantilesrE   s          rG   Ú$test_groupby_quantile_nullable_arrayr°   å   sµ   € ô 
˜#˜ ™ c U¨Q¡YÑ.°VÑ<Ó	=€BØ�Z‰Z˜‹_˜SÑ!×*Ñ*¨1Ó-€Fä�!”TÔÜ�m‰m×(Ñ(¨3°¨*°a¨ÀÀdÀÐ(ÓLˆÚ(‰ä�S˜#�J SÔ)ˆØ˜ˆä�y‰y˜¨!Ñ+°3¸SÈ	ÔR€HÜ×Ñ˜6 8Õ,rI   Únumeric_onlyTFc                 óŠ  — t        dgdgdgdœ«      }|r]|j                  d«      j                  | |¬«      }|j                  d«      dg   j                  | «      }t        j                  ||«       y t        j                  t        d¬	«      5  |j                  d«      j                  | |¬«       d d d «       y # 1 sw Y   y xY w)
Nr   r   r‡   )r!   r"   ru   r!   )r±   r"   z7'quantile' cannot be performed against 'object' dtypes!r�   )r   r;   r4   r<   r=   r.   r‘   r’   )r   r±   rD   rF   rE   s        rG   Ú-test_groupby_quantile_raises_on_invalid_dtyper³   ý   s²   € ô 
˜!˜ C 5°¨uÑ5Ó	6€BÙØ—‘˜C“×)Ñ)¨!¸,Ð)ÓGˆØ—:‘:˜c“? C 5Ñ)×2Ñ2°1Ó5ˆÜ
×Ñ˜f hÕ/ä�]‰]ÜÐVô
ñ 	Cð �J‰J�s‹O×$Ñ$ Q°\Ð$ÔB÷	C÷ 	Cñ 	Cús   Â#B9Â9Cc                 ó  — t        ddgdt        j                  gdœ| ¬«      }|j                  d«      d   j	                  d«      }t        dg| d¬	«      }| d
v r| }nd }t        j                  dg||d¬«      }t        j                  ||«       |j                  d«      d   j	                  ddg«      }t        j                  dgdz  t        j                  j                  |ddgfdd g¬«      d|¬«      }t        j                  ||«       y )Nr   çš™™™™™É?©r‡   rª   rM   r‡   rª   r   r   )r6   r'   )ÚFloat32r«   ©r6   r)   r'   r   r   re   r¬   )r   rU   r›   r;   r4   r   r1   r3   r<   r­   rX   rY   )Úany_float_dtyperD   rF   Ú	exp_indexÚexpected_dtyperE   s         rG   Útest_groupby_quantile_NA_floatr¼     sù   € ä	˜!˜Q˜ s¬B¯F©F mÑ4¸OÔ	L€BØ�Z‰Z˜‹_˜SÑ!×*Ñ*¨3Ó/€FÜ�s�e ?¸Ô=€IàÐ0Ñ0Ø(‰àˆä�y‰y˜#˜ n¸IÈCÔP€HÜ×Ñ˜6 8Ô,à�Z‰Z˜‹_˜SÑ!×*Ñ*¨C°¨;Ó7€FÜ�y‰yØ	ˆ�‰	Ü�m‰m×(Ñ(¨)°c¸4°[Ð)AÈ#ÈtÈÐ(ÓUØØô	€Hô ×Ñ˜6 8Õ,rI   c           	      ó’  — t        ddgddgdœ| ¬«      }|j                  d«      d   j                  d«      }t        j                  d	gd
t        dgd| ¬«      d¬«      }t        j                  ||«       |j                  d«      j                  d«      }t        dd	id
t        dgd| ¬«      ¬«      }t        j                  ||«       y )Nr   r   r   r¶   rM   r‡   rª   r   rq   r«   )r'   r6   r¸   ©r6   r)   )	r   r;   r4   r1   r3   r   r<   r­   r=   )Úany_int_ea_dtyperD   rF   rE   s       rG   Útest_groupby_quantile_NA_intrÀ   $  s¾   € ä	˜!˜Q˜ q¨! fÑ-Ð5EÔ	F€BØ�Z‰Z˜‹_˜SÑ!×*Ñ*¨3Ó/€FÜ�y‰yØ	ˆØÜ�Q�C˜cÐ)9Ô:Øô	€Hô ×Ñ˜8 VÔ,à�Z‰Z˜‹_×%Ñ% cÓ*€FÜØ	ˆcˆ
˜)¬5°!°¸3ÐFVÔ+Wô€Hô ×Ñ˜& (Õ+rI   zinterpolation, val1, val2))r   r   r   )r   r   r   )r	   r   r   c           
      ó˜  — t        g d¢dddt        j                  gdœ|¬«      }|j                  d«      j	                  dd	g| ¬
«      }t        d||t        j                  t        j                  gi|t        j
                  j                  t        j                  g d¢|¬«      g d¢gdd g¬«      ¬«      }t        j                  ||«       y )N©r   r   r   r   r   r   r   r•   rM   r!   r   ra   ©r   r   r"   ©r   r   r   r   )r   ra   r   ra   re   r¾   ©
r   r1   ÚNAr;   r4   rX   Úfrom_arraysr3   r<   r=   )r   Úval1Úval2Úany_numeric_ea_dtyperD   rF   rE   s          rG   Ú)test_groupby_quantile_all_na_group_maskedrË   7  s´   € ô 
Ú ! Q¨¬2¯5©5Ð!1Ñ2Ð:Nô
€Bð �Z‰Z˜‹_×%Ñ%¨¨c¨
À-Ð%ÓP€FÜØ	ˆt�Tœ2Ÿ5™5¤"§%¡%Ð(Ð)Ø"Ü�m‰m×'Ñ'Ü�Y‰Y’|Ð+?Ô@ÒBVÐWØ˜�+ð (ó 
ô€Hô ×Ñ˜& (Õ+rI   r
   r   c           
      ó¬  — t        g d¢dddt        j                  gdœ|¬«      }|j                  d«      j	                  dd	g| ¬
«      }|dk(  r|}nd}t        dddt        j                  t        j                  gi|t        j
                  j                  t        j                  g d¢|¬«      g d¢gdd g¬«      ¬«      }t        j                  ||«       y )NrÂ   r   r   r   r•   rM   r!   r   r   rÃ   r·   r«   r"   r   rN   rÄ   )r   r   r   r   re   r¾   rÅ   )r   rÊ   rD   rF   r»   rE   s         rG   Ú0test_groupby_quantile_all_na_group_masked_interprÍ   M  sÊ   € ô
 
Ú ! Q¨¬2¯5©5Ð!1Ñ2Ð:Nô
€Bð �Z‰Z˜‹_×%Ñ%¨¨d¨À=Ð%ÓQ€Fà˜yÒ(Ø-‰à"ˆäØ	ˆs�CœŸ™¤§¡Ð&Ð'ØÜ�m‰m×'Ñ'ä—	‘	š,Ð.BÔCÚ&ðð ˜�+ð (ó 
ô
€Hô ×Ñ˜& (Õ+rI   r6   r«   r·   c                 óH  — t        ddgt        j                  gdz  dœ| ¬«      }|j                  d«      d   j	                  d«      }t        j
                  t        j                  g| t        dg| ¬«      d¬	«      }d|j                  _
        t        j                  ||«       y )
Nr   r   r¶   rM   r‡   rª   r   r   r¸   )r   r1   rÆ   r;   r4   r3   rU   r›   r   r)   r'   r<   r­   )r6   rD   rF   rE   s       rG   Ú"test_groupby_quantile_allNA_columnrÏ   j  s„   € ô 
˜!˜Q˜¤r§u¡u g°¡kÑ2¸%Ô	@€BØ�Z‰Z˜‹_˜SÑ!×*Ñ*¨3Ó/€FÜ�y‰yÜ	�‰ˆ˜¤U¨C¨5¸Ô%>ÀSô€Hð €H‡N�NÔÜ×Ñ˜8 VÕ,rI   c                  óh  — t        t        j                  t        j                  d«      d¬«      g d¢dœ«      } | j                  d«      j                  d«      }t        dt        j                  d	«      t        j                  d
«      git        ddgd¬«      ¬«      }t        j                  ||«       y )Nr   r   )ÚunitrÄ   )ÚvalueÚgrouprÓ   g®Gáz®ï?rÒ   z0 days 00:00:00.990000z0 days 00:00:02.990000r   r   r&   rO   )r   r1   Úto_timedeltarU   r€   r;   r4   Ú	Timedeltar   r<   r=   )rD   rF   rE   s      rG   Útest_groupby_timedelta_quantilerÖ   v  s•   € ä	Ü—/‘/¤"§)¡)¨A£,°SÔ9ÂLÑQó
€Bð �Z‰Z˜Ó ×)Ñ)¨$Ó/€FÜàÜ—‘Ð5Ó6Ü—‘Ð5Ó6ðð	
ô �Q˜�F Ô)ô€Hô ×Ñ˜& (Õ+rI   c            	      ó  — t        t        j                  d«      j                  dd«      t	        d«      t        j                  t	        d«      d¬«      ¬«      } d	}t        j                  t        |¬
«      5  | j                  dd¬«      }d d d «       j                  ddg¬«      }t        g d¢g d¢g d¢gt	        d«      t
        j                  j                  g d¢dd g¬«      ¬«      }t        j                  ||«       y # 1 sw Y   ŒwxY w)Né   r   r   ÚXYZÚABABÚcolr&   r~   z+DataFrame.groupby with axis=1 is deprecatedr�   r   )Úaxisçš™™™™™é?rµ   )r   )gš™™™™™ù?gš™™™™™Ù?rb   gffffffö?)gffffff@gš™™™™™@gffffff@gš™™™™™@)g333333#@gÍÌÌÌÌÌ @g333333%@gÍÌÌÌÌÌ"@)©rK   rÝ   ©rK   rµ   ©rS   rÝ   ©rS   rµ   re   )r   rU   r€   Úreshaper,   r1   r3   r<   Úassert_produces_warningÚFutureWarningr;   r4   rX   rw   r=   )rD   ÚmsgÚgbrF   rE   s        rG   Útest_columns_groupby_quantilerç   ˆ  sâ   € ä	Ü
�	‰	�"‹×Ñ˜a Ó$Ü�5‹kÜ—	‘	œ$˜v›,¨UÔ3ô
€Bð
 8€CÜ	×	#Ñ	#¤M¸Ô	=ñ 'Ø�Z‰Z˜ AˆZÓ&ˆ÷'à�[‰[˜C ˜:ˆ[Ó&€FÜâ Ú Ú!ð	
ô
 �5‹kÜ—‘×)Ñ)Ú<ÀUÈDÀMð *ó 
ô
€Hô ×Ñ˜& (Õ+÷'ð 'ús   Á6D Ä D	c           
      óÀ  — t        j                  dddd| ¬«      j                  d«      }t        |t	        t        dd«      «      t	        t        dd	«      «      d
œ«      }|j                  d«      j                  ddg«      }t         j                  j                  |d d d…   ddggd¬«      }t        dddœdddœdddœdddœg|¬«      }t        j                  ||«       y )Nz2020-04-19 00:00:00Ú1minrz   ÚUTC)Ústartr   r   ÚtzrÑ   Ú1hr   ée   éÉ   )Ú	timestampÚcategoryrÒ   rð   rµ   rÝ   éc   )rð   Nre   gš™™™™™)@g333333\@)rñ   rÒ   gš™™™™H@gfffff†b@g333333Q@gš™™™™e@gÍÌÌÌÌW@gfffffh@rO   )r1   Ú
date_rangeÚfloorr   r,   r   r;   r4   rX   rY   r<   r=   )rÑ   ÚdtirD   rF   ÚmirE   s         rG   Útest_timestamp_groupby_quantiler÷   ¢  sç   € ä
�-‰-Ø#¨&¸#À%Èdôç�eˆDƒkð ô 
àÜœU 1 c›]Ó+Üœ%  S›/Ó*ñ	
ó
€Bð �Z‰Z˜Ó$×-Ñ-¨s°C¨jÓ9€Fä	�‰×	#Ñ	# S©¨2¨¡Y°°c°
Ð$;ÐCVÐ	#Ó	W€BÜà¨Ñ.Ø¨Ñ.Ø¨Ñ.Ø¨Ñ.ð		
ð ô€Hô ×Ñ˜& (Õ+rI   c                  óþ  — t        j                  dd¬«      } t        j                  | «      j                  «       j	                  «       }| j                  d«      |d<   | j                  d«      |d<   | | d   z
  |d	<   t         j                  |j                  d
<   t        j                  t        j                  d«      d«      }|j                  |«      }|j                  d«      }t        d«      D �ci c]$  }||j                  |d d…   j                  d«      “Œ& }}t        |«      j                   j#                  «       }|j$                  j'                  t(        «      |_        t+        j,                  ||«       y c c}w )Nz
2016-01-01iè  )r   z
US/Pacificr   r   r   r   r   r   r   éÈ   r   )r1   ró   r3   Úto_frameÚcopyÚtz_localizeÚ	to_periodÚNaTÚilocrU   Útiler€   r;   r4   r   r   ÚTÚinfer_objectsr)   r:   Úintr<   r=   )rõ   rD   Úbyræ   rF   ÚiÚexprE   s           rG   Ú#test_groupby_quantile_dt64tz_periodr  ¿  s$  € ä
�-‰-˜¨dÔ
3€CÜ	�‰�3‹×	 Ñ	 Ó	"×	'Ñ	'Ó	)€BØ�O‰O˜LÓ)€B€q�EØ�M‰M˜#Ó€B€q�EØ�#�a‘&‰L€B€q�EÜ—&‘&€B‡G�GˆB�Kä	�‰”—‘˜1“˜sÓ	#€BØ	�‰�B‹€Bà�[‰[˜Ó€Fô 49¸³8Ö
<¨aˆ1ˆb�g‰g�a�d˜�d‰m×$Ñ$ SÓ)Ñ)Ð
<€CÐ
<Ü˜‹~×Ñ×-Ñ-Ó/€HØ—^‘^×*Ñ*¬3Ó/€H„Nä×Ñ˜& (Õ+ùò	 =s   Ã1)E:c                  óD  — t         j                  j                  g d¢g d¢¬«      } t        j                  t	        d«      | ¬«      }|j                  dd¬«      j                  d	d
g«      }t         j                  j                  g d¢dd g¬«      }t        j                  g d¢|¬«      }t        j                  ||«       t         j                  j                  j                  j                  ddgd	d
gg«      }t        j                  |j                  j                  |«       y )N))r   r!   rS   )r   r!   rK   )r   r"   rS   )r   r"   rK   )r   r!   rS   )r   r!   rK   )r   r"   rS   )r   r"   rK   )ÚsampleÚcat0Úcat1re   é   rO   r  F)Úlevelro   rµ   rÝ   )rá   rà   rß   rÞ   )rc   g333333@gš™™™™™@g333333@rS   rK   )r1   rX   rw   r3   r   r;   r4   r<   r­   ÚcoreÚindexesÚfrozenÚ
FrozenListÚassert_equalr)   r|   )ÚindÚserrF   ÚqindrE   Úexpected_levelss         rG   Ú+test_groupby_quantile_nonmulti_levels_orderr  Õ  sæ   € ä
�-‰-×
#Ñ
#ò		
ò )ð $ó €Cô �)‰)”E˜!“H CÔ
(€CØ�[‰[˜v¨Eˆ[Ó2×;Ñ;¸SÀ#¸JÓG€Fä�=‰=×$Ñ$Ú8ÀÈÀð %ó €Dô �y‰yÒ-°TÔ:€Hä×Ñ˜6 8Ô,ô —g‘g—o‘o×,Ñ,×7Ñ7¸#¸s¸ÀcÈ3ÀZÐ8PÓQ€OÜ‡O�O�F—L‘L×'Ñ'¨Õ9rI   )'ÚnumpyrU   r.   Úpandasr1   r   r   Úpandas._testingÚ_testingr<   r/   Úparametrizer›   ró   Úas_unitrH   rZ   rk   rs   rx   rŒ   r“   r™   rž   r   rV   r¥   r°   r³   r¼   rÀ   rË   rÍ   rÏ   rÖ   rç   r÷   r  r  © rI   rG   ú<module>r     si  ðÛ Û ã ÷õ ð ‡�×ÑØÒIóð ‡�×ÑØò 
š/Ð*Ú	’|Ð$Ú	š,Ð'â	"Ò$=Ð>à
ˆr�v‰v�s˜BŸF™F CÐ	(¨3°·±¸¸R¿V¹VÀSÐ*IÐJØ
�&‰&�#�r—v‘v˜s B§F¡FÐ	+¨b¯f©f°c¸2¿6¹6À3ÈÏÉÐ-OÐPð ˆB�M‰M˜(¨°aÔ8ØˆB�M‰M˜(¨°aÔ8¹¸2¸Ñ>ð	
ð
 ˆB�M‰M˜(¨°aÔ8×@Ñ@ÀÓEØˆB�M‰M˜(¨°aÔ8¹¸2¸Ñ>×FÑFÀsÓKð	
ð
 �&‰&ˆ�A‰˜Ÿ™�x !‘|Ð$ð)óð2 ‡�×Ñ˜Ò5Ó6ñ,ó 7ó3óð:,ò@,ò*,ò",ò$,ð ‡�×Ñ˜¨°	Ð':Ó;Ø‡�×Ñ˜ a S¨1¨a¨& MÓ2Ø‡�×Ñ˜  S˜z˜lÓ+ñ,ó ,ó 3ó <ð,ò0%òòð ‡�×ÑØ*à
ˆr�v‰v�s˜BŸF™FÐ	#¡U¨1£X°°S¨z¸CÀ¸:ÐFØ
ˆr�v‰v�s˜CÐ	 ¡%¨£(¨S°#¨J¸¸c¸
ÐCØ
ˆs�C˜Ÿ™Ð	 ¡%¨£(¨S°#¨J¸¸C¸ÐAØ
ˆˆrˆd�Q�C˜$˜Ð Ø	ˆR��—‘˜" IÔ.°°·±¸À9Ô0MÐNðó	ñ,ó	ð,ð& ‡�×ÑØàˆ�‰’ Ñ!¨Ô1Øˆ�‰Ò$ qÑ(°	Ô:ðóð ‡�×Ñ˜˜s¢OÐ4Ó5ñ-ó 6óð-ð  ‡�×Ñ˜˜s¢OÐ4Ó5Ø‡�×Ñ˜¨$°¨Ó7ñ
Có 8ó 6ð
Cò-ò0,ð& ‡�×ÑØÒ!Wóñ,óð,ð& ‡�×Ñ˜¨:°xÐ*@ÓAñ,ó Bð,ð8 ‡�×Ñ˜ 9¨iÐ"8Ó9ñ-ó :ð-ò,ò$,ò4,ò:,ó,:rI   