
    dZ}                        d dl m 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 e ddd           e ddd	          d
          Z eej                             ee                    e          Z e
ee ej         ee                    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+d$ Z,ej"        #                    d%d& d' gd(d)g*          d+             Z-d, Z.d- Z/d. Z0d/ Z1d0 Z2ej"        #                    d%d1gd2d3gdd4id5d6d7g          d8             Z3d9 Z4d: Z5d; Z6d< Z7ej"        #                    d=g d>          d?             Z8d@ Z9ej"        #                    dAdBdCdDg dEdFdGdg dHdIdJf
dKdLdMdgdNz  dDdGdOg dPdQd	f
dBdCdDg dEdFdGdRg dSdTdUf
dBdCdDg dEdFdVdg dWdXdJf
g          dY             Z:ej"        #                    dZg d4d[d\d]gifd4d^d_gd]gd`fd4ej;        d_gd]gd`fd5d[d\dagifdbd5ej;        dcfd1d[d\dUgifd1d^ddgdUgd`fd1ej;        ddgdUgd`fd3d[d\degifd3d^dfgdegd`fd3ej;        dfgdegd`fdgd[d\dUgifdgd^ddgdUgd`fdgej;        ddgdUgd`fdhd[d\degifdhd^dfgdegd`fdhej;        dfgdegd`fd2d[d\digifdjd2ej;        dkfd6d[d\digifdld6ej;        dmfdnd[d\dogifdpdnej;        dqfdrd[d\dsgifdtdrej;        dqfdud[d\dvgifdwduej;        dqf          dx             Z<ej"        #                    dZdyd4d^d_gfd4ej;        d_gfdzd{d5ej;        d|fd}d1d^ddgfd1ej;        ddgfd~d3d^dfgfd3ej;        dfgfddgd^ddgfdgej;        ddgfddhd^dfgfdhej;        dfgfg          d             Z=ej"        #                    dg d          d             Z>d Z?dS )    )datetimeN)lib)UnsupportedFunctionCall)	DataFrameNamedAggSeries)
date_range     
   Min)startendfreqABCc                  4    t                                           S N)_test_framecopy     c/var/www/html/t/fyr/venv311/lib/python3.11/site-packages/pandas/tests/resample/test_resample_api.py
test_framer      s    r   c                      t                               d          } dt          |           v sJ t                               dd          } dt          |           v sJ d S )NHziDatetimeIndexResampler [freq=<Hour>, axis=0, closed=left, label=left, convention=start, origin=start_day]
2000-01-01)originzsDatetimeIndexResampler [freq=<Hour>, axis=0, closed=left, label=left, convention=start, origin=2000-01-01 00:00:00])test_seriesresamplestr)rs    r   test_strr%      s    S!!A	:=@VV	D 	D 	D 	D 	S66A	DGJ1vv	N 	N 	N 	N 	N 	Nr   c                     t                               d          } |                                 }t          |t                    sJ t          |          dk    sJ t                                                               d          } |                                 }t          |t                    sJ t          |          dk    sJ d S )Nr      )r!   r"   mean
isinstancer   lento_framer   )r$   results     r   test_apir-   +   s    S!!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      r3   )            )dategroupvalr8   z
2016-01-03r7   r2   z
2016-01-17r   r3   r9   namesr:   r4   r6   r5   indexc                 P    |                      d                                          S )N1D)r"   ffillxs    r   <lambda>z+test_groupby_resample_api.<locals>.<lambda>J   s    D1A1A1G1G1I1I r   )r   r	   	set_indextolistpd
MultiIndexfrom_arraysgroupbyapplytmassert_frame_equal)dfir?   expectedr,   s        r   test_groupby_resample_apirR   7   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                                          }t          j
        ||           d S )Nr   r   r4   r/   r   r;   )keydatesvaluesrU   rT   Don)r   r	   nprandomrandnrF   rK   r"   r(   rM   rN   rO   rQ   r,   s      r   test_groupby_resample_on_apir^   N   s     
