o
    թZh                     @   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 ]"\}}t | jD ]\}}| 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   U/var/www/html/lang_env/lib/python3.10/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s$J 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.w   Y  t| d| |d	< W d    n1 sIw   Y  |jd	 dksWJ |sb|jd
 dksbJ |jd dkskJ 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    	