
    Piz                        d dl m Z  d dlZd dlZd dlZd dlmZ d dlmZ d dl	Z
d dl	mZmZmZ d dlmZ d dlmZ ej        d             Zej        d             Zej        d	             Zd
 Zd Zd Zd Zd Zd Zd Zej                            ddgddgg          d             Z d Z!ej                            dg d          d             Z"d Z#d Z$d Z%d Z&ej                            dddg          d             Z'ej                            d d! d" gd#d$g%          d&             Z(d' Z)d( Z*d) Z+ej        d*             Z,ej        d+             Z-ej        d,             Z.ej        d-             Z/ej        d.             Z0ej        d/             Z1ej        d0             Z2ej        d1             Z3ej        d2             Z4ej        d3             Z5ej        d4             Z6ej        d5             Z7ej        d6             Z8ej        d7             Z9 ej        g d89          d:             Z:d; Z;ej                            d<d ej<        d= d>idej<        fd?d@ fd> edej<                   ed?dA           d>g          dB             Z=dC Z>ej                            d<d dDdEgidDdEdFg          dG             Z?ej                            d<ddDdEdFidDdEdFdDdEdHd>g          dI             Z@dJ ZAej                            d<d ejB        dK d>idejB        fd?dL fd> edejB                   ed?dM           d>g          dN             ZCej                            d<d ejB        ej<        dOidPejB        fdQej<        fd> edPejB                   edQej<                  d>g          dR             ZDej                            dSddEdTgdDdTgd>gdd?gdEdTgdDdTgd>gg          dU             ZEej                            d<ddEdTgidEdTgdDdTgd>g          dV             ZFdW ZGdX ZHdY ZIdZ ZJd[ ZKej                            d\g d]          d^             ZLd_ ZMej                            d`dadbdcg dddedfdg dgdhdif
djdkdldmgdnz  dcdfdog dpdqdrf
dadbdcg dddedfdsg dtdudvf
dadbdcg dddedwdg dxdydif
g          dz             ZNej                            d{g dEd|d}d~gifdEddgd~gdfdEejO        dgd~gdfdd|d}dgifddejO        dfdd|d}dvgifdddgdvgdfdejO        dgdvgdfdd|d}dgifdddgdgdfdejO        dgdgdfdd|d}dvgifdddgdvgdfdejO        dgdvgdfdd|d}dgifdddgdgdfdejO        dgdgdfdDd|d}dgifddDejO        dfdd|d}dgifddejO        dfdTd|d}dgifddTejO        dfdd|d}dgifddejO        dfdd|d}dgifddejO        df          d             ZPej                            d{ddEddgfdEejO        dgfdddejO        dfddddgfdejO        dgfddddgfdejO        dgfddddgfdejO        dgfddddgfdejO        dgfg          d             ZQd ZRd ZSdS )    )datetimeN)lib)Day)	DataFrameNamedAggSeries)
date_rangec                  b    t          t          ddd          t          ddd          d          S )N     
   Min)startendfreq)r	   r        {/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/pandas/tests/resample/test_resample_api.pydtir      s0    HT1a00htQ6K6KRWXXXXr   c                     t          t          j                            d                              t	          |                     |           S )N   )r   nprandomdefault_rnglen)r   s    r   _test_seriesr      s6    ")''**11#c((;;SAAAr   c           	      f    t          ||t          j        t          |                     d          S )NABC)r   r   aranger   )r   r   s     r   
test_framer#      s*    <l3s88ATATUUVVVr   c                     |                      d          }dt          |          v sJ |                      dd          }dt          |          v sJ d S )NhzaDatetimeIndexResampler [freq=<Hour>, closed=left, label=left, convention=start, origin=start_day]
2000-01-01originzkDatetimeIndexResampler [freq=<Hour>, closed=left, label=left, convention=start, origin=2000-01-01 00:00:00])resamplestr)r   rs     r   test_strr,   #   s    c""A	:=@VV	D 	D 	D 	D 	c,77A	DGJ1vv	N 	N 	N 	N 	N 	Nr   c                 ~   |                      d          }|                                }t          |t                    sJ t	          |          dk    sJ |                                                      d          }|                                }t          |t                    sJ t	          |          dk    sJ d S )Nr%      )r)   mean
isinstancer   r   to_framer   )r   r+   results      r   test_apir3   1   s    c""AVVXXFff%%%%%v;;#((--AVVXXFfi(((((v;;#r   c                  >   t          t          ddd          g dg dd                              d          } t          d	d
                                          t          dd
                                          z   }t          j                            dgd
z  dgd
z  z   |gddg          }t          ddgdz  dgz   dgdz  z   d
gz   i|          }|                     d                              d           dg         }t          j
        ||           d S )N