:>b999ioob))	
 	

 
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           }t          j        ||           |                     d          }|                    d           }t          j        ||           t          j        	                    t          j
        ddg                              d          |j        g          |_        |                     dd
          }|                    d           }t          j        ||           d S )Nr   r3   r   r   2000r   r;   r>   5DF
group_keysc                     | S r   r   rC   s    r   rE   z*test_resample_group_keys.<locals>.<lambda>e       q r   c                     | S r   r   rC   s    r   rE   z*test_resample_group_keys.<locals>.<lambda>j   rf   r   r   z
2000-01-06r4   Tc                     | S r   r   rC   s    r   rE   z*test_resample_group_keys.<locals>.<lambda>r   rf   r   )r   r	   r   r"   rL   rM   rN   rH   rI   rJ   to_datetimerepeatr?   )rO   rQ   gr,   s       r   test_resample_group_keysrl   _   s1   	##:fb+I+I+I	J	J	JBwwyyH 	DU++AWW[[!!F&(+++ 	DAWW[[!!F&(+++ ]..	|4	5	5	<	<Q	?	?P HN 	DT**AWW[[!!F&(+++++r   c                    t                               d          }|                                |                                z
  }|                    d           }t          j        ||           |                     d          }|                                |                                z
  }|                    d           }t          j        ||           d S )Nr   c                 T    |                                  |                                 z
  S r   maxr(   rC   s    r   rE   ztest_pipe.<locals>.<lambda>|       aeegg0 r   c                 T    |                                  |                                 z
  S r   ro   rC   s    r   rE   ztest_pipe.<locals>.<lambda>   rq   r   )r!   r"   rp   r(   piperM   assert_series_equalrN   )r   r$   rQ   r,   s       r   	test_piperu   v   s     	S!!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   r`   r   )r"   rM   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   rT   rW   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)rT   r   rk   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"   rM   rt   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r2   r   r>   30s)r	   r   rZ   aranger*   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                 H   | }d|d<   |                     d          }|ddg                                         }t          j        |j                                        |j                                        gd          }t          j        ||           |g d                                         }t          j	        t          d	
          5  |                                 d d d            n# 1 swxY w Y   |                    d          }t          j        ||           d S )NfoorW   r   r   r   r   axisr   Could not convertr~   Tnumeric_only)r"   r(   rH   concatr   r   rM   rN   r   r   	TypeError)r   rO   r$   r,   rQ   s        r   tests_raises_on_nuisancer      s8   	BBsG
CASz]!!Fy!#((**achhjj1:::H&(+++!&&((H	y(;	<	<	<  	              VVV&&F&(+++++s   C,,C03C0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dtyper>   20s)r      (   <   P   2012-01-01 00:00:00r4   r   r2   )	r	   r   rZ   r   r*   r"   asfreqrM   rt   )r   r   r,   rQ   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          } t          t          j        t	          |                     |           }|j        g d         }|                    d                                                                          }t          g dt          j
        g d	d
d                    }t          j        ||           d S )Nr   r   r   r   r>   )r   r   r3      r4   r6                  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]2S)r   r   )r	   r   rZ   r   r*   ilocr"   r(   rB   rH   DatetimeIndexrM   rt   )r   r   ts2r,   rQ   s        r   *test_combined_up_downsampling_of_irregularr      s    
 Z3
7
7
7C		#c((##3	/	/	/B
'888
9C\\$$$&&,,..F	
 	
 	
$   $ #)
 
 
') ) )HT 68,,,,,r   c                     t                               d          } t                               t          j        d                                        d          }|                     d          }t          j        ||           d S )N20minr   r(   )r!   r"   rK   rH   Grouper	transformrM   rt   )r$   rQ   r,   s      r   test_transform_seriesr     sm    W%%A""2:7#;#;#;<<FFvNNH[[  F68,,,,,r   rY   r8   c                 F   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   rW   r   r8   r3   ABry   r?   r   r(   T)droprX   )r	   r   rz   r   rZ   r[   randlistrK   rH   r   r   reset_indexr"   rM   rN   )rY   r?   rO   rQ   r$   r,   s         r   test_transform_framer     s     xa++XdAr-B-BMMMEEJ	29>>"a(($t**E	J	J	JBzz"*'22233==fEEH	V||''T'22^^
G##A[[  F&(+++++r   c                  J   t          ddd          } t          t          j        t	          |           d          |           }|                    d          }|                                }|                    d	
          }t          j	        ||           |
                                }|                    d
          }t          j	        ||           d}t          j        t          |          5  |                    d           d d d            d S # 1 swxY w Y   d S )Nr   r   r   r   r   r   r>   srB   )methodbfillzTInvalid fill method\. Expecting pad \(ffill\), backfill \(bfill\) or nearest\. Got 0r~   r   )r	   r   rZ   r   r*   r"   rB   fillnarM   rt   r   r   r   
