
    .i1                     2    d dl Zd dlmZmZm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          \/var/www/app/trading-bot/venv/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   )	numpyr   pandasr   r   r   r   r   r$    r   r   <module>r(      s!     &r   