Ë
    ž…j~P  ã                   óØ  — d dl mZ d dlZd dlZd dlmZ d dlZd dlm	Z	m
Z
mZmZmZ d dlmZ d dlmZ ej&                  d„ «       Zd„ Zd„ Zd	„ Zd
„ Zd„ Zd„ Zd„ Zej8                  j;                  dg d¢«      d„ «       Zd„ Zej8                  j;                  dddg«      d„ «       Z d„ Z!d„ Z"d„ Z#d„ Z$d„ Z%d„ Z&d„ Z'd„ Z(d„ Z)d„ Z*ej8                  j;                  dd gd d!gg«      d"„ «       Z+ej8                  j;                  d#d$d%g«      d&„ «       Z,ej8                  j;                  d'd d(g«      d)„ «       Z-d*„ Z.ej8                  j;                  dd gd d!gg«      d+„ «       Z/d,„ Z0d-„ Z1d.„ Z2d/„ Z3d0„ Z4y)1é    )ÚdedentN)Úis_platform_windows)Ú	DataFrameÚIndexÚSeriesÚTimedeltaIndexÚ	Timestamp)Ú
date_rangec            	      ó†   — t        dgdz  dgdz  z   dgdz  z   t        j                  d«      dœt        d	d
dd¬«      ¬«      S )Né   é   é   é   é   é   é(   ©ÚAÚBú1/1/2000ÚsÚns©ÚfreqÚperiodsÚunit©Úindex)r   ÚnpÚaranger
   © ó    úm/root/aria/tools/markitdown-venv/lib/python3.12/site-packages/pandas/tests/resample/test_resampler_grouper.pyÚ
test_framer$      sI   € äØˆc�B‰h˜!˜˜r™Ñ! Q C¨!¡GÑ+´"·)±)¸B³-Ñ@Ü˜¨#°rÀÔEôð r"   c                 ó$  — ddl m} t        d«      }| j                  |«       t	        j
                  d d¬«      5   |d«      5  t        | j                  j                  dd«      «       d d d «       d d d «       y # 1 sw Y   ŒxY w# 1 sw Y   y xY w)	Nr   )Úprovisionalcompleterzã    import numpy as np
    from pandas import Series, date_range
    data = np.arange(10, dtype=np.float64)
    index = date_range("2020-01-01", periods=len(data))
    s = Series(data, index=index)
    rs = s.resample("D")
    F)Úraise_on_extra_warningsÚignorezrs.r   )	ÚIPython.core.completerr&   r   Úrun_cellÚtmÚassert_produces_warningÚlistÚ	CompleterÚcompletions)Úipr&   Úcodes      r#   Ú"test_tab_complete_ipython6_warningr2      s„   € Ý;äð	ó	€Dð ‡K�K�Ôô 
×	#Ñ	# DÀ%Ô	Hñ 5Ù! (Ó+ñ 	5Ü�—‘×)Ñ)¨%°Ó3Ô4÷	5÷5ð 5÷	5ð 	5ú÷5ð 5ús#   º	BÁ&A:Á)BÁ:B	Á?BÂBc                  óÀ  — g d¢g d¢g d¢g d¢g d¢g d¢g d¢g d¢g d	¢g d
¢g
} t        | g d¢¬«      }t        j                  |j                  «      |_        d„ }|j	                  d«      j                  |«      }|j                  d«      j	                  d«      j                  d«      j                  «       }t        j                  ||«       t        t        ddd¬«      g d¢g d¢dœ«      j                  d«      }d„ }|j	                  d«      j                  |«      }|j	                  d«      j                  d«      j                  «       }t        j                  ||«       y )N)ú
2010-01-01r   r   )z
2010-01-02r   r   )z
2010-01-05r   r   )z
2010-01-10r   é   )z
2010-01-13r   r   )r4   r   é   )z
2010-01-03r   r   )z
2010-01-04r   r   )z
2010-01-11r   r5   )z
2010-01-14r   r   )ÚdateÚidÚscore©Úcolumnsc                 ó^   — | j                  d«      j                  d«      j                  «       S )Nr7   ÚD)Ú	set_indexÚresampleÚasfreq©Úxs    r#   Úf_0z'test_deferred_with_groupby.<locals>.f_0E   s%   € Ø�{‰{˜6Ó"×+Ñ+¨CÓ0×7Ñ7Ó9Ð9r"   r8   r7   r=   ú
2016-01-01é   ÚW)Ústartr   r   ©r   r   r   r   )r6   é   r5   r   ©r7   ÚgroupÚvalc                 ó@   — | j                  d«      j                  «       S )NÚ1D)r?   ÚffillrA   s    r#   Úf_1z'test_deferred_with_groupby.<locals>.f_1T   s   € Ø�z‰z˜$Ó×%Ñ%Ó'Ð'r"   rK   rN   )r   ÚpdÚto_datetimer7   ÚgroupbyÚapplyr>   r?   r@   r+   Úassert_frame_equalr
   rO   )ÚdataÚdfrC   ÚexpectedÚresultrP   s         r#   Útest_deferred_with_groupbyrZ   2   s+  € ò 	ÚÚÚÚÚÚÚÚÚð€Dô 
