
    d                     d    d dl Z d dlZd dlmZmZ d dlmZmZm	Z	m
Z
 d dlmZ  G d d          ZdS )    N)IS64is_platform_windows)Categorical	DataFrameSeries
date_rangec                   J    e Zd Zd Zd Zd Zd Zd Zd Zd Z	d Z
d	 Zd
 ZdS )TestIterationc                 @    |                                 |j        u sJ d S N)keyscolumnsselffloat_frames     ]/var/www/html/t/fyr/venv311/lib/python3.11/site-packages/pandas/tests/frame/test_iteration.py	test_keyszTestIteration.test_keys   s(    !![%8888888    c                     t          g dg dgg d          }|                                D ]!\  }}t          |t           j                  sJ "d S )N                  )ar   br   )r   items
isinstance_constructor_sliced)r   dfkvs       r   test_iteritemszTestIteration.test_iteritems   sj    			999-GGGHHJJ 	@ 	@DAqa!>??????	@ 	@r   c                    g d}t          g dg dg|          }t          ||                                          D ]G\  }\  }}||k    sJ t          |t                    sJ ||         |k                                    sJ Hd S )N)r   r   cr   r   r    )r   zipr!   r"   r   all)r   colsr$   r)   r%   r&   s         r   
test_itemszTestIteration.test_items   s    			999-t<<<T288::.. 	& 	&IAv16666a(((((qEQJ##%%%%%%	& 	&r   c                 T    |                                 D ]\  }}|j        |k    sJ d S r   )r!   name)r   float_string_framer%   r&   s       r   test_items_nameszTestIteration.test_items_names%   s<    &,,.. 	 	DAq6Q;;;;;	 	r   c                 X    t          j        t          |          |j                  sJ d S r   )tmequalContentslistr   r   s     r   	test_iterzTestIteration.test_iter)   s,    [ 1 1;3FGGGGGGGr   c                     |                                 D ]'\  }}|j        |         }t          j        ||           (|                                 D ]'\  }}|j        |         }t          j        ||           (d S r   )iterrowslocr3   assert_series_equal)r   r   r0   r%   r&   exps         r   test_iterrowszTestIteration.test_iterrows,   s    ((** 	+ 	+DAq/!$C"1c****&//11 	+ 	+DAq$(+C"1c****	+ 	+r   c                     t          g dt          ddd          d          }|                                D ]'\  }}|j        |         }t	          j        ||           (d S )N)M1701M1802M1903M2004z
2000-01-01r   M)periodsfreq)non_iso8601iso8601)r   r   r8   r9   r3   r:   )r   sr%   r&   r;   s        r   test_iterrows_iso8601z#TestIteration.test_iterrows_iso86015   s~    CCC%lACHHH 
 
 JJLL 	+ 	+DAq%(C"1c****	+ 	+r   c           
         t          t          j        ddd          gd gd gdgg gt                      gi gd          }t          t          j        ddd          d d dg t                      i gt	          d          dd          }t          |                                          \  }}t          j        ||           d S )	Ni  r    )r   r   r)   defgabcdefgr   object)indexr/   dtype)	r   datetimesetr   r5   nextr8   r3   r:   )r   r$   expected_results        r   test_iterrows_cornerz"TestIteration.test_iterrows_cornerA   s    'a334VVTTeeWT 

 

 tQ**D$BrJy//	
 
 
 ''	6
vx00000r   c                 &   t          |                                          D ]h\  }}t          j        |dd                    }|d         |_        |j        |d d f                             d          }t          j        ||           it          t          j
                            d          t          d          dddg	          }|                    d
          D ]}t          |d         t                    sJ  t          g dg dd          }|ddg         }t          |                                          g dk    sJ t!                      s=t"          r6t%          t          |                    d                               dk    sJ t'          |                    d                    }|j        dk    sJ |j        |j        |j        f|k    sJ t1          |          j        dk    sJ ddg|_        t'          |                    d                    }|dk    sJ |j        dk    sJ t          d t          d          D                       }	t'          |	                                          }
t          |
t6                    sJ t9          |
d          sJ t          d t          d          D             g          }t'          |                    d
                    }t          |t6                    sJ t9          |d          sJ t          d t          d           D             g          }t'          |                    d
                    }t          |t6                    sJ t9          |d          sJ d S )!Nr   r   T)dropr   )floatsintsr\   r]   r    FrQ   r   r   )r   r   )datar   ))r   r   r   )r   r   r   )r   r   r   )r/   z![(0, 1, 4), (1, 2, 5), (2, 3, 6)]TestName)Indexr   r   defreturn)r   r   r   )ra   _1_2c                 6    i | ]}d t          |          z   |gS )rM   )str.0is     r   
<dictcomp>z1TestIteration.test_itertuples.<locals>.<dictcomp>{   s&    @@@qs1vvs@@@r   i   _fieldsc                      i | ]}d | d| S foo_bar_ rh   s     r   rk   z1TestIteration.test_itertuples.<locals>.<dictcomp>   &    $P$P$PZAZZ$P$P$Pr      c                      i | ]}d | d| S rn   rq   rh   s     r   rk   z1TestIteration.test_itertuples.<locals>.<dictcomp>   rr   r      )	enumerate
itertuplesr   r#   r/   ilocreset_indexr3   r:   nprandomrandnranger"   intr5   r   r   reprrU   rl   ra   r   r   type__name__r   tuplehasattr)r   r   rj   tupserrV   r$   dfaatup2df3tup3df_254_columnsresult_254_columnsdf_255_columnsresult_255_columnss                  r   test_itertupleszTestIteration.test_itertuplesW   s    6 6 8 899 	2 	2FAs/ABB88C1vCH"'111-99t9DDH"31111yq))588<<xQWFX
 
 
 ==u=-- 	+ 	+Cc!fc******))))))<<===3*~DOO%%&&*K*K*KKKKK $%% 	T 	T"--T-22334467 7 7 7 2==j=1122{11111	35#%(C////Cyy!Z////X&
BMMzM2233y    |44444@@E$KK@@@AACNN$$%%$&&&&&tY''''' #$P$PU3ZZ$P$P$P#QRR!.";";%";"H"HII,e44444)955555"$P$PU3ZZ$P$P$P#QRR!.";";%";"H"HII,e44444)95555555r   c                    t          g dg dd          }t          |d                   |d<   t          |j        j                  }t          j        |j        j                                                  }t          j	        ||           |
                    d          D ]}t          |           |                                D ]\  }}t          |           |                                D ]\  }}t          |           d S )N)r   r   r   r   r   r   )r   r   r   r   r   rL   )id	raw_grader   gradeFr^   )r   r   r5   r   valuesrz   arraytolistr3   assert_almost_equalrw   rg   r8   r!   )	r   r$   rX   rV   trowrG   r)   cols	            r   #test_sequence_like_with_categoricalz1TestIteration.test_sequence_like_with_categorical   s    %%%4R4R4RSS
 
 ""[/227 bho&&8BHO,,3355
vx000 U++ 	 	AFFFFkkmm 	 	FCFFFFhhjj 	 	FAsHHHH	 	r   N)r   
__module____qualname__r   r'   r-   r1   r6   r<   rH   rY   r   r   rq   r   r   r
   r
      s        9 9 9@ @ @
& & &  H H H+ + +
+ 
+ 
+1 1 1,36 36 36j    r   r
   )rS   numpyrz   pandas.compatr   r   pandasr   r   r   r   pandas._testing_testingr3   r
   rq   r   r   <module>r      s               
                 N N N N N N N N N Nr   