U
    c                  
   @   s   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dddge j
d	dd
ge j
dddddgdd ZdS )    N)DatetimeIndex
date_rangec                 C   s   |dkr| S | j }|j}|d| d}| jd kr<|j}nt| j| j|d}t|j||d}t|| jd}|j|ks~t	|S )NnszM8[])tzunit)dtype)name)
_dataZ_ndarrayZastyper   r   typeZ_simple_newr   r	   AssertionError)Zdti_nanor   Zdta_nanoZarr_nanoZarrr   Zdtadti r   T/tmp/pip-unpacked-wheel-g7fro6k3/pandas/tests/indexes/datetimes/methods/test_snap.pyastype_non_nano
   s    
r   zignore::DeprecationWarningr   zAsia/ShanghaizEurope/Berlinr	   Zmy_dtir   r   usmssc                 C   s  t dddddddg| |dd	}t||}|jd
d}tdd| |dd	}|ddg}t||}t|| |j|jkszt|j	d kst|j	d kst|jdd}tdd| |dd	}|dddddg}t||}t|| |j|jkst|j	d kst|j	d kstd S )Nz1/1/2002z1/2/2002z1/3/2002z1/4/2002z1/5/2002z1/6/2002z1/7/2002D)r	   r   freqzW-MON)r   z
12/31/2001zw-mon      Bb      )
r   r   Zsnapr   repeattmZassert_index_equalr   r   r   )r	   r   r   r   resultexpectedr   r   r   test_dti_snap   s<    	


r    )ZpytestZpandasr   r   Zpandas._testingZ_testingr   r   markfilterwarningsZparametrizer    r   r   r   r   <module>   s   