�4Ò!8Ô	9€BÜ�n‰n˜RŸW™WÓ%€B„Gò:ð �z‰z˜$Ó×%Ñ% cÓ*€HØ�\‰\˜&Ó!×)Ñ)¨$Ó/×8Ñ8¸Ó=×DÑDÓF€FÜ×Ñ˜& (Ô+ä	ä \¸1À3ÔGÚ!Úñ	
ó
÷ �i�Óð ò(ð �z‰z˜'Ó"×(Ñ(¨Ó-€HØ�Z‰Z˜Ó ×)Ñ)¨$Ó/×5Ñ5Ó7€FÜ×Ñ˜& (Õ+r"   c                 óØ  — | j                  d«      }|j                  j                  d„ «      }|j                  d«      j                  j	                  «       }t        j                  ||«       |j                  j                  d«      j	                  «       }t        j                  ||«       |j                  d«      j	                  «       j                  }t        j                  ||«       y )Nr   c                 ó@   — | j                  d«      j                  «       S ©NÚ2s)r?   ÚmeanrA   s    r#   ú<lambda>ztest_getitem.<locals>.<lambda>_   s   €  1§:¡:¨dÓ#3×#8Ñ#8Ó#:€ r"   r^   )rS   r   rT   r?   r_   r+   Úassert_series_equal)r$   ÚgrX   rY   s       r#   Útest_getitemrc   \   sª   € Ø×Ñ˜3Ó€Aà�s‰s�y‰yÑ:Ó;€Hà�Z‰Z˜Ó×Ñ×$Ñ$Ó&€FÜ×Ñ˜6 8Ô,à�S‰S�\‰\˜$Ó×$Ñ$Ó&€FÜ×Ñ˜6 8Ô,à�Z‰Z˜Ó×"Ñ"Ó$×&Ñ&€FÜ×Ñ˜6 8Õ,r"   c                  óª  — dddœdddœg} t        | t        dd¬«      ¬«      }|j                  d	«      j                  d
«      }|d   j	                  «       }t
        j                  j                  ddg|j                  gd¬«      }t        ddg|d¬«      }t        j                  ||«       |d   j	                  «       }t        j                  ||«       y )Nr   r   )r8   Úbuyerr   r   rD   ©r   r   r8   rN   re   )r8   N©Únames)r   Úname)r   r
   rS   r?   ÚcountrQ   Ú
MultiIndexÚfrom_arraysr   r   r+   ra   )rV   rW   ÚrrY   Úexp_mirX   s         r#   Útest_getitem_multiplero   k   sÇ   € ð ˜sÑ#¨A¸Ñ%<Ð=€DÜ	�4œz¨,ÀÔBÔ	C€BØ
�
‰
�4Ó×!Ñ! $Ó'€AØˆw‰Z×ÑÓ€Fä�]‰]×&Ñ&¨¨A¨°·±Ð'9ÀÐ&ÓN€FÜØ	
ˆAˆØØô€Hô
 ×Ñ˜6 8Ô,àˆw‰Z×ÑÓ€FÜ×Ñ˜6 8Õ,r"   c                  ó\  — t        t        d«      t        dd¬«      ddœ«      } | j                  d«      j	                  d«      j                  d	«      d
   j                  «       }| j	                  d«      j                  d	d¬«      d
   j                  «       }t        j                  ||«       y )NÚaabbbz1-1-2016r6   rf   r   )r8   r7   rV   r7   r8   Ú2DrV   ©Úon)	r   r-   r
   r>   rS   r?   Úsumr+   ra   )rW   ÚexprY   s      r#   Ú)test_groupby_resample_on_api_with_getitemrw      s’   € ä	Ü�G‹}¤j°ÀQÔ&GÐQRÑSó
€Bð �,‰,�vÓ
×
&Ñ
& tÓ
,×
5Ñ
5°dÓ
;¸FÑ
C×
GÑ
GÓ
I€CØ�Z‰Z˜Ó×&Ñ& t°Ð&Ó7¸Ñ?×CÑCÓE€FÜ×Ñ˜6 3Õ'r"   c                  ó~  — d} d\  }}d}t        ||d¬«      }t        t        j                  j	                  d«      j                  t        |«      «      |¬«      }||| }t        j                  | ¬«      }|j                  |«      j                  d«      }||| }|j                  |«      j                  d«      }	t        j                  t        d	¬
«      5  t        j                  |j                   |	j                   «       d d d «       t#        d«      }
t        j                  | |
¬«      }|j                  |«      j                  d«      }||| }|j                  |«      j                  d«      }t        j$                  ||«       t#        d«      t        j&                  d«      dz  z   }t        j                  | |¬«      }|j                  |«      j                  d«      }||| }|j                  |«      j                  d«      }t        j$                  ||«       t        j$                  ||«       y # 1 sw Y   �Œ4xY w)NÚ1399min)z1/1/2000 00:00:00z1/31/2000 00:00z1/15/2000 00:00:00Ú1231min©r   r   r   rj   zIndex are different©Úmatchr   )r   Úorigini0u  )r
   r   r   ÚrandomÚdefault_rngÚstandard_normalÚlenrQ   ÚGrouperrS   ÚaggÚpytestÚraisesÚAssertionErrorr+   Úassert_index_equalr   r	   ra   Ú	Timedelta)r   rG   ÚendÚmiddleÚrngÚtsÚts2Úsimple_grouperÚcount_tsÚ	count_ts2r~   Úadjusted_grouperÚadjusted_count_tsÚadjusted_count_ts2Úorigin_futureÚadjusted_grouper2Úadjusted2_count_tsÚadjusted2_count_ts2s                     r#   Útest_groupby_with_originr™   ‰   sí  € ð €DØ7�J€Eˆ3Ø!€Fä
