
    >ie6                       d dl mZ 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mZ d&dZerdnd	Zej                            g d
 e eddgg dddgddddegddgddg                              d             Zej                            dg d          d             Zd Zd Zd Zd Zd Zd Zd  Zd! Zd" Zej                            d#g d$          d%             ZdS )'    )annotations)productN)PANDAS_GE_220)	assert_eqmeanc                J     t           | j        |fi ||                      S N)getattrresample)dffreqhowkwargss       Jlib/python3.11/site-packages/dask/dataframe/tseries/tests/test_resample.pyr   r      s.    47;2;t..v..44666    MEM)objmethodnpartitionsr   closedlabelseriesframe)countr   ohlc      30minhdwrightleftc                x   t          j        ddd          }|                    t          j        ddd                    }| dk    r1t          j        t	          t          |                    |          }n8| d	k    r2t          j        d
t	          t          |                    i|          }t          j        ||          }t          |||||          }	t          |||||          }
t          |	|
d           |	j        }|
j        d         |d         k    sJ |
j        d         |d         k    sJ d S )N1-1-2000	2-15-2000r    r   	4-15-2000	5-15-2000r   indexr   ar   )r   r   r   Fcheck_dtyper   )pd
date_rangeunionSeriesrangelen	DataFrameddfrom_pandasr   r   	divisionsr,   )r   r   r   r   r   r   r,   psdsresultexpectedr;   s               r   test_series_resampler@      s:    M*k<<<EKKk;SIIIJJE
hYuSZZ((666	\3c%jj 1 12%@@@		4	4	4B b$F6GGGFDfV5IIIHfhE2222 I>!	!,,,,>"2......r   r   )r   nuniquesizesumc           	        t          j        ddd          }|                    t          j        ddd                    }t          j        t	          t          |                    |          }t          j        |d	          }t           t          |
                    d
          |                        t          |
                    d
          |                                  d S )N
2000-01-01
2000-02-15r    r(   r)   r*   r+   r   r.   r   )r2   r3   r4   r5   r6   r7   r9   r:   r   r
   r   )r   r,   r<   r=   s       r   $test_resample_has_correct_fill_valuerG   6   s    M,3???EKKk;SIIIJJE	5U$$E	2	2	2B		*	*	*B-G$$f--//1VW9M9Mv1V1V1X1X    r   c                 8   t          j        ddd          } t          j        t          t	          |                     |           }t          j        |d          }t          |                    d          	                    d	          |                    d          	                    d	                     t          |                    d          	                    d	d
g          |                    d          	                    d	d
g                     d S )NrE   rF   r    r(   r+   r   r.   10minr   min)
r2   r3   r5   r6   r7   r9   r:   r   r   aggr,   r<   r=   s      r   test_resample_aggrM   B   s    M,3???E	5U$$E	2	2	2B		*	*	*Bbkk'""&&v..G0D0D0H0H0P0PQQQ
G  &%11
G  &%11    r   c                 0   t          j        ddd          } t          j        t          t	          |                     |           }t          j        |d          }dd	}t          |                    d
          	                    |d          |                    d
          	                    |d                     |                    d
          	                    |d          dk    
                                                                sJ d S )NrE   rF   r    r(   r+   r   r.      c                    |S r	    )r   barargsr   s       r   fooz,test_resample_agg_passes_kwargs.<locals>.fooS   s    
r   2h)rR   )rO   )r2   r3   r5   r6   r7   r9   r:   r   r   rK   computeall)r,   r<   r=   rT   s       r   test_resample_agg_passes_kwargsrX   N   s   M,3???E	5U$$E	2	2	2B		*	*	*B    bkk$##CQ#//T1B1B1F1FsPQ1F1R1RSSSKK!!#1!--2;;==AACCCCCCCr   c                    t          j        ddd          } |                     t          j        ddd                    } t          j        t	          t          |                     |           }t          j        |d	          }t          j	        t          d
          5  |                    dt                                                                                      d d d            d S # 1 swxY w Y   d S )Nr&   r'   Dr(   r)   r*   r+   r   r.   z(Index is not contained within new index.)match2)r2   r3   r4   r5   r6   r7   r9   r:   pytestraises