ValueError)r   r   r$   rQ   r,   r   s         r   test_fillnar     sH   
Z$
7
7
7C		#c(('222#	>	>	>B
CAwwyyHXXWX%%F68,,,wwyyHXXWX%%F68,,,	(  
z	-	-	-  	                 s   5DDDfuncc                 0    |                      dd          S )Nr   Frc   )r"   rC   s    r   rE   rE   7  s    !**W*77 r   c                 V    |                      t          j        d          d          S )Nr   r   Frc   )rK   rH   r   rC   s    r   rE   rE   8  s#    !))BJG444)GG r   r"   rK   )idsc                      | t                     }|                    d           }t          j        |t                      d S )Nc                     | S r   r   rC   s    r   rE   z0test_apply_without_aggregation.<locals>.<lambda>?  rf   r   )r!   rL   rM   rt   )r   tr,   s      r   test_apply_without_aggregationr   4  s?     	[AWW[[!!F6;/////r   c                      t                               d                              dd          } | d                             d           }t	          j        |t                               d                     d S )Nr   )rz   r   Frc   c                     | S r   r   rC   s    r   rE   z1test_apply_without_aggregation2.<locals>.<lambda>E  s    A r   )r!   r+   r"   rL   rM   rt   rename)groupedr,   s     r   test_apply_without_aggregation2r   C  sk    """..77E7RRGU^!!++..F6;#5#5e#<#<=====r   c                  P   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 )N  r   r   r   r   r   r?   ry   3Tz'Column\(s\) \['r1', 'r2'\] do not existr~   r(   r   r1r2
r   rZ   r[   r\   r	   r"   r   r   r   aggrO   r$   r   s      r   test_agg_consistencyr   I  s     

	a  #t<<<
 
 
B 	DA
4C	xs	+	+	+ + +	V5))***+ + + + + + + + + + + + + + + + + +   5BB"Bc                  P   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   r3   r   r   r   r   ar   r   z#Column\(s\) \[2, 'b'\] do not existr~   r(   r   )r3   br   r   s      r   'test_agg_consistency_int_str_column_mixr   Y  s    	
	a  #t<<<C
 
 
B 	DA
0C	xs	+	+	+ ' '	&u%%&&&' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' 'r   c                  
   t           j                            d           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                                         }|d                                         }|d                                         }	|d                                         }