�U˜C iÔ
0€CÜ	”—	‘	×%Ñ% aÓ(×8Ñ8¼¸S»ÓBÈ#Ô	N€BØ
ˆV�Cˆ.€Cô —Z‘Z TÔ*€NØ�z‰z˜.Ó)×-Ñ-¨gÓ6€HØ˜˜sÐ#€HØ—‘˜NÓ+×/Ñ/°Ó8€IÜ	�‰”~Ð-BÔ	Cñ ?Ü
×Ñ˜hŸn™n¨i¯o©oÔ>÷?ô �q‹\€FÜ—z‘z t°FÔ;ÐØŸ
™
Ð#3Ó4×8Ñ8¸ÓAÐØ)¨&°Ð5ÐØŸ™Ð%5Ó6×:Ñ:¸7ÓCÐÜ×ÑÐ,Ð.@ÔAô ˜a“L¤2§<¡<°	Ó#:¸VÑ#CÑC€MÜŸ
™
¨°]ÔCÐØŸ™Ð$5Ó6×:Ñ:¸7ÓCÐØ+¨F°3Ð7ÐØŸ+™+Ð&7Ó8×<Ñ<¸WÓEÐÜ×ÑÐ-Ð/BÔCô ×ÑÐ,Ð.AÕB÷+?ñ ?ús   Ã+H2È2H<c                  ó  — t        dddd¬«      } t        t        d«      | ¬«      j                  d«      j	                  «       }t        g d¢t        j                  g d	¢d
d¬«      ¬«      }t        j                  ||«       y )Nr   r   Úminr   ©r   r   r   r   Ú20s)r   r   r   r   r   r   r   )z2000-01-01 00:00:00z2000-01-01 00:00:20z2000-01-01 00:00:40z2000-01-01 00:01:00z2000-01-01 00:01:20z2000-01-01 00:01:40z2000-01-01 00:02:00zdatetime64[ns])Údtyper   )	r
   r   Úranger?   ÚnearestrQ   ÚDatetimeIndexr+   ra   )r   rY   rX   s      r#   Útest_nearestr¢   ²   sr   € ô �z¨1°5¸tÔD€EÜ”E˜!“H EÔ*×3Ñ3°EÓ:×BÑBÓD€FäÚÜ×Ñòð #Øô
ô€Hô  ×Ñ˜6 8Õ,r"   Úf)ÚfirstÚlastÚmedianÚsemru   r_   r›   ÚmaxÚsizerj   r    ÚbfillrO   r@   Úohlcc                 óÀ   ‡ — |j                  d«      }|j                  d«      } t        |‰ «      «       }|j                  ˆ fd„«      }t	        j
                  ||«       y )Nr   r^   c                 óD   •—  t        | j                  d«      ‰«      «       S r]   ©Úgetattrr?   ©rB   r£   s    €r#   r`   ztest_methods.<locals>.<lambda>ä   s   ø€ Ð!=¤¨¯©°DÓ)9¸1Ó!=Ó!?€ r"   )rS   r?   r¯   rT   r+   Úassert_equal©r£   r$   rb   rm   rY   rX   s   `     r#   Útest_methodsr³   Ë   sO   ø€ ð* 	×Ñ˜3Ó€AØ	�
‰
�4Ó€AàŒW�Q˜‹]‹_€FØ�w‰wÓ?Ó@€HÜ‡O�O�F˜HÕ%r"   c                 óà   — | j                  d«      }|j                  d«      }|j                  j                  «       }|j                  j	                  d„ «      }t        j                  ||«       y )Nr   r^   c                 ó@   — | j                  d«      j                  «       S r]   )r?   ÚnuniquerA   s    r#   r`   z&test_methods_nunique.<locals>.<lambda>í   s   €  1§:¡:¨dÓ#3×#;Ñ#;Ó#=€ r"   )rS   r?   r   r¶   rT   r+   ra   )r$   rb   rm   rY   rX   s        r#   Útest_methods_nuniquer·   è   sR   € à×Ñ˜3Ó€AØ	�
‰
�4Ó€AØ�S‰S�[‰[‹]€FØ�s‰s�y‰yÑ=Ó>€HÜ×Ñ˜6 8Õ,r"   ÚstdÚvarc                 óÄ   ‡ — |j                  d«      }|j                  d«      } t        |‰ «      d¬«      }|j                  ˆ fd„«      }t	        j
                  ||«       y )Nr   r^   r   ©Úddofc                 óH   •—  t        | j                  d«      ‰«      d¬«      S )Nr^   r   r»   r®   r°   s    €r#   r`   z&test_methods_std_var.<locals>.<lambda>ö   s   ø€ Ð!=¤¨¯©°DÓ)9¸1Ó!=À1Ô!E€ r"   )rS   r?   r¯   rT   r+   rU   r²   s   `     r#   Útest_methods_std_varr¾   ñ   sR   ø€ à×Ñ˜3Ó€AØ	�
‰
�4Ó€AØŒW�Q˜‹] Ô"€FØ�w‰wÓEÓF€HÜ×Ñ˜& (Õ+r"   c                 ó.  — | j                  d«      }|j                  d«      }|j                  d«      j                  «       }d„ }|j                  |«      }t	        j
                  ||«       d„ }|j                  |«      }t	        j
                  ||«       y )Nr   r^   c                 ó@   — | j                  d«      j                  «       S r]   )r?   ru   rA   s    r#   rC   ztest_apply.<locals>.f_0  s   € Ø�z‰z˜$Ó×#Ñ#Ó%Ð%r"   c                 óD   — | j                  d«      j                  d„ «      S )Nr^   c                 ó"   — | j                  «       S ©N©ru   )Úys    r#   r`   z)test_apply.<locals>.f_1.<locals>.<lambda>  s   € °·±³€ r"   )r?   rT   rA   s    r#   rP   ztest_apply.<locals>.f_1  s   € Ø�z‰z˜$Ó×%Ñ%Ñ&7Ó8Ð8r"   )rS   r?   ru   rT   r+   rU   )r$   rb   rm   rX   rC   rY   rP   s          r#   Ú
