
    Owg                     D    d dl Zd dlmZmZmZ d dlmZ d Z	d Z
d Zd Zy)    N)	DataFrameSeriesperiod_rangec                     t        | j                        D ]K  \  }}t        | j                        D ].  \  }}| j                  ||f   }| j                  ||f   }||k(  r.J  M y )N)	enumerateindexcolumnsiatat)float_frameirowjcolresultexpecteds          U/var/www/horilla/myenv/lib/python3.12/site-packages/pandas/tests/indexing/test_iat.pytest_iatr      sp    K--. &3 3 34 	&FAs __QT*F"~~c3h/HX%%%	&&    c                  P    t        ddggddg      } | j                  d   dk(  sJ y )N      x)r	   )r   r   )r   r
   )dfs    r   test_iat_duplicate_columnsr      s.    	QF8c3Z	0B66$<1r   c                      t        dd      } t        t        j                  j	                  d      j                  d      |       }|| d      }|j                  d   }||k(  sJ y )Nz1/1/2001
   )periodsr   )r   r   )r   r   nprandomdefault_rngstandard_normalr
   )r   serr   r   s       r   )test_iat_getitem_series_with_period_indexr$      s\    R0E
&&q)99"=U
KC58}HWWQZFvr   c                    t        j                  dt         j                        t        j                  d      d}t        |      j	                         }|d   }t        j                  |      5  d | |      d<   d d d        t        j                  |      5  d | |      d	<   d d d        |j                  d	   dk(  sJ |s|j                  d
   dk(  sJ |j                  d   dk(  sJ y # 1 sw Y   qxY w# 1 sw Y   TxY w)N   )dtyper   )r   yr(   i'  )   r   i  )r)   r   )r*   r*   )	r   arangeint64r   copytmassert_cow_warningr
   iloc)indexer_ialusing_copy_on_writewarn_copy_on_writedatar   r#   s         r   #test_iat_setitem_item_cache_clearedr5   "   s     1BHH-BHHQK@D	4			B
S'C 
		1	2 % $B% 
		1	2 % $B% 66$<4xx|t###776?d"""% %% %s   /C+C7+C47D )numpyr   pandasr   r   r   pandas._testing_testingr.   r   r   r$   r5    r   r   <module>r;      s+     
 &#r   