|d                                         }t          j        |||	|
gd          }t          j                            ddgddgg          |_        |D ]}||d         k    r|d                                         }|d                                         }t          j        |||gd          }t          j                            g dddgg          |_        |                    t           j        t           j        g          }t5          j        ||           |                    t           j        t           j        g          }t5          j        ||           	t          j        ||
gd          }|D ]}|                    t           j        t           j        d          }t5          j        ||d           |                    dt           j        fdt           j        f          }t5          j        ||d           |                    t9          dt           j                  t9          dt           j                            }t5          j        ||d           t          j        ||gd          }t          j                            ddg          |_        |D ]0}|                    dddgi          }t5          j        ||           1t          j        ||gd          }ddg|_        |D ]f}|d                             ddg          }t5          j        ||           |d                             dd          }t5          j        ||           gd}|D ]Y}t=          j        t          j         j!        |          5  |                    ddddi           d d d            n# 1 swxY w Y   Zt          j        |||	|gd          }t          j                            g d          |_        |D ]]}t=          j        t          j         j!        |          5  |                    dddddd d           d d d            n# 1 swxY w Y   ^t          j        |||	|
gd          }t          j                            g d!          |_        |D ]5}|                    ddgddgd          }t5          j        ||d           6t          j        |||	|gd          }t          j                            g d"          |_        d S )#N  r
   r   r   rW   r   r8   r3   r   r   r?   r<   2DrX   levelr   r   r   r(   std)r8   r   r   r`   T
check_liker   r(   r   r   r   )r(   r   nested renamer is not supportedr~   )r   r   r   )r   mean2)r   sum2)r   r   )r   r   r   r(   r   r   ))r   r   r(   )r   r   r   )r   r   r(   )r   r   r   )"rZ   r[   seedr	   r   rz   r   r   r   r   r   rH   rI   rJ   ranger?   r"   rK   r   r(   r   r   r   from_productry   	aggregaterM   rN   r   from_tuplesr   r   errorsSpecificationError)r?   rO   df_coldf_multr$   casesa_meana_stda_sumb_meanb_stdb_sumrQ   r   	date_meandate_stdexpr,   r   s                      r   test_aggr  l  s    INN4xa++XdAr-B-BMMMEEJ	29>>"a(($t**E	J	J	JB^^FkkmmGM--	rBHgv%6 .  GM 	DA	((V,,


2:4((())	E sV[[]]FcFJJLLEcFJJLLEsV[[]]FcFJJLLEcFJJLLEy&%7a@@@H}11C:2OPPH 4 4a==&	((Iy}}H)Y(;!DDDC-44###fe_5 CK [["'26!233F!&#....[["'26!233F!&(3333y&%q111H A A278899
fh4@@@@RW~#rv??
fh4@@@@xRW55#rv9N9NOO
fh4@@@@@y&%q111H}00-1NOOH 0 0cFE?344
fh////y&%q111HH 0 03!!65/22
fh///3!!v5!99
fh////
+C ? ?]297sCCC 	? 	?KKve<<=>>>	? 	? 	? 	? 	? 	? 	? 	? 	? 	? 	? 	? 	? 	? 	? y&%7a@@@H}00DDD H   ]297sCCC 	 	KK"(77#)599   	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 y&%7a@@@H}00BBB H  A AFE?&%IIJJ
fh4@@@@@y&%7a@@@H}00	
 	
 	
 Hs$   8VV#	&V#	X??Y	Y	c            	      n   t           j                            d           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 ]}|                    t           j        d d          }|d                             d           }t          j        |d                                         |gd          }	t/          j        ||	d           |                    dt           j        fdd f          }t/          j        ||	d           |                    t3          dt           j                  t3          dd                     }t/          j        ||	d           t          j        |d                                         |d                                         |d                                         |d                                         gd          }	t          j                            g d          |	_        d}
|D ]V}t;          j        t>          |
          5  |ddg                             t           j        t           j        d           d d d            n# 1 swxY w Y   t;          j        t>          |
          5  |ddg                             dt           j        fdt           j        f           d d d            n# 1 swxY w Y   t;          j        t>          |
          5  |ddg                             t3          dt           j                  t3          dt           j                             d d d            n# 1 swxY w Y   Xt          j        |d                                         |d                                          |d                                         |d                                          gd          }	t          j                            g d           |	_        |D ]5}|                    d!d"gd#d"gd          }t/          j        ||	d           6|D ]=}|ddg                             d!d"gd#d"gd          }t/          j        ||	d           >d$}
|D ]}t;          j        t          j!        j"        |
          5  |d                             dd!d"gi           d d d            n# 1 swxY w Y   t;          j        t          j!        j"        |
          5  |d                             d!d"gd#d"gd           d d d            n# 1 swxY w Y   d%}
|D ]X}t;          j        t>          |
          5  |dg                             d!d"gd#d"gd           d d d            n# 1 swxY w Y   Yd S )&Nr   r
   r   r   rW   r   r8   r3   r   r   r?   r<   r   rX   r   c                 .    t          j        | d          S Nr   )ddofrZ   r   rC   s    r   rE   ztest_agg_misc.<locals>.<lambda>  s    BF114E4E4E r   r`   r   c                 .    t          j        | d          S r  r  rC   s    r   rE   ztest_agg_misc.<locals>.<lambda>  s    ):):): r   r   r   Tr   c                 .    t          j        | d          S r  r  rC   s    r   rE   ztest_agg_misc.<locals>.<lambda>  s    "&:K:K:K r   c                 .    t          j        | d          S r  r  rC   s    r   rE   ztest_agg_misc.<locals>.<lambda>  s    rvaa?P?P?P r   ))result1r   )r  r   )result2r   )r  r   z1Column\(s\) \['result1', 'result2'\] do not existr~   )r  r  r  r  )r   r   r   r  r   r   r(   r   z Column\(s\) \['B'\] do not exist)#rZ   r[   r  r	   r   rz   r   r   r   r   r   rH   rI   rJ   r  r?   r"   rK   r   r   r   rL   r   rM   rN   r   r(   r  ry   r   r   r   r   r  r  )r?   rO   r	  r
  r$   r  r   r,   rcustomrQ   r   s              r   test_agg_miscr!    s	    INN4xa++XdAr-B-BMMMEEJ	29>>"a(($t**E	J	J	JB^^FkkmmGM--	rBHgv%6 .  GM 	DA	((V,,