test_applyrÆ   ú   s�   € Ø×Ñ˜3Ó€AØ	�
‰
�4Ó€Að �z‰z˜$Ó×#Ñ#Ó%€Hò&ð �W‰W�S‹\€FÜ×Ñ˜& (Ô+ò9ð �W‰W�S‹\€FÜ×Ñ˜& (Õ+r"   c                  óf  — t        ddd¬«      } t        dt        j                  j	                  d«      j                  t        | «      «      i| ¬«      }d„ }|j                  t        j                  d	¬«      «      j                  |«      }|j                  d	«      j                  |«      }t        j                  ||«       |d   j                  t        j                  d	¬«      d
¬«      j                  |«      }|d   j                  d	«      j                  |«      }t        j                  ||«       y )Nz1-1-2015z12-31-15r=   r{   Úcol1r   ©rV   r   c                 ó(   — t        ddgddg¬«      }|S )Nr   r   ÚaÚbr   )r   )rB   r   s     r#   r£   z(test_apply_with_mutated_index.<locals>.f  s   € Ü�A�q�6 # s Ô,ˆØˆr"   ÚMEF)Ú
group_keys)r
   r   r   r   r€   r‚   rS   rQ   rƒ   rT   r?   r+   rU   ra   )r   rW   r£   rX   rY   s        r#   Útest_apply_with_mutated_indexrÏ     só   € ä�z :°CÔ8€EÜ	Ø”b—i‘i×+Ñ+¨AÓ.×5Ñ5´c¸%³jÓAÐBÈ%ô
€Bòð �z‰zœ"Ÿ*™*¨$Ô/Ó0×6Ñ6°qÓ9€Hà�[‰[˜Ó×$Ñ$ QÓ'€FÜ×Ñ˜& (Ô+ð �&‰z×!Ñ!¤"§*¡*°$Ô"7ÀEÐ!ÓJ×PÑPÐQRÓS€HØ�‰Z× Ñ  Ó&×,Ñ,¨QÓ/€FÜ×Ñ˜6 8Õ,r"   c                  ój  ‡— t         j                  j                  ddg«      } t        ddd¬«      }t	        t        j                  dgdz  t
        j                  ¬	«      j                  dd
«      || ¬«      }|j                  D �ci c]+  }||d   dk(  rt
        j                  nt
        j                  “Œ- c}Š|j                  d«      j                  ˆfd„«      }t	        d
ddggz  t        ddd
¬«      t         j                  j                  ddg«      ¬«      }t        j                  ||«       y c c}w )N)r   rË   Ú Úone)r   rÌ   ÚiÚtwoz
2017-01-01Ú15Minr   )rG   r   r   r   é   ©rž   r   ©r   r;   r   rÒ   Úhc                 ó.   •—  ‰| j                      | «      S rÃ   ©ri   )rB   Úagg_dicts    €r#   r`   z/test_apply_columns_multilevel.<locals>.<lambda>,  s   ø€ Ð.>¨h°q·v±vÑ.>¸qÓ.A€ r"   g        Ú1h)rQ   rk   Úfrom_tuplesr
   r   r   ÚarrayÚint64Úreshaper;   ru   r_   r?   rT   r+   rU   )ÚcolsÚindrW   ÚcolrY   rX   rÜ   s         @r#   Útest_apply_columns_multilevelrå   $  s   ø€ ä�=‰=×$Ñ$Ð&;Ð=SÐ%TÓU€DÜ
˜<¨g¸qÔ
A€CÜ	Ü
�‰�!��r‘¤§¡Ô*×2Ñ2°1°aÓ8ÀÈTô
€Bð JLÏÉÖTÀ#�  A¡¨%¢”b—f’f´R·W±WÑ<ÒT€HØ�[‰[˜Ó×#Ñ#Ó$AÓB€FÜØ	ˆa�ˆXˆJ‰Ü˜|°$ÀÔBÜ—‘×)Ñ)Ø"Ð$:Ð;ó
ô€Hô ×Ñ˜& (Õ+ùò Us   Â0D0c                  ó  — d„ } t        dddd¬«      }t        g d¢|¬«      }t        g d	¢|¬«      }|j                  d
«      j                  | |d¬«      }t        ddd
d¬«      }t        dg|¬«      }t	        j
                  ||«       y )Nc                 óà   — | j                  «       } |j                  | j                  «      j                  d«      j	                  «       }|j
                  d   |z  }| ||k\     j
                  d   S )Nr   éÿÿÿÿ)Úsort_valuesÚreindexr   ÚfillnaÚcumsumÚiloc)ÚseriesÚweightsÚqrì   Úcutoffs        r#   Úweighted_quantilez5test_apply_non_naive_index.<locals>.weighted_quantile8  sc   € Ø×#Ñ#Ó%ˆØ—‘ §¡Ó.×5Ñ5°aÓ8×?Ñ?ÓAˆØ—‘˜R‘ 1Ñ$ˆØ�f Ñ&Ñ'×,Ñ,¨QÑ/Ð/r"   z2017-6-23 18:00r   Ú15minÚUTC)r   r   Útz)ç      ð?r   r   r   r   r   r   r   r   )g      d@é[   éA   é+   é   é
   r   r   r=   g      à?)rï   rð   z2017-06-23 00:00:00+00:00)r   rõ   rö   )r
   r   r?   rT   r+   ra   )rò   ÚtimesrV   rï   rY   rã   rX   s          r#   Útest_apply_non_naive_indexrý   7  s‰   € ò0ô Ð(°!¸'ÀeÔL€EÜÒ,°EÔ:€DÜÒ6¸eÔD€Gà�]‰]˜3Ó×%Ñ%Ð&7ÀÈCÐ%ÓP€FÜ
