o
    5c                     @   sz   d dl Z d dlmZmZ d dlmZ dd Ze j	de j
dg de j
ddd	ge j
d
g ddd ZdS )    N)DatetimeIndex
date_rangec                 C   s   |dkr| S | j }|j}|d| d}| jd u r|j}n
t| j| j|d}t|j||d}t|| jd}|j|ks?J |S )NnszM8[])tzunit)dtype)name)	_data_ndarrayastyper   r   type_simple_newr   r	   )dti_nanor   dta_nanoarr_nanoarrr   dtadti r   f/var/www/html/gps/gps/lib/python3.10/site-packages/pandas/tests/indexes/datetimes/methods/test_snap.pyastype_non_nano
   s   
r   zignore::DeprecationWarningr   )NzAsia/ShanghaizEurope/Berlinr	   my_dtir   )r   usmssc                 C   s   t g d| |dd}t||}|jdd}tdd| |dd}|d	d
g}t||}t|| |j|jks8J |jd u s?J |jd u sFJ |jdd}tdd| |dd}|g d}t||}t|| |j|jksoJ |jd u svJ |jd u s}J d S )N)1/1/2002z1/2/2002z1/3/2002z1/4/2002z1/5/2002z1/6/20021/7/2002D)r	   r   freqzW-MON)r   z
12/31/2001r   zw-mon      Br   b)   r$   r$      r%   )	r   r   snapr   repeattmassert_index_equalr   r   )r	   r   r   r   resultexpectedr   r   r   test_dti_snap   s.   	


r,   )pytestpandasr   r   pandas._testing_testingr(   r   markfilterwarningsparametrizer,   r   r   r   r   <module>   s    