2016-01-01   W)r   periodsr   )r   r   r   r   )            )dategroupvalr=   z
2016-01-03r<   r8   z
2016-01-17r   r   r>   namesr?   r9   r;   r:   indexc                 P    |                      d                                          S )N1D)r)   ffillxs    r   <lambda>z+test_groupby_resample_api.<locals>.<lambda>P   s    D1A1A1G1G1I1I r   )r   r	   	set_indextolistpd
MultiIndexfrom_arraysgroupbyapplytmassert_frame_equal)dfirD   expectedr2   s        r   test_groupby_resample_apirW   =   sB    
\13GGG!\\<<	
 	

 
 i  	<+++2244
\1
-
-
-
4
4
6
6	7  M%%sQw!q'8!&<WfDU%VVE%!qA3!q!8A3!>?uMMMHZZ  &&'I'IJJE7SF&(+++++r   c                     t          ddgdz  t          dd          t          j                            d                              d          d          } |                     d	                              d
                              d          	                                }|                     d
                              dd	          	                                }t          j        ||           d S )Nr   r    r9   r5   r   r@   r   )keydatesvaluesrZ   rY   Don)r   r	   r   r   r   standard_normalrK   rP   r)   r/   rR   rS   rT   rV   r2   s      r   test_groupby_resample_on_apira   T   s     