2:4((())	E  A ARV*E*EFFGGC&,,::;;9afjjllG41===
fh4@@@@#rv30K0K*LMM
fh4@@@@sBF##x5P5P'Q'Q  
 
 	fh4@@@@@ y	
3qvzz||QsV[[]]AcFKKMMB  H }00PPP H ?C 
 
]83/// 	G 	GsCjM"&RWEEFFF	G 	G 	G 	G 	G 	G 	G 	G 	G 	G 	G 	G 	G 	G 	G ]83/// 	M 	MsCjMBF 3	277KLLL	M 	M 	M 	M 	M 	M 	M 	M 	M 	M 	M 	M 	M 	M 	M ]83/// 	 	sCjM9bf--)RW1M1M    	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 y	
3qvzz||QsV[[]]AcFJJLLA  H }00AAA H  A AeU^65/BBCC
fh4@@@@@  A AC:""%vuo#N#NOO
fh4@@@@@
+C  D D]297sCCC 	. 	.cFJJeU^,---	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	. 	. ]297sCCC 	D 	DcFJJeU^65/BBCCC	D 	D 	D 	D 	D 	D 	D 	D 	D 	D 	D 	D 	D 	D 	D
 .C F F]83/// 	F 	FseHLLu~VUODDEEE	F 	F 	F 	F 	F 	F 	F 	F 	F 	F 	F 	F 	F 	F 	FF Fsm   5NN		N		+8O//O3	6O3	AQ11Q5	8Q5	 W--W1	4W1	#YY	Y	9$Z))Z-	0Z-	minr(   rp   r   prodmedianr`   c                    t           j                            d           t          t	          ddd          t	          ddd          d          }d|_        t          t           j                            dd          t          d	          |
          j	        }|
                    dd          }t          j        t          d          5  |                    |            d d d            d S # 1 swxY w Y   d S )Nr   r
   r   r   rW   r   r8   r3   r   r   Mr   z"axis other than 0 is not supportedr~   )rZ   r[   r  r	   r   rz   r   r   r   Tr"   r   r   NotImplementedErrorr   )r   r?   rO   ress       r   test_multi_agg_axis_1_raisesr*  2  s   
 INN4xa++XdAr-B-BMMMEEJ	29>>"a(($t**E	J	J	J	LB
++c+
"
"C	*2V	W	W	W                   s   C..C25C2c                      t           j                            d           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   rW   r   r8   r3   r   r   r?   r<   r   rX   r   r   r~   r   r(   r   r   r   rar   rbr`   )rZ   r[   r  r	   r   rz   r   r   r   r   r   rH   rI   rJ   r  r?   r"   rK   r   r   r   r  r  r  r   )r?   rO   r	  r
  r$   r  r   r   s           r   test_agg_nested_dictsr.  @  se   INN4xa++XdAr-B-BMMMEEJ	29>>"a(($t**E	J	J	JB^^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;;F?	F?	0)H%%H)	,H)	!JJ		J	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  r5   r   r   g      ?g       @)dtrD   yr3   g      @g      ?r0  z Column\(s\) \['z'\] do not existr~   30Tr(   r$  r   )rD   r1  z)r   r   rF   r   r   r   r"   r   )datarO   r   s      r   &test_try_aggregate_non_existing_columnr5  a  s;    aA&&Ss;;aA&&Ss;;aA&&Ss;;D
 