Ø#Ð%@ÀsÈuô€Cô �s�e 3Ô'€HÜ×Ñ˜6 8Õ,r"   c                 óÆ  — t        ddd| ¬«      }t        |g d¢g d¢dœ¬«      }|j                  d	«      j                  d
d¬«      j	                  «       }t        j                  g d¢t
        j                  ¬«      t        j                  g d¢d| › d�¬«      g}t        j                  j                  |d	d g¬«      }t        dg d¢i|¬«      }t        j                  ||«       y )Nú
2000-01-01rr   r6   r   )r   r   r   r   r   )r   r   r   r   r   )Úcol0rÈ   )r   rV   r   Ú1WÚleft)Úlabel)r   r   r   r   r×   )z
1999-12-26ú
2000-01-02r  r  zM8[ú]rg   rÈ   )r   r   r   r   rÉ   )r
   r   rS   r?   ru   r   rß   rà   rQ   rk   rl   r+   rU   )r   r   rW   rY   ÚmiÚmindexrX   s          r#   Ú test_resample_groupby_with_labelr  J  s¿   € ä�|¨$¸ÀÔE€EÜ	˜ªoÂÑ%WÔ	X€BØ�Z‰Z˜Ó×(Ñ(¨°VÐ(Ó<×@Ñ@ÓB€Fô 	�‰’¤R§X¡XÔ.Ü
�‰ÚDØ˜�v˜Q�-ô	
ð
€Bô �]‰]×&Ñ& r°&¸$°Ð&Ó@€FÜ˜v¢|Ð4¸FÔC€Hä×Ñ˜& (Õ+r"   c                 óþ  — | }t        g d¢d¬«      }|j                  d«      j                  d«      j                  «       }|j                  j
                  dk(  sJ ‚t        j                  |j                  j                  d   |«       |j                  d«      j                  d«      j                  «       }|j                  j
                  dk(  sJ ‚t        j                  |j                  j                  d   |«       y )N©r   r   r   r   rÛ   r^   r   r   r   )
r   rS   r?   r_   r   Únlevelsr+   rˆ   ÚlevelsÚrolling)r$   rW   rX   rY   s       r#   Útest_consistency_with_windowr  ]  sÃ   € à	€BÜ’Y SÔ)€HØ�Z‰Z˜‹_×%Ñ% dÓ+×0Ñ0Ó2€FØ�<‰<×Ñ 1Ò$Ð$Ð$Ü×Ñ˜&Ÿ,™,×-Ñ-¨aÑ0°(Ô;à�Z‰Z˜‹_×$Ñ$ RÓ(×-Ñ-Ó/€FØ�<‰<×Ñ 1Ò$Ð$Ð$Ü×Ñ˜&Ÿ,™,×-Ñ-¨aÑ0°(Õ;r"   c            	      ó~  — t        t        j                  j                  d«      j	                  d«      t        d«      t        ddd¬«      ¬«      } | j                  d	«      j                  «       }g d
¢| _	        | j                  d	«      j                  «       }|j                  |_	        t        j                  ||«       y )Nr   )r   r   Úaaaú
2012-01-01r   r   ©r   r   ©r;   r   Ú5s)rË   rÌ   Úc)r   r   r   r€   r�   r-   r
   r?   r¦   r;   r+   rU   ©rW   rY   rX   s      r#   Útest_median_duplicate_columnsr  j  s‘   € ô 
Ü
�	‰	×Ñ˜aÓ ×0Ñ0°Ó9Ü�U“Ü˜¨r¸Ô<ô
€Bð
 �[‰[˜Ó×%Ñ%Ó'€FÚ €B„JØ�{‰{˜4Ó ×'Ñ'Ó)€HØ—~‘~€HÔÜ×Ñ˜& (Õ+r"   c                  ót  — t        t        d«      t        dd«      dœt        ddd¬«      ¬«      } | j                  d«      j	                  d	„ «      }t        g d
¢t        ddd¬«      ¬«      }t        j                  ||«       | j                  d«      j	                  d„ «      }t        j                  ||«       y )Nrû   r   )rä   rÈ   r  Ú20minr  r   rÙ   c                 ó6   — | j                   j                  «       S rÃ   )rä   ru   ©rK   s    r#   r`   z0test_apply_to_one_column_of_df.<locals>.<lambda>�  s   € °%·)±)·-±-³/€ r"   )r   r   é   é	   rE   c                 ó(   — | d   j                  «       S )Nrä   rÄ   r  s    r#   r`   z0test_apply_to_one_column_of_df.<locals>.<lambda>ˆ  s   € °%¸±,×2BÑ2BÓ2D€ r"   )r   rŸ   r
   r?   rT   r   r+   ra   r  s      r#   Útest_apply_to_one_column_of_dfr  y  sœ   € ä	Ü�b“	¤5¨¨R£=Ñ1Ü˜¨r¸Ô@ô