:>b999i++A..>>rBB	
 	

 
B ||G$$,,U33<<SAAFFHHHZZ'''88==??F&(+++++r   c                     t          dddt          ddd                    } |                                 }|                     d	d
          }|                    d           }t          j        ||           |                     d	          }|                    d           }t          j        ||           t          j        	                    t          j
        ddg                              d                              d          |j        g          |_        |                     d	d          }|                    d           }t          j        ||           d S )Nr   r   r   r    2000r   ns)r8   unitrC   5DF
group_keysc                     | S Nr   rH   s    r   rJ   z*test_resample_group_keys.<locals>.<lambda>k       q r   c                     | S rk   r   rH   s    r   rJ   z*test_resample_group_keys.<locals>.<lambda>p   rl   r   r&   z
2000-01-06r9   Tc                     | S rk   r   rH   s    r   rJ   z*test_resample_group_keys.<locals>.<lambda>{   rl   r   )r   r	   copyr)   rQ   rR   rS   rM   rN   rO   to_datetimeas_unitrepeatrD   )rT   rV   gr2   s       r   test_resample_group_keysrt   e   sD   	##:fbt+T+T+T	U	U	UBwwyyH 	DU++AWW[[!!F&(+++ 	DAWW[[!!F&(+++ ]..NL,788@@FFMMaPPN	
 HN 	DT**AWW[[!!F&(+++++r   c                    |                     d          }|                                |                                z
  }|                    d           }t	          j        ||           |                      d          }|                                |                                z
  }|                    d           }t	          j        ||           d S )Nr%   c                 T    |                                  |                                 z
  S rk   maxr/   rH   s    r   rJ   ztest_pipe.<locals>.<lambda>       aeegg0 r   c                 T    |                                  |                                 z
  S rk   rw   rH   s    r   rJ   ztest_pipe.<locals>.<lambda>   ry   r   )r)   rx   r/   piperR   assert_series_equalrS   )r#   r   r+   rV   r2   s        r   	test_piper}      s     	c""Auuww!HVV0011F68,,, 	C  Auuww!HVV0011F&(+++++r   c                    |                      d          }t          j        |j        j        | j                   |                      d          d         }|j        j        | j        d         k    sJ |                      d          d         }t          j        |j        j        | j        ddg                    |                      d          d         }t          j        |j        j        | j        ddg                    d S )Nr%   r    r   rc   r   )r)   rR   assert_index_equal_selected_objcolumnsnamer#   r+   s     r   test_getitemr      s    C  A!/1:3EFFFC  %A?:#5a#88888 	C  *A!/1:3Eq!f3MNNNC  *A!/1:3Eq!f3MNNNNNr   rY   r\   r   c                     |                     d          }d}t          j        t          |          5  ||           d d d            d S # 1 swxY w Y   d S )Nr%   z^\"Columns not found: 'D'\"$match)r)   pytestraisesKeyError)rY   r#   rs   msgs       r   test_select_bad_colsr      s    C  A *C	xs	+	+	+  	#                 s   	A		AAc                     |                      d          }t          j        |j                                        |d                                                    d S )Nr%   r   )r)   rR   r|   r   sumr   s     r   test_attribute_accessr      sD    C  A137799afjjll33333r   attr)groupsngroupsindicesc                    t          ddd          }t          t          j        t	          |                    |          }|                    d          }t          ||            |                                 t          ||            d S )N1/1/2012d   sr8   r   rC   30s)r	   r   r   r"   r   r)   getattrr/   )r   rngtsrss       r   test_api_compat_before_user      s     Z3
7
7
7C		#c((##3	/	/	/B	U		B B GGIIIBr   c                 x   | }d|d<   |                     d          }|ddg                                         }t          j        |j                                        |j                                        gd          }t          j        ||           |g d                                         }t          j	        d	          }|rd
}t          j        t          |          5  |                                 d d d            n# 1 swxY w Y   |                    d          }t          j        ||           d S )Nfoor\   r%   r   r    r   axisr   z&agg function failed [how->mean,dtype->z-dtype 'str' does not support operation 'mean'r   Tnumeric_only)r)   r/   rM   concatr   r    rR   rS   reescaper   r   	TypeError)r#   using_infer_stringrT   r+   r2   rV   r   s          r   tests_raises_on_nuisancer      sT   	BBsG
CASz]!!Fy!#((**achhjj1:::H&(+++!&&((H
)<
=
=C >=	y	,	,	,  	              VVV&&F&(+++++s   #DDDc                  L   t          ddd          } t          t          j        t	          |           d          |           }|                    d                                          }t          g d	t          d
dd                    }t          j        ||           d S )Nr   r   r   r   int64dtyperC   20s)r      (   <   P   2012-01-01 00:00:00r9   r   r8   )	r	   r   r   r"   r   r)   asfreqrR   r|   )r   r   r2   rV   s       r   'test_downsample_but_actually_upsamplingr      s    
Z3
7
7
7C		#c(('222#	>	>	>B[[&&((F.UAFFF  H 68,,,,,r   c                     t          dddd          } t          t          j        t	          |                     |           }|j        g d         }|                    d                                                                          }t          g d	t          j
        g d
dd                    }t          j        ||           d S )Nr   r   r   re   )r8   r   rf   rC   )r   r   r      r9   r;                  2s)g      ?g      @g      @      @r         &@r   g      .@      0@r   r   r         9@r   r   g      >@)r   z2012-01-01 00:00:02z2012-01-01 00:00:04z2012-01-01 00:00:06z2012-01-01 00:00:08z2012-01-01 00:00:10z2012-01-01 00:00:12z2012-01-01 00:00:14z2012-01-01 00:00:16z2012-01-01 00:00:18z2012-01-01 00:00:20z2012-01-01 00:00:22z2012-01-01 00:00:24z2012-01-01 00:00:26z2012-01-01 00:00:28z2012-01-01 00:00:30zdatetime64[ns]r   r   )r	   r   r   r"   r   ilocr)   r/   rG   rM   DatetimeIndexrR   r|   )r   r   ts2r2   rV   s        r   *test_combined_up_downsampling_of_irregularr      s    
 Z3T
B
B
BC		#c((##3	/	/	/B
'888
9C\\$$$&&,,..F	
 	
 	
$   $ #)
 
 
') ) )HT 68,,,,,r   c                     |                      d          }|                     t          j        d                                        d          }|                    d          }t          j        ||           d S )N20minr   r/   )r)   rP   rM   Grouper	transformrR   r|   )r   r+   rV   r2   s       r   test_transform_seriesr     sm    g&&A##BJG$<$<$<==GGOOH[[  F68,,,,,r   r^   r=   c                 j   t          t          ddd          t          ddd          d          }d|_        t          t          j                            d                              d          t          d	          |
          }|                    t          j
        d                                        d          }| dk    r*|                    d          }|                                }|                    d|           }|                    d          }t          j        ||           d S )Nr   r   r   r\   r   r=   r   r   r   ABr   rD   r   r/   T)dropr]   )r	   r   r   r   r   r   r   listrP   rM   r   r   reset_indexr)   rR   rS   )r^   rD   rT   rV   r+   r2   s         r   test_transform_framer     s    xa++XdAr-B-BMMMEEJ	
	a  ''00$t**E
 
 
B zz"*'22233==fEEH	V||''T'22^^
G##A[[  F&(+++++r   funcc                 0    |                      dd          S )Nr   Frh   r)   rH   s    r   rJ   rJ   /  s    !**W*77 r   c                 V    |                      t          j        d          d          S )Nr   r   Frh   rP   rM   r   rH   s    r   rJ   rJ   0  s#    !))BJG444)GG r   r)   rP   )idsc                 r     | |          }|                     d           }t          j        ||           d S )Nc                     | S rk   r   rH   s    r   rJ   z0test_apply_without_aggregation.<locals>.<lambda>7  rl   r   )rQ   rR   r|   )r   r   tr2   s       r   test_apply_without_aggregationr   ,  s?     	\AWW[[!!F6<00000r   c                     |                      d                              dd          }|d                             d           }t          j        ||                     d                     d S )Nr   )r   r   Frh   c                     | S rk   r   rH   s    r   rJ   z1test_apply_without_aggregation2.<locals>.<lambda>=  s    A r   )r1   r)   rQ   rR   r|   rename)r   groupedr2   s      r   test_apply_without_aggregation2r   ;  sk    ###//88U8SSGU^!!++..F6<#6#6u#=#=>>>>>r   c                  t   t          t          j                            d                              d          t          ddd          g d          } |                     d	          }d
}t          j        t          |          5  |
                    ddd           d d d            d S # 1 swxY w Y   d S )Nr   )  r   r   r   r   r   r   rD   r   3minz&Label\(s\) \['r1', 'r2'\] do not existr   r/   r   r1r2r   r   r   r   r_   r	   r)   r   r   r   aggrT   r+   r   s      r   test_agg_consistencyr   A  s     

	a  00;;#t<<<
 
 
B 	FA
3C	xs	+	+	+ + +	V5))***+ + + + + + + + + + + + + + + + + +   B--B14B1c                  t   t          t          j                            d                              d          t          ddd          ddg	          } |                     d
          }d}t          j        t          |          5  |
                    ddd           d d d            d S # 1 swxY w Y   d S )Nr   )r   r   r   r   r   r   r   ar   r   z"Label\(s\) \[2, 'b'\] do not existr   r/   r   )r   br   r   s      r   'test_agg_consistency_int_str_column_mixr   Q  s    	
	a  00;;#t<<<C
 
 
B 	FA
/C	xs	+	+	+ ' '	&u%%&&&' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' 'r   c                  v    t          t          ddd          t          ddd          dd          } d| _        | S )Nr   r   r   r\   re   r   rf   r=   )r	   r   r   rC   s    r   rD   rD   d  s=    xa++XdAr-B-BSWXXXEEJLr   c                     t          t          j                            d                              d          t	          d          |           }|S )Nr   r   r   r   )r   r   r   r   r   )rD   frames     r   rT   rT   k  sH    
	a  ''00$t**E  E Lr   c                 *    |                                  S rk   )r   rT   s    r   df_colr   s  s    >>r   c                     |                                  }t          j                            t	          d          |gddg          |_        |S )Nr   rD   r=   rA   )ro   rM   rN   rO   rangerD   )r   rD   df_mults      r   r   r   x  sI    kkmmGM--	rE7F"3 .  GM Nr   c                 \    |                      d          d                                         S N2Dr   r)   r/   r   s    r   a_meanr    %    ;;tS!&&(((r   c                 \    |                      d          d                                         S r  r)   stdr   s    r   a_stdr	    %    ;;tS!%%'''r   c                 \    |                      d          d                                         S r  r)   r   r   s    r   a_sumr    r
  r   c                 \    |                      d          d                                         S Nr  r    r  r   s    r   b_meanr    r  r   c                 \    |                      d          d                                         S r  r  r   s    r   b_stdr    r
  r   c                 \    |                      d          d                                         S r  r  r   s    r   b_sumr    r
  r   c                 ,    |                      d          S )Nr  r   r   s    r   df_resampler    s    ;;tr   c                 0    |                      dd          S )Nr  r=   r]   r   )r   s    r   df_col_resampler    s    ??4F?+++r   c                 0    |                      dd          S )Nr  r=   levelr   )r   s    r   df_mult_resampler    s    D///r   c                 R    |                      t          j        d                    S )Nr  r   r   r   s    r   df_grouper_resampler    s!    ::bjd+++,,,r   )r  r  r  r  )paramsc                 6    |                      | j                  S rk   )getfixturevalueparam)requests    r   casesr$    s     ""7=111r   c                    t          j        ||||gd          }t           j                            ddgddgg          |_        d|j        j        j        v rw| d                                         }| d         	                                }t          j        |||gd          }t           j                            g d	ddgg          |_        | 
                    t          j        d
 g          }	t          j        |	|           d S )Nr   r   r   r    r/   z