4	"	"4	(	(B .C	xs	+	+	+ O O
EfXXJeWMMNNNO O O O O O O O O O O O O O O O O O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 )NrD   )r   r3   r   
2020-01-01r   rW   r   r>   r   r   c                     | |z   |z   S r   r   )rD   r   cs      r   foo1z/test_agg_list_like_func_with_args.<locals>.foo1v      1uqyr   r3   c                     | |z   |z   S r   r   )rD   r   r9  s      r   foo2z/test_agg_list_like_func_with_args.<locals>.foo2y  r;  r   z/foo1\(\) got an unexpected keyword argument 'b'r~   r0   )r   r9  )r9  r7   	   r   )rD   r:  )rD   r=  r   )r   r   )r3   r   )r   r	   r   r   r   r"   r   rH   rI   r  rM   rN   )rO   r:  r=  r   r,   rQ   s         r   !test_agg_list_like_func_with_argsr?  p  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   rW   r   r   )r8   r   vdr<   r>   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!r8   )rY   r   zunhashable type: 'list'rX   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-likez$datetime64 type does not support sumTr   )r	   r   rZ   r   r*   r   r   rH   rI   rJ   r   r   r   r"   r   r   r   r   r?   rz   rM   rN   )r?   r   rO   df_expr   r  r,   s          r   test_selection_api_validationrD    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	y(N	O	O	O + +
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   r2   tzr   1dr(   )g     G@g     a@g     ph@rW   r   r   )levelscodes)
r   r   r  r	   r"   r  rH   rI   rM   rN   )rE  rO   r,   rQ   s       r   *test_agg_with_datetime_index_list_agg_funcrU    s     
U3ZZWco
 
 
 

 
 
B [[((&22F|#q_UUUxj6(%;QC!:NNN  H
 &(+++++r   c                     t          ddd          } t          j        |           }|                    d           t	          ||           }|                    d          }t	          t          j        d	          t          j        d	          g| d d d
                   }|                    d          }t          j
        ||           |                    d          }t          j
        ||           |                    d          }t          j
        ||           |                    d          }t          j
        ||           d S )Nr7  z
2020-01-021hr   F)writer>   rA   r      lastfirstrp   r"  )r	   rZ   
zeros_likesetflagsr   r"   rH   	Timestampr   rM   rt   )r?   arrserr   rQ   r,   s         r   test_resample_agg_readonlyra    s   |\===E
-

CLLuL
E
"
"
"C	d		Br|AQ8dddLLLH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   r5   r>     r         rY  17minr   )r   rd     ?   z20001002 00:26:00r0   z20200101 8:26:35z20200101 9:31:5877s3   right)
r   r5   r4   r5   r4   r5   r4   r5   r4   r5   z2020-01-01 09:30:45left)r   rd  rg  '   rY  z20001002 00:43:00r4   end_day)r   r   -   ro  z2000-10-02 00:29:00c
                    t          | ||          }
t          ||
          }|                    |||                                          }t          |t          |||	                    }t	          j        ||           d S )Nr   r>   )r    closed)r   r   r2   )r	   r   r"   r   rM   rt   )r   r   r   r4  resample_freqr    rq  exp_dataexp_endexp_periodsr   r   r)  rQ   s                 r   test_end_and_end_day_originrv    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catrw  r   )r#  Fcan't multiply sequencer{  cat_1r   cat_2r[  rZ  g      )@)r(   Fr   r   )r$  Fcould not convertr~  r   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                    rK| dv rt          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-01r3   rW   r   