€Bð �[‰[˜Ó×#Ñ#Ñ$AÓB€FÜÚœj¨¸qÀsÔKô€Hô ×Ñ˜6 8Ô,ð �[‰[˜Ó×#Ñ#Ñ$DÓE€FÜ×Ñ˜6 8Õ,r"   c                  ó"  — t        g d¢g d¢g d¢dœ«      } t        j                  | d   «      | d<   | j                  d«      j	                  dd¬«      }|d	g   j                  «       }|j                  d	d
i«      }t        j                  ||«       y )N)Úcat_1r!  Úcat_2r!  r"  r!  r"  r!  )r6   r   é   r   rE   é   rû   é2   )z2019-2-1z
2018-02-03ú	2020-3-11ú2019-2-2r'  z	2018-12-4r&  z
2020-12-12)ÚcatÚnumr7   r7   r(  ÚYErs   r)  ru   )	r   rQ   rR   rS   r?   ru   r„   r+   rU   )rW   Ú	resampledrX   rY   s       r#   Útest_resample_groupby_aggr,  Œ  s�   € ä	ò	ò 1ò	ñ	
ó
€Bô2 —‘  6¡
Ó+€B€v�Jà—
‘
˜5Ó!×*Ñ*¨4°FÐ*Ó;€IØ˜%˜Ñ!×%Ñ%Ó'€HØ�]‰]˜E 5˜>Ó*€Fä×Ñ˜& (Õ+r"   c                  ó\  — t        d«      } t        dgdgdœt        | gd¬«      ¬«      }|j                  d«      j	                  d	«      d
   }|j                  ddg«      }t        ddggt        j                  j                  d| fgddg¬«      ddg¬«      }t        j                  ||«       y )Nz2021-02-28 00:00:00ÚbetaéE   )ÚclassÚvaluer7   rÛ   r   r0  rÍ   r1  ru   r©   r   rg   rØ   )r	   r   r   rS   r?   r„   rQ   rk   rÞ   r+   rU   )r�   rW   r+  rY   rX   s        r#   Ú"test_resample_groupby_agg_listliker2  °  s©   € ä	Ð(Ó	)€BÜ	˜f˜X°°Ñ5¼UÀBÀ4ÈfÔ=UÔ	V€BØ—
‘
˜7Ó#×,Ñ,¨TÓ2°7Ñ;€IØ�]‰]˜E 6˜?Ó+€FÜØ
ˆaˆˆ	Ü�m‰m×'Ñ'¨&°"¨¨¸wÈÐ>OÐ'ÓPØ˜�ô€Hô
 ×Ñ˜& (Õ+r"   ÚkeysrË   rÌ   c                 ó¸  — t        g ddgt        g «      ¬«      }|j                  | «      j                  t	        j
                  d«      ¬«      j                  «       }| dgk(  rdgng }t        ddg¬«      j                  | d¬«      j                  t        g «      d	¬
«      |   }t        | «      dk(  r| d   |j                  _
        t        j                  ||«       y )NrË   rÌ   r  ú00:00:01©Úruler:   F©ÚdropT)Úappendr   r   )r   r   rS   r?   rQ   Úto_timedeltar_   r>   r‚   r   ri   r+   rU   ©r3  rW   rY   Úexpected_columnsrX   s        r#   Ú
test_emptyr>  ¾  sÃ   € ô 
�2  S˜z´ÀÓ1CÔ	D€BØ�Z‰Z˜Ó×&Ñ&¬B¯O©O¸JÓ,GÐ&ÓH×MÑMÓO€FØ $¨¨¢˜‘u°2Ðä˜3 ˜*Ô%ß	‰�4˜eˆÓ	$ß	‰”> "Ó%¨dˆÓ	3Ð4Dñ	Fð ô
 ˆ4ƒy�A‚~Ø" 1™gˆ�‰Ôä×Ñ˜& (Õ+r"   ÚconsolidateTFc                 ór  — t        dddd¬«      }t        d|t        d«      ddœ«      }t        d	|t        d«      d
œ«      }t        j                  ||gd¬«      }| r|j                  «       }|j                  dg«      j                  dd¬«      j                  «       }t        j                  j                  dgdz  d	gdz  z   t        j                  g d¢dz  «      j                  d«      gddg¬«      }t        g d¢dz  dgdz  t        j                  gdz  z   dœ|¬«      }t        j                   ||«       y )Nz
2020-01-01é   r=   r   rœ   r   rL   )Úkeyr7   rÈ   Ú
col_objectr   )rB  r7   rÈ   T)Úignore_indexrB  rF   r7   rs   r   )z
2020-01-05z
2020-01-12z
2020-01-19r   rg   )r   r6   r   )rÈ   rC  r   )r
   r   rŸ   rQ   ÚconcatÚ_consolidaterS   r?   r›   rk   rl   rR   Úas_unitr   Únanr+   rU   )r?  ÚdatesÚdf1Údf2rW   rY   ÚidxrX   s           r#   Ú.test_resample_groupby_agg_object_dtype_all_nanrM  Ï  s/  € ô �|¨R°cÀÔE€EÜ