<lambda_0>r   r=   )r=   r   r    c                 .    t          j        | d          S Nr   )ddofr   r  rH   s    r   rJ   z3test_agg_mixed_column_aggregation.<locals>.<lambda>  s    1B1B1B r   )rM   r   rN   from_productr   nodecallspecidr/   r  	aggregater   rR   rS   )
r$  r  r	  r  r  r#  rV   	date_meandate_stdr2   s
             r   !test_agg_mixed_column_aggregationr1    s   y&%7a@@@H}11C:?U2VWWHGL),,,&M&&((	=$$&&9i8<1EEE=55&,!78
 
 __bg'B'BCDDF&(+++++r   r   c                 .    t          j        | d          S r'  r)  rH   s    r   rJ   rJ     s    rvaa/@/@/@ r   rc   r    c                 .    t          j        | d          S r'  r)  rH   s    r   rJ   rJ     s    26!!3D3D3D r   c                 .    t          j        | d          S r'  r)  rH   s    r   rJ   rJ     s    26!RSCTCTCT r   c                 ~    t          j        ||gd          } | j        di |}t          j        ||d           d S )Nr   r   T
check_liker   )rM   r   r.  rR   rS   )r$  r  r  r   rV   r2   s         r   #test_agg_both_mean_std_named_resultr8    sQ     y&%q111HU_##s##F&(t<<<<<<r   c                     t          j        ||gd          }t           j                            ddg          |_        |                     dddgi          }t          j        ||           d S )Nr   r   r   r/   r   r  r   r/   r  rM   r   rN   from_tuplesr   r.  rR   rS   )r$  r  r	  rV   r2   s        r   #test_agg_both_mean_std_dict_of_listr>    sj    y&%q111H}00-1NOOH__cFE?344F&(+++++r   r/   r   )r/   r   c                     t          j        ||gd          }ddg|_         | d         j        di |}t	          j        ||           d S )Nr   r   r/   r   r   r   )rM   r   r   r.  rR   rS   )r$  r  r  r   rV   r2   s         r   test_agg_both_mean_sumr@    s]     y&%q111HH!U3Z!((C((F&(+++++r   )mean2sum2c                     d}t          j        t          j        j        |          5  |                     |           d d d            d S # 1 swxY w Y   d S )Nnested renamer is not supportedr   )r   r   rM   errorsSpecificationErrorr.  r$  r   r   s      r   (test_agg_dict_of_dict_specificationerrorrH    s     ,C	ry33	?	?	?                   s   AAAc                     t          j        ||||gd          }t           j                            g d          |_        |                     ddgddgd          }t          j        ||d           d S )	Nr   r   )r:  r;  r    r/   r    r  r/   r  rc   Tr6  r<  )r$  r  r	  r  r  rV   r2   s          r   test_agg_dict_of_listsrL    s    y&%7a@@@H}00BBB H __FE?&%IIJJF&(t<<<<<<r   c                 .    t          j        | d          S r'  r)  rH   s    r   rJ   rJ     s    bfQQ.?.?.? r   c                 .    t          j        | d          S r'  r)  rH   s    r   rJ   rJ     s    "&2C2C2C r   c                 .    t          j        | d          S r'  r)  rH   s    r   rJ   rJ     s    "&QRBSBSBS r   c                     | d                              d           }t          j        | d                                         |gd          } | j        di |}t          j        ||d           d S )	Nr    c                 .    t          j        | d          S r'  r)  rH   s    r   rJ   z&test_agg_with_lambda.<locals>.<lambda>  s    ):):): r   r   r   r   Tr6  r   )rQ   rM   r   r   r   rR   rS   )r$  r   rcustomrV   r2   s        r   test_agg_with_lambdarS    s{     Cj::;;Gy%*..**G41===HUYF&(t<<<<<<r   )result1result2rT  rU  c                     d}t          j        t          |          5   | ddg         j        di | d d d            d S # 1 swxY w Y   d S )Nz0Label\(s\) \['result1', 'result2'\] do not existr   r   r    r   r   r   r   r   rG  s      r   test_agg_no_columnrX    s     >C	xs	+	+	+ % %sCj$$$$$% % % % % % % % % % % % % % % % % %s   AAAz	cols, aggr  c                     t          j        ||||gd          }t           j                            g d          |_        |	| |         }n| }|                    |          }	t          j        |	|d           d S )Nr   r   ))r   r   r;  rJ  rK  Tr6  )rM   r   rN   r=  r   r   rR   rS   )
r$  colsr   r  r	  r  r  rV   objr2   s
             r   "test_agg_specificationerror_nestedr\    s      y%6Q???H}00AAA H DkWWS\\F&(t<<<<<<r   c                     d}t          j        t          j        j        |          5  | d                             |           d d d            d S # 1 swxY w Y   d S )NrD  r   r   )r   r   rM   rE  rF  r   rG  s      r   "test_agg_specificationerror_seriesr^  <  s     ,C 
