a
    Pf                     @   sL   d dl Zd dlmZmZmZ d dl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J qq
d S )N)	enumerateindexcolumnsiatat)Zfloat_frameirowjcolresultexpected r   W/var/www/ai-form-bot/venv/lib/python3.9/site-packages/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(J d S )N      x)r   )r   r   )r   r   )dfr   r   r   test_iat_duplicate_columns   s    r   c                  C   sL   t ddd} ttjdd| d}|| d  }|jd }||ksHJ d S )Nz1/1/2001
   )Zperiodsr   )r   r   )r   r   nprandomZdefault_rngZstandard_normalr   )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 }t| d| |d< W d    n1 s\0    Y  t| d| |d	< W d    n1 s0    Y  |jd	 dksJ |s|jd
 dksJ |jd dksJ d S )N   )Zdtyper   )r   yr   i'  )   r   i  )r   r   )r    r    )	r   ZarangeZint64r   copytmZassert_cow_warningr   Ziloc)Zindexer_ialZusing_copy_on_writeZwarn_copy_on_writedatar   r   r   r   r   #test_iat_setitem_item_cache_cleared"   s    **r$   )numpyr   Zpandasr   r   r   Zpandas._testingZ_testingr"   r   r   r   r$   r   r   r   r   <module>   s   	