˜C¨¼¸b»	ÐQVÑWÓ
X€CÜ
˜C¨¼¸b»	ÑBÓ
C€CÜ	�‰�C˜�:¨DÔ	1€BÙØ�_‰_Óˆà�Z‰Z˜˜Ó ×)Ñ)¨#°&Ð)Ó9×=Ñ=Ó?€FÜ
�-‰-×
#Ñ
#àˆE�A‰I˜˜ ™	Ñ!Ü�N‰NÒEÈÑIÓJ×RÑRØóð	
ð �fˆoð $ó €Cô â ‘NØ ˜' A™+¬¯©¨°1©Ñ4ñ	
ð ô€Hô ×Ñ˜& (Õ+r"   Ú	min_countr   c                 óü  — | }|j                  t        j                  t        j                  gt	        |«      z  |¬«      ¬«      }|j                  d«      j                  d«      }|j                  |¬«      }t        j                  g d¢t        j                  dd¬	«      j                  d«      ggg d
¢g d¢gdd g¬«      }|dk(  rdnt        j                  }t        d|i||¬«      }t        j                  ||«       y )Nr×   )r   r   Ú40s)rN  r
  rÿ   r   )r   )r   r   r   )r   r   r   ©r  Úcodesrh   r   rÑ   r   )r   rž   )ÚassignrQ   rß   ÚNAr‚   rS   r?   ru   rk   rR   rG  r   r+   rU   )	Ústring_dtype_no_objectr$   rN  rž   ÚgbrsrY   r   r1  rX   s	            r#   Ú&test_groupby_resample_empty_sum_stringrW  î  sÔ   € ð
 #€EØ×"Ñ"¤R§X¡X¬r¯u©u¨g¼¸J»Ñ.GÈuÔ%UÐ"ÓV€JØ×Ñ˜cÓ"×+Ñ+¨EÓ2€DØ�X‰X 	ˆXÓ*€Fä�M‰MÚœBŸN™N¨<¸dÔC×KÑKÈDÓQÐRÐSÚš)Ð$Ø�Dˆkô€Eð
 ˜q’.‰B¤b§e¡e€EÜ˜#˜u˜¨U¸%Ô@€HÜ×Ñ˜& (Õ+r"   c                  ól  — t        t        dd¬«      g d¢g d¢dœ¬«      } | j                  d«      j                  d	d
¬«      dg   j	                  «       }t
        j                  j                  g d¢| d
   j                  d d d…   gdd
g¬«      }t        dg d¢i|¬«      }t        j                  ||«       y )NrD   r   ©rG   r   ©r   r   r   r   r   r   r   r   )r   r5   r6   r   r   rû   r6   r   rJ   )rV   rK   rr   r7   rs   rL   ©r   r   r   r   r   rg   )ç      @g      @g      @g      @rÉ   )r   r
   rS   r?   r_   rQ   rk   rl   Ú_valuesr+   rU   ©rW   rY   Úmi_exprX   s       r#   Ú'test_groupby_resample_with_list_of_keysr`    sº   € ä	ä \¸1Ô=Ú-Ú,ñ
ô
€Bð �Z‰Z˜Ó ×)Ñ)¨$°6Ð)Ó:¸E¸7ÑC×HÑHÓJ€Fä�]‰]×&Ñ&Ú	�r˜&‘z×)Ñ)©#¨A¨#Ñ.Ð/¸ÀÐ7Hð 'ó €Fô àÒ'ð
ð ô	€Hô ×Ñ˜& (Õ+r"   c                 ó&  — t        g g d¢¬«      }t        j                  |d   «      |d<   |j                  d«      }|j	                  | «      j                  t        j                  d«      ¬«      j                  «       }| dgk(  rdgng }t        g d¢¬«      j                  | d¬	«      }t        j                  |d   «      |d<   |j                  dd
d
¬«      |   }t        | «      dk(  r| d   |j                  _
        t        j                  ||«       y )N)rË   rÌ   r7   r:   r7   r5  r6  rË   rÌ   Fr8  T)r:  r9  r   r   )r   rQ   rR   r>   rS   r?   r;  r_   r‚   r   ri   r+   rU   r<  s        r#   Útest_resample_no_indexrb    sð   € ô 
�2Ò1Ô	2€BÜ—‘  6¡
Ó+€B€v�JØ	�‰�fÓ	€BØ�Z‰Z˜Ó×&Ñ&¬B¯O©O¸JÓ,GÐ&ÓH×MÑMÓO€FØ $¨¨¢˜‘u°2ÐÜÒ!3Ô4×>Ñ>¸tÈ%Ð>ÓP€HÜ—~‘~ h¨vÑ&6Ó7€HˆVÑØ×!Ñ! &°¸DÐ!ÓAÐBRÑS€HÜ
ˆ4ƒy�A‚~Ø" 1™gˆ�‰Ôä×Ñ˜& (Õ+r"   c                  óö  — t        t        t        j                  g d¢«      d¬«      ¬«      } | j	                  g d¢«      j                  t        j                  d«      ¬«      j                  «       }t        j                  g d¢«      }t        t        j                  t        j                  d	d
gt        j                  ¬«      |gg d¢g d¢gd dg¬«      ¬«      }t        j                  ||t        «        ¬«       y )N)ú2018-01-01 00:00:00ú2018-01-01 12:00:00ú2018-01-02 00:00:00r7   rÛ   r   )r   r   r   z06:00:00r6  )rd  z2018-01-01 06:00:00re  rf  r   r   r×   )r   r   r   r   )r   r   r   r   rQ  )Úcheck_index_type)r   r   rQ   rR   rS   r?   r;  r_   rk   r   rß   Úintpr+   rU   r   )rW   rY   r   rX   s       r#   Útest_resample_no_columnsri  *  sÈ   € ä	ÜÜ�N‰NÚUóð ô	
ô
€Bð �Z‰Zš	Ó"×+Ñ+´·±ÀÓ1LÐ+ÓM×RÑRÓT€FÜ�N‰Nò	
ó€Eô Ü�m‰mÜ—H‘H˜a ˜V¬2¯7©7Ô3°UÐ;Ú¢Ð.Ø˜�.ô
ô€Hô ×Ñ˜& (ÔATÓAVÐ=VÖWr"   c            	      óº  — t        dgdz  dgdz  z   dgdz  z   dgdz  z   t        j                  d«      dœt        dddd¬	«      ¬
«      } | j	                  d«      j                  d«      j                  «       }t        j                  j                  g d¢t        j                  ddgdz  d¬«      gdd g¬«      }t        d|¬
«      }t        j                  ||«       y )Nr   r   r   r   r   z31/12/2000 18:00rÙ   r   r   r   r   r=   rH   z
2000-12-31z
2001-01-01zM8[ns]r×   rg   )r   r   r    r
   rS   r?   r©   rQ   rk   rl   r¡   r   r+   ra   r^  s       r#   Ú)test_groupby_resample_size_all_index_samerk  I  sÝ   € ä	Øˆc�A‰g˜˜˜a™Ñ 1 #¨¡'Ñ)¨Q¨C°!©GÑ3¼"¿)¹)ÀB»-ÑHÜÐ+°#¸rÈÔMô