ry33	?	?	?  c
s                 s   AAAc                     d}t          j        t          |          5  | dg                             ddgddgd           d d d            d S # 1 swxY w Y   d S )NzLabel\(s\) \['B'\] do not existr   r   r   r  r/   rc   rW  )r$  r   s     r   )test_agg_specificationerror_invalid_namesr`  G  s     -C	xs	+	+	+ F Fseu~VUODDEEEF F F F F F F F F F F F F F F F F Fs   $AAAc                     t          t          ddd          t          ddd          d          } d| _        t          t          j                            d                              d          t          d	          | 
          }|                                }|	                                }t          j                            t          d          |j        gddg          |_        |                    d          }||                    dd          |                    dd          |                    t          j        d                    g}d}|D ]_}t%          j        t          j        j        |          5  |                    dddgidddgid           d d d            n# 1 swxY w Y   `|D ]}t%          j        t          j        j        |          5  |ddg                             dddgidddgid           d d d            n# 1 swxY w Y   t%          j        t          j        j        |          5  |                    dddgidddgid           d d d            n# 1 swxY w Y   d S )Nr   r   r   r\   r   r=   r   r   r   r   rD   rA   r  r]   r  rD  r   r   r/   r   r    r   rar  rbrc   )r	   r   r   r   r   r   r   r   r   ro   rM   rN   rO   r   rD   r)   rP   r   r   r   rE  rF  r.  r   )rD   rT   r   r   r+   r$  r   r   s           r   test_agg_nested_dictsrd  O  sh   xa++XdAr-B-BMMMEEJ	
	a  ''00$t**E
 
 