2018-12-31r   Yr|  r}  r4   r   ry  r>   r   )r  r(   r$  r#  r~   r   )r	   r   r"   r   
no_defaultr   r)   r#   r   r   r   r   rM   rN   )r   r   expected_datar?   expected_indexrO   	resampledkwargsr   klass_r,   rQ   s                r   test_frame_downsample_methodr  <  s|   T |QS999EacBBBN	GW-q"g>>e	L	L	LBC  Is~%% ,/9f%%D-%% 0#'HHH		j]5666 	 	vA	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 ].AAA
fh/////s   4	C

CC)r   Tr   )r#  Tr   )r#  Fr   r   )r"  Tr   )rp   Tr   )r[  Tr   )rZ  Tr   c                 p   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    r>t          j        t          d          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  r3   rW   r   r  r   r  r|  r}  r>   r   z1Cannot use numeric_only=True with SeriesGroupBy\.r~   r#  z"can't multiply sequence by non-intr   )r	   r   r"   r   r  r   r   r   r   rM   rt   )r   r   r  r?   r  rO   r  r  r   r   r,   rQ   s               r   test_series_downsample_methodr  z  s   4 |QS999EacBBBN	'"%	0	0	0BC  I3>11RR7UF9f%%D 
1CN::K6KK]9C000 	 	DNN6NNN	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 		6		]9,PQQQ 	 	DNN6NNN	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 -~>>>
vx00000s$   !	B77B;>B;%	C;;C?C?zmethod, raises))r   T)r#  T)r"  T)rp   T)r[  F)rZ  F)r$  F)r(   T)r   T)r  T)r  F)ohlcF)nuniqueFc                    t          ddd          }t          g d|          }|                    d          }d}t          ||           }d	}d
}d|  }	|rwt	          j        t          |	          5  t          j        t          |          5   |g |dddR   d d d            n# 1 swxY w Y   d d d            d S # 1 swxY w Y   d S t	          j        t          |	          5  t          j        t          |          5   |g |dddR   d d d            n# 1 swxY w Y   d d d            d S # 1 swxY w Y   d S )N20180101r   hr   )r3   r0   r5   r>   30minr   z-numpy operations are not valid with resample.ztoo many arguments passed inz2Passing additional args to DatetimeIndexResampler.r~   r   r3   )r	   r   r"   r   rM   assert_produces_warningFutureWarningr   r   r   r   )
r   r   r?   rO   r  argsr   	error_msgerror_msg_typewarn_msgs
             r   test_args_kwargs_deprr    s   & z13777E					'	'	'BG$$ID9f%%D?I3NLFLLH %'XFFF 	% 	%6iHHH % %$d$A$q$!$$$$% % % % % % % % % % % % % % %	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 'XFFF 	% 	%y??? % %$d$A$q$!$$$$% % % % % % % % % % % % % % %	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	%sl   3CB*C*B.	.C1B.	2CC	C*D9D!D9!D%	%D9(D%	)D99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:00r>   )r  z2018-01-01 08:00:00z2018-01-01 16:00:00r  8H)r   rH   ri   r"   r(   rM   rN   r]   s      r   test_resample_emptyr    s    	nQQQ
 

 
 
B
 n  
 
	 	 	H [[##%%F&(+++++r   )@r   numpyrZ   r   pandas._libsr   pandas.errorsr   pandasrH   r   r   r   pandas._testing_testingrM   pandas.core.indexes.datetimesr	   dtir[   r   r*   r!   r   r   fixturer   r%   r-   rR   r^   rl   ru   r|   markparametrizer   r   r   r   r   r   r   r   r   r   r   r   r   r  r!  r*  r.  r5  r?  rD  rU  ra  rv  r  r  r  r  r  r   r   r   <module>r     s	                    1 1 1 1 1 1             
       4 4 4 4 4 4jxxa++$21F1FUSSSfRY^^CCHH--s33ik)")CCPSHHBUBUVVWW     	 	 	, , ,., , ,", , ,., , , 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   0 0 0> > >+ + + ' ' '&h h hVXF XF XFv 
eWvuoU|65Q5QR   P P PBO O O, , ,22' 2' 2'j III , , ,,- - -0 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%$)-". 	//"0 
3>#671"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0. )	~&	/	$	y!		*	y!		*	%'#	#.7),		"	'+% .1 1/ .1.    $% %% $%,, , , , ,r   