ValueErrorr   r   r   rV   rL   s      r   Ctest_resample_throws_error_when_parition_index_does_not_match_indexr`   Z   s   M*k<<<EKKk;SIIIJJE	5U$$E	2	2	2B		*	*	*B	z)S	T	T	T 0 0
HHH##%%--///0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0s   #AC22C69C6c                    g d} t          j        | t          t          |                     d          }t          j        |d         d          |d<   |                    d                              d                                          }t          j	        |d                              d          }|                    d                                          
                                }t          ||           d S )	N)l   @Brk+ l   @-I$
l+ l   D60l+ l   >Fn+ l   Pln+ l   cn+ l   vn+ l   Y/b;2. l   pWta2. l   O2. l   2. l   .o-2. l    zl@2. l   h#S 3. l   }Fk}/ l   qXk}/ l   (]~;l}/ l   *bl}/ l   06l}/ l   XIl}/ l   Lh\l}/ l   HJGm}/ l   xbOyn	- )TimeCountsrb   T)utc1Qr   r.   )r2   r8   r6   r7   to_datetime	set_indexr   rB   r9   r:   rV   r   )timesr   r?   ddfactuals        r   4test_resample_pads_last_division_to_avoid_off_by_onerk   c   s      E4 
uc%jj0A0ABB	C	CB6
555BvJ||F##,,T227799H
.
+
+
+
5
5f
=
=C\\$$$&&..00Ffhr   c                    dd l } t          j        ddd          }|                    t          j        ddd                    }t          j        d| j                            t          |                    i|	          }t          j        |d
          }|	                    d          
                                }|	                    d          
                                                                }t          ||           d S )Nr   z
2012-01-02z
2012-02-02r    r(   z
2012-03-02z
2012-04-02pr+   r   r.   2D)numpyr2   r3   r4   r8   randomr7   r9   r:   r   r   rV   r   )npr,   r   ri   r?   r>   s         r   (test_resample_does_not_evenly_divide_dayrr      s    M,3???EKKlLsKKKLLE	sBI,,SZZ889	G	G	GB
.
+
+
+C{{4  &&((H\\$%%''//11Ffhr   c                    t          j        ddd          } |                     t          j        ddd                    } t          j        t	          t          |                     |           }t          j        |d	          }|                    d
          	                                }|                    d
          	                                
                                }t          ||           d S )Nz2012-01-02 00:00:00z2012-01-02 01:00:00rJ   r(   z2012-01-02 06:00:00z2012-01-02 08:00:00r+   r   r.   57min)r2   r3   r4   r5   r6   r7   r9   r:   r   r   rV   r   )r,   sr=   r?   r>   s        r   /test_series_resample_does_not_evenly_divide_dayrv      s    M/1FUSSSEKK
+-BOOO E 		%E

##5111A	q	)	)	)Bzz'""''))H[[!!&&((0022Ffhr   c                    t          j        dg di          } t          j        | dd          }	 |j                            d                                           J # t          $ r}dt          |          v sJ Y d }~d S d }~ww xY w)Nx)rO   r      r   F)r   sort1mr;   )	r2   r8   r9   r:   rx   r   r   r_   str)r   ri   es      r   test_unknown_divisions_errorr~      s    	sIII&	'	'B
.
7
7
7C%t!!### % % %c!ff$$$$$$$$$$%s   .A 
B)BBc                    ddl m } m} dd l}|                                 }t	          j        || |d          z   d          }|j                            ddt          |                    }t	          j	        ||d	          }|
                    d
          }t          j        |d          }|                    d                                                                          j        j        d
k    sJ d S )Nr   )datetime	timedelta   rZ   r(   rO   d   )highrB   )datevaluesr      r.   )r   r   ro   nowr2   r3   rp   randintr7   r8   rg   r9   r:   r   r   headr,   name)r   r   rq   
date_todaydaysdatar   ri   s           r   test_resample_index_namer      s    ,,,,,,,,J=Z))B--%?cJJJD9QSs4yy99D	tt44	5	5B	f		B
.
+
+
+C<<!!##((**05??????r   c                    t          j        d          t          j        d          t          j        d          g} t          j        dgdgdgg|           }|j                            d          |_        t          j        |d	          }|                    d
                                          }|                    d
                                          }t          ||d           d S )Nz2016-10-15 00:00:00z2016-10-16 10:00:00z2016-10-17 00:00:00rO   r   ry   r+   zAmerica/Sao_Paulor.   1DF)
check_freq)
r2   	Timestampr8   r,   tz_localizer9   r:   r   r   r   )r,   r   ri   r>   r?   s        r   *test_series_resample_non_existent_datetimer      s    
*++
*++
*++E
 
sQC!oU	3	3	3Bx##$788BH
.
+
+
+C\\$$$&&F{{4  %%''Hfh5111111r   rK   )rA   r   r   rB   quantilec                h    t          j        ddd          }t          j        t          t	          |                    |          }t          j        |d          } fd} ||                    d	                    } ||                    d	                    }t          ||d
           d S )NrE   rF   r    r(   r+   r   r.   c                4     t          |                       S r	   )r
   )r   rK   s    r   <lambda>z"test_common_aggs.<locals>.<lambda>   s    #72s##%% r   1dFr/   )	r2   r3   r5   r6   r7   r9   r:   r   r   )rK   r,   r<   r=   fresr?   s   `      r   test_common_aggsr      s    M,3???E	5U$$E	2	2	2B		*	*	*B%%%%A
!BKK

CqT""##Hc8//////r   )r   )
__future__r   	itertoolsr   pandasr2   r]   dask.dataframe	dataframer9   dask.dataframe._compatr   dask.dataframe.utilsr   r   r   markparametrizelistr@   rG   rM   rX   r`   rk   rr   rv   r~   r   r   r   rQ   r   r   <module>r      s)   " " " " " "                  0 0 0 0 0 0 * * * * * *7 7 7 7 #TT ???Dw%%%Fc3R(ff	
 	
	 	 / / /* #F#F#FGG  HG	 	 		D 	D 	D0 0 0"  "  " J          % % %@ @ @"2 2 2  P P PQQ
0 
0 RQ
0 
0 
0r   