B ^^FkkmmGM--	rBHgv%6 .  GM 	DA	((V,,


2:4((())	E ,C V V]297sCCC 	V 	VKKfe_5cFE?=STTUUU	V 	V 	V 	V 	V 	V 	V 	V 	V 	V 	V 	V 	V 	V 	V  P P]297sCCC 	 	sCjMfe_-TFE?4KLL  	 	 	 	 	 	 	 	 	 	 	 	 	 	 	
 ]297sCCC 	P 	PEE/tfe_6MNNOOO	P 	P 	P 	P 	P 	P 	P 	P 	P 	P 	P 	P 	P 	P 	PP Ps6   !F..F2	5F2	#)HH	H	!I55I9	<I9	c                     t          dddd          dddt          dddd          dddt          dddd          d	d
dg} t          |                               d          }d}t          j        t
          |          5  |                    d                              dgdgdgd           d d d            d S # 1 swxY w Y   d S )Ni  r:   r   r   g      ?g       @)dtrI   yr   g      @g      ?rf  zLabel\(s\) \['z'\] do not existr   30minr/   medianr   )rI   rg  z)r   r   rK   r   r   r   r)   r   )datarT   r   s      r   &test_try_aggregate_non_existing_columnrl  q  s;    aA&&Ss;;aA&&Ss;;aA&&Ss;;D
 
4	"	"4	(	(B -C	xs	+	+	+ Q Q
G  xxjw!O!OPPPQ Q Q Q Q Q Q Q Q Q Q Q Q Q Q Q Q Qs   0B>>CCc                  >   t          dg dit          ddd                    } dd
}dd}d}t          j        t          |          5  |                     d                              ||gddd           d d d            n# 1 swxY w Y   |                     d                              ||gdd          }t          ddgddgddggt          ddd          t          j        	                    ddg                    }t          j        ||           d S )NrI   )r   r   r   
2020-01-01r   r\   r   rC   r   r   c                     | |z   |z   S rk   r   )rI   r   cs      r   foo1z/test_agg_list_like_func_with_args.<locals>.foo1      1uqyr   r   c                     | |z   |z   S rk   r   )rI   r   rp  s      r   foo2z/test_agg_list_like_func_with_args.<locals>.foo2  rr  r   z/foo1\(\) got an unexpected keyword argument 'b'r   r6   )r   rp  )rp  r<   	   r   )rI   rq  )rI   rt  r   )r   r   )r   r   )r   r	   r   r   r   r)   r   rM   rN   r=  rR   rS   )rT   rq  rt  r   r2   rV   s         r   !test_agg_list_like_func_with_argsrv    s   		iii
< M M M
 
 
B       =C	y	,	,	, 8 8
CdD\1Q7778 8 8 8 8 8 8 8 8 8 8 8 8 8 8 [[!!4,Q!77F
Q!Q"b"qs;;;))=-*HII  H
 &(+++++s   /BB
B
c                  d   t          t          ddd          t          ddd          d          } t          j        t	          |           t          j                  }t          | |dt          j        	                    || gdd	g
                    }t          d|i|           }d}t          j        t          |          5  |                    dd           d d d            n# 1 swxY w Y   d}t          j        t          |          5  |                    ddd	           d d d            n# 1 swxY w Y   d}t          j        t          |          5  |                    dddg           d d d            n# 1 swxY w Y   d}t          j        t          |          5  |                    dddg           d d d            n# 1 swxY w Y   d}t          j        t          |          5  |                    dd	                                           d d d            n# 1 swxY w Y   t          j        t          |          5  |                    dd                                           d d d            n# 1 swxY w Y   |                    d                                          }d|j        _        |                    dd                                          }t)          j        ||           d	|j        _        t          j        t          d          5  |                    dd	                                           d d d            n# 1 swxY w Y   |                    dd	                              d          }t)          j        ||           d S )Nr   r   r   r\   r   r   )r=   r   vdrA   rC   r   z\Only valid with DatetimeIndex, TimedeltaIndex or PeriodIndex, but got an instance of 'Index'r   r  r  z2The Grouper cannot specify both a key and a level!r=   )r^   r  zunhashable type: 'list'r]   z#\"Level \['a', 'date'\] not found\"z|Upsampling from level= or on= selection is not supported, use \.set_index\(\.\.\.\) to explicitly set index to datetime-likez0datetime64 type does not support operation 'sum'Tr   )r	   r   r   r"   r   r   r   rM   rN   rO   r   r   r   r)   
ValueErrorr   r   r   rD   r   rR   rS   )rD   r   rT   df_expr   expr2   s          r   test_selection_api_validationr}    s-   xa++XdAr-B-BMMME
)CJJbh
/
/
/C	S!!m''eS#J'GG
 
 
B Sz///F	)  
y	,	,	, % %
D$$$% % % % % % % % % % % % % % % ?C	z	-	-	- 0 0
DV3///0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 $C	y	,	,	, , ,
Dc6]+++, , , , , , , , , , , , , , , 1C	xs	+	+	+ / /
Df.../ / / / / / / / / / / / / / /
	J  
z	-	-	- . .
D$$++---. . . . . . . . . . . . . . .	z	-	-	- . .
DV$$++---. . . . . . . . . . . . . . . //$