€Bð �Z‰Z˜‹_×%Ñ% cÓ*×/Ñ/Ó1€Fä�]‰]×&Ñ&âÜ×Ñ˜l¨LÐ9¸AÑ=ÀXÔNð	
ð �Dˆkð 'ó €Fô Ø	Øô€Hô ×Ñ˜6 8Õ,r"   c                  ód  — t        g d¢g d¢dœt        ddd¬«      ¬«      } | j                  d	«      j                  d
«      dg   j	                  «       }t
        j                  j                  g d¢| j                  d d d…   gd	dg¬«      }t        dg d¢i|¬«      }t        j                  ||«       y )NrZ  ©r   r   rE   r   r6   r  r   rI   ©rK   rL   rD   r   r7   ©rG   r   ri   rÉ   rK   rr   rL   r[  r   rg   ©g       @g      @g      @r\  ©r   r
   rS   r?   r_   rQ   rk   rl   r   r+   rU   r^  s       r#   Ú0test_groupby_resample_on_index_with_list_of_keysrr  _  s²   € ä	â-Ú+ñ
ô ˜|°Q¸VÔDô
€Bð �Z‰Z˜Ó ×)Ñ)¨$Ó/°°Ñ8×=Ñ=Ó?€Fä�]‰]×&Ñ&Ú	�r—x‘x¡ ! ‘}Ð%¨g°vÐ->ð 'ó €Fô àÒ'ð
ð ô	€Hô ×Ñ˜& (Õ+r"   c                  óx  — t        g d¢g d¢g d¢g d¢dœt        ddd¬	«      ¬
«      } | j                  d«      j                  d«      ddg   j	                  «       }t
        j                  j                  g d¢| j                  d d d…   gddg¬«      }t        g d¢g d¢dœ|¬
«      }t        j                  ||«       y )NrZ  rm  )r   r5   r   r   r   r   r   r   )r   rE   r   rE   r   r   r   r6   )rK   Ú	first_valÚ
second_valÚ	third_valrD   r   r7   ro  rÉ   rK   rr   rt  ru  r[  r   rg   rp  )ç      @rw  g      @rw  )rt  ru  rq  r^  s       r#   Ú>test_groupby_resample_on_index_with_list_of_keys_multi_columnsrx  v  s»   € ä	â-Ú1Ú2Ú1ñ	
ô ˜|°Q¸VÔDô
€Bð �Z‰Z˜Ó ×)Ñ)¨$Ó/°¸lÐ0KÑL×QÑQÓS€Fä�]‰]×&Ñ&Ú	�r—x‘x¡ ! ‘}Ð%¨g°vÐ->ð 'ó €Fô â-Ú.ñ
ð ô€Hô ×Ñ˜& (Õ+r"   c            	      ó  — t        g d¢g d¢dœt        t        dd¬«      d¬«      ¬	«      } | j                  d
«      }|j	                  d«      }t        j                  t        d¬«      5  |dg    d d d «       y # 1 sw Y   y xY w)NrZ  rm  rn  rD   r   rY  r7   rÛ   rÉ   rK   rr   zColumns not foundr|   Úval_not_in_dataframe)r   r   r
   rS   r?   r…   r†   ÚKeyError)rW   ÚgbÚrss      r#   Ú?test_groupby_resample_on_index_with_list_of_keys_missing_columnr~  �  s|   € ä	â-Ú+ñ
ô Ü˜\°1Ô5Øô
ô	
€Bð 
�‰�GÓ	€BØ	�‰�TÓ	€BÜ	�‰”xÐ':Ô	;ñ %Ø
Ð"Ð#Ò$÷%÷ %ñ %ús   Á(A8Á8B)5Útextwrapr   Únumpyr   r…   Úpandas.compatr   ÚpandasrQ   r   r   r   r   r	   Úpandas._testingÚ_testingr+   Úpandas.core.indexes.datetimesr
   Úfixturer$   r2   rZ   rc   ro   rw   r™   r¢   ÚmarkÚparametrizer³   r·   r¾   rÆ   rÏ   rå   rý   r  r  r  r  r,  r2  r>  rM  rW  r`  rb  ri  rk  rr  rx  r~  r!   r"   r#   ú<module>r‰     sÁ  ðÝ ã Û å -ã ÷õ õ Ý 4ð ‡�ñó ðò5ò,',òT-ò-ò((ò&CòR-ð2 ‡�×ÑØòóñ(&ó)ð(&ò-ð ‡�×Ñ˜˜u e˜nÓ-ñ,ó .ð,ò,ò(-ò,,ò&-ò&,ò&
<ò,ò-ò&!,òH,ð ‡�×Ñ˜ 3 %¨#¨s¨Ð!4Ó5ñ,ó 6ð,ð  ‡�×Ñ˜¨¨u¨Ó6ñ,ó 7ð,ð< ‡�×Ñ˜ q¨! fÓ-ñ,ó .ð,ò&,ð. ‡�×Ñ˜ 3 %¨#¨s¨Ð!4Ó5ñ,ó 6ð,ò Xò>-ò,,ò.,ó4%r"   