U
    sVc                     @   s@   d dl Zd dlmZmZmZ dd Zdd Zdd Zd	d
 Z	dS )    N)	DataFrameSeriesperiod_rangec                 C   sT   t | jD ]D\}}t | jD ]0\}}| j||f }| j||f }||kstqq
d S )N)	enumerateindexcolumnsiatatAssertionError)Zfloat_frameirowjcolresultexpected r   B/tmp/pip-unpacked-wheel-xj8nt62q/pandas/tests/indexing/test_iat.pytest_iat
   s
    r   c                  C   s,   t ddggddgd} | jd dks(td S )N      x)r   )r   r   )r   r   r
   )dfr   r   r   test_iat_duplicate_columns   s    r   c                  C   sF   t ddd} ttjd| d}|| d  }|jd }||ksBtd S )Nz1/1/2001
   )Zperiods)r   r   )r   r   nprandomZrandnr   r
   )r   serr   r   r   r   r   )test_iat_getitem_series_with_period_index   s
    
r   c                 C   s   t jdt jdt dd}t| }|d }d| |d< d| |d	< |jd	 dksZt|sp|jd
 dkspt|jd dkstd S )N   )Zdtyper   )r   yr   i'  )   r   i  )r    r   )r!   r!   )r   ZarangeZint64r   copyr   r
   Ziloc)Zindexer_ialZusing_copy_on_writedatar   r   r   r   r   #test_iat_setitem_item_cache_cleared"   s    r$   )
Znumpyr   Zpandasr   r   r   r   r   r   r$   r   r   r   r   <module>   s
   		