#
#
%
%CCIN[[&[))--//F#v&&&CIN	K
 
 
 + + 	D$$((***+ + + + + + + + + + + + + + + [[S[))--4-@@F#v&&&&&s~   C))C-0C-D66D:=D:FFF,GGG:*H00H47H4*JJJ/*M%%M),M)col_name)t2t2xt2qT_2Mt2pt2mt2m1T2Mc           	         t          t          t          d                    t          dddd          | g          }|                    d                              dg          }t          g d	t          dd
dd          t          j        | gdggdgdgg                    }t          j	        ||           d S )N   z
2017-01-0115minzEurope/Berlin)r   r   r8   tzr   rF   r/   )g     G@g     a@g     ph@r\   r   r   )levelscodes)
r   r   r   r	   r)   r.  rM   rN   rR   rS   )r~  rT   r2   rV   s       r   *test_agg_with_datetime_index_list_agg_funcr    s     
U3ZZWco
 
 
 

 
 
B [[((&22F|#q_UUUxj6(%;QC!:NNN  H
 &(+++++r   c                     t          dddd          } t          j        |           }|                    d           t	          ||           }|                    d	          }t	          t          j        d
          t          j        d
          g| d d d                   }t          d          |j	        _
        |                    d          }t          j        ||           |                    d          }t          j        ||           |                    d          }t          j        ||           |                    d          }t          j        ||           d S )Nrn  z
2020-01-021hre   r   F)writerC   rF   r      r   lastfirstrx   min)r	   r   
zeros_likesetflagsr   r)   rM   	Timestampr   rD   r   r   rR   r|   )rD   arrserr   rV   r2   s         r   test_resample_agg_readonlyr    s-   |\4HHHE
-

CLLuL
E
"
"
"C	d		Br|AQ8dddLLLHa&&HNVVF^^F68,,,VVG__F68,,,VVE]]F68,,,VVE]]F68,,,,,r   zLstart,end,freq,data,resample_freq,origin,closed,exp_data,exp_end,exp_periodsz2000-10-01 23:30:00z2000-10-02 00:26:007min)	r   r   r:   ru     r         r  17minr   )r   r     ?   z20001002 00:26:00r6   z20200101 8:26:35z20200101 9:31:5877sr   3   right)
r   r:   r9   r:   r9   r:   r9   r:   r9   r:   z2020-01-01 09:30:45r   left)r   r  r  '   r  z20001002 00:43:00r9   end_day)r   r   -   r  z2000-10-02 00:29:00c
                    t          | ||          }
t          ||
          }|                    |||                                          }t          |t          |||	                    }t	          j        ||           d S )Nr   rC   )r(   closed)r   r   r8   )r	   r   r)   r   rR   r|   )r   r   r   rk  resample_freqr(   r  exp_dataexp_endexp_periodsr   r   resrV   s                 r   test_end_and_end_day_originr    s    B UCd
+
+
+C	C	 	 	 B
++mF6+
B
B
F
F
H
HCW=+NNN  H
 3)))))r   z#method, numeric_only, expected_dataTnumr   F
cat_1cat_2catr  prodr   )r  Fcan't multiply sequencer  r  cat_1rx   r   cat_2r  r  g      )@)r/   FCould not convertr  ri  )ri  F-Cannot convert \['cat_1' 'cat_2'\] to numericr  gH6%@)r  F!could not convert string to floatr  varg      \@)r  Fr  semg      @)r  Fr  c                 |   t          ddd          }t          ddd          }t          dd	gd
dgd|          }|                    d          }|t          j        u ri }nd|i}t          ||           }	t          |t                    r| dv r(t          }
t          j
        d|  d          }|rd|  d}n| dv r|rt          }
d|  d}n	t          }
|}t          j        |
|          5   |	di |}d d d            d S # 1 swxY w Y   d S  |	di |}t          ||          }t          j        ||           d S )N
2018-01-01r   r\   r   
2018-12-31r   YEr  r  r9   r   r  rC   r   )r  r/   ri  r  zagg function failed [how->z,dtype->z(dtype 'str' does not support operation '')r   r  r  r   r   )r	   r   r)   r   
no_defaultr   r0   r*   r   r   r   rz  r   r   rR   rS   )methodr   expected_datar   rD   expected_indexrT   	resampledkwargsr   klassr   _r2   rV   s                  r   test_frame_downsample_methodr  O  s   X |QS999EadCCCN	GW-q"g>>e	L	L	LBD!!Is~%% ,/9f%%D-%% 0666E)IIIIJJC! KJJJJ,,,1C,EFVFFFCCEC]5,,, 	 	vA	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 ].AAA
fh/////s   +	DDD)r   Tr   )r  Tr   )r  Fr   r   )r  Tr   )rx   Tr   )r  Tr   )r  Tr   c                    t          ddd          }t          ddd          }t          dd	g|
          }|                    d          }|t          j        u ri nd|i}t          ||           }	|rQ|t          j        urCd|  }
t          j        t          |
          5   |	di | d d d            d S # 1 swxY w Y   d S | dk    rVt          j
        d          }
