U
    sVc                     @   s:   d dl Z d dlmZmZmZ d dlmZ G dd dZdS )    N)IndexSeries
date_rangec                   @   s,   e Zd Zdd Zdd Zdd Zdd Zd	S )
TestSeriesDelItemc              	   C   s   t td}|d= t tddtddd}t|| |d= t tddtddd}t|| t d}|d= t|t dtg ddd d|d< t|t d |d= t|t dtg ddd d S )	N   r      index   int64dtyper   r	   )r   rangetmassert_series_equalr   )selfsexpected r   M/tmp/pip-unpacked-wheel-xj8nt62q/pandas/tests/series/indexing/test_delitem.pytest_delitem   s    zTestSeriesDelItem.test_delitemc              	   C   st   t ddgd}|d= t|t dtg ddd d|d< t|t ddgd |d= t|t dtg ddd d S )Nr   ar   r   objectr   r   )r   r   r   r   r   r   r   r   r   test_delitem_object_index"   s      z+TestSeriesDelItem.test_delitem_object_indexc              	   C   s.   t td}tjtdd |d= W 5 Q R X d S )Nr   z^0$)matchr   )r   r   pytestZraisesKeyErrorr   r   r   r   test_delitem_missing_key0   s    
z*TestSeriesDelItem.test_delitem_missing_keyc                 C   s   t dddd}t|}|ddg }|d= |j|jks8tt|| |d d}t|}|d d }|d= |j|jks~tt|| d S )	Nz
2016-01-01   z
US/Pacific)Zperiodstzr   r
   r   D)r   r   r   AssertionErrorr   r   Ztz_localizeZ	to_period)r   ZdtiZserr   pir   r   r   test_delitem_extension_dtype7   s    z.TestSeriesDelItem.test_delitem_extension_dtypeN)__name__
__module____qualname__r   r   r   r%   r   r   r   r   r      s   r   )	r   Zpandasr   r   r   Zpandas._testingZ_testingr   r   r   r   r   r   <module>   s   