|rd}
t          j        t          |
          5   |	di | d d d            d S # 1 swxY w Y   d S  |	di |}t          ||
          }t          j        ||           d S )Nr  r   r\   r   r  r   r  r  r  rC   r   z1Cannot use numeric_only=True with SeriesGroupBy\.r   r  z&agg function failed [how->prod,dtype->z-dtype 'str' does not support operation 'prod'r   )r	   r   r)   r   r  r   r   r   r   r   r   rR   r|   )r  r   r  r   rD   r  rT   r  r  r   r   r2   rV   s                r   test_series_downsample_methodr    s   8 |QS999EadCCCN	'"%	0	0	0BD!!I3>11RR7UF9f%%D 1CN::K6KK]9C000 	 	DNN6NNN	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 		6		i@AA 	BAC]9C000 	 	DNN6NNN	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 -~>>>
vx00000s$   !	B77B;>B;=	DDDc                     t          t          j        g d                    } t          t          j        g d                    }|                     d                                          }t          j        ||           d S )N)2018-01-01 00:00:00z2018-01-01 12:00:002018-01-02 00:00:00rC   )r  z2018-01-01 08:00:00z2018-01-01 16:00:00r  8h)r   rM   rp   r)   r/   rR   rS   r`   s      r   test_resample_emptyr    s    	nQQQ
 

 
 
B
 n  
 
	 	 	H [[##%%F&(+++++r   c            
      >   t          dddddd          t          dddddd          t          dddddd          g} t          dg di|           }|                    d	d
                                          }t	          j        t	          j        d          t	          j        d          t	          j        d          g|j        j        d          }t          dt          j
        t          j
        t          j
        gi|          }t          j        ||           d S )Ni  r      r   r   value)r   r   r   rC   1min	start_dayr'   z2025-10-17 17:15:00z2025-10-17 17:16:00z2025-10-17 17:17:00r  r   )r   r   r)   r   rM   r   r  rD   r   r   nanrR   rS   )idxrT   r2   exp_idxr|  s        r   =test_asfreq_respects_origin_with_fixed_freq_all_seconds_equalr    s    	r2r2r**r2r2r**r2r2r**C
 
GYYY's	3	3	3B[[[44;;==F L.//L.//L.//	

 l   G Wrvrvrv67w
G
G
GC&#&&&&&r   )Tr   r   numpyr   r   pandas._libsr   pandas._libs.tslibsr   pandasrM   r   r   r   pandas._testing_testingrR   pandas.core.indexes.datetimesr	   fixturer   r   r#   r,   r3   rW   ra   rt   r}   r   markparametrizer   r   r   r   r   r   r   r   r   r   r   r   rD   rT   r   r   r  r	  r  r  r  r  r  r  r  r  r$  r1  r/   r8  r>  r@  rH  rL  r   rS  rX  r\  r^  r`  rd  rl  rv  r}  r  r  r  r  r  r  r  r  r   r   r   <module>r     s         				            # # # # # #             
       4 4 4 4 4 4 Y Y Y B B B W W W  	 	 	, , ,., , ,", , ,4, , , O O O #c
 344  544 4 4
 !A!A!ABB  CB, , ,$	- 	- 	-4- 4- 4-n- - - f~.., , /.,$ 
77GG 
Y   1 1 1? ? ?+ + + ' ' '&             ) ) ) ( ( ( ( ( ( ) ) ) ( ( ( ( ( (    , , , 0 0 0 - - - XXX  2 2 2, , , 		rw%@%@AABBGnC)D)D#EFFhsBG$$88C9T9T+U+UVV = = =, , , 	Vfe_%u'E'EF , , , 		ve,,- //!511	
 	
	 	 	 	= = = 		rv$?$?@@ABFm3(C(C"DEEhsBF##((38S8S*T*TUU = = = 		RV889"&!BG(<==hy"&))BG0L0LMM % % % 	eU^65/::;  %.77	
	 = = =  	S5%.!%vuo#N#NO   F F FP P PDQ Q Q, , ,24' 4' 4'n III , , ,,- - -2 R "!,,,OO	
 C"H***!	
 "!,,,	
 "!,,,OO!	
K14 4j* *k4 4j*0 )"	urdm$"	~rd;;<" 
t D DE" 
u~&	"
 	3" 
!:;" 
uqcl#" 
y!556" 
	1# > >?" 
urdm$" 
y"667" 
	2$ ? ?@" 
$%" 
%'A3778" 
#.7)QC"@"@A"  
t}%!"" 
	2$778#"$ 
'B4!@!@A%"& 
v''"( 	-)"* 
!45+", 
4%$)-". 	L/"0 
3>#ST1"2 
u12343"4 	<5"6 
 CD7"8 
uug&'9": 	<;"< 
 CD="> 
usen%?"@ 	<A"B 
 CDC"& &N 0  0O& &N 0F )	~&	/	$	y!		*	y!		*	%'#	#.7),		"	'+% .1 1/ .1:, , ,*' ' ' ' 'r   