U
|
ý°d» ã @ sV d dl Z d dlZd dlmZmZ d dlmZmZm Z m
|
Z
|
d dl m Z G dd dZdS )é N)ÚIS64Úis_platform_windows)Ú CategoricalÚ DataFrameÚSeriesÚ
|
date_rangec @ s\ e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dS )Ú TestIterationc C s | ¡ |jkstd S ©N)ÚkeysÚcolumnsÚAssertionError©ÚselfÚ float_frame© r úXd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/test_iteration.pyÚ test_keys s zTestIteration.test_keysc C sH t dddgdddggdddgd }| ¡ D ]\}}t|t js*tq*d S )
|
Né é é é é é ÚaÚb©r )r ÚitemsÚ
|
isinstanceÚ_constructor_slicedr )r ÚdfÚkÚvr r r Útest_iteritems s "z TestIteration.test_iteritemsc C st dddg}t dddgddd gg|d
|
}t|| ¡ D ]:\}\}}||ksLtt|tsZt|| |k ¡ s4tq4d S ) Nr r Úcr r r r r r r )r Úzipr r r r Úall)r Úcolsr r# r r! r r r Ú
|
test_items s
|
zTestIteration.test_itemsc C s$ | ¡ D ]\}}|j|kstqd S r )r Únamer )r Úfloat_string_framer r! r r r Útest_items_names% s z TestIteration.test_items_namesc C s t t||j¡std S r )ÚtmZ equalContentsÚlistr r r r r r Ú test_iter) s zTestIteration.test_iterc C sT | ¡ D ] \}}|j| }t ||¡ q| ¡ D ] \}}|j| }t ||¡ q0d S r )ÚiterrowsÚlocr+ Úassert_series_equal)r r r) r r! Úexpr r r Ú test_iterrows, s
|
|
zTestIteration.test_iterrowsc C sL t ddddgtddddd }| ¡ D ] \}}|j| }t ||¡ q(d S )
|
NZM1701ZM1802ZM1903ZM2004z
|
2000-01-01r ÚM)ZperiodsÚfreq)Z non_iso8601Ziso8601)r r r. r/ r+ r0 )r Úsr r! r1 r r r Útest_iterrows_iso86015 s
|
þÿ
|
z#TestIteration.test_iterrows_iso8601c C s t t ddd¡gd gd gdgg gt gi gd}tt ddd¡d d dg t i gtdddd}t| ¡ \}}t ||¡ d S ) Niß r Ú )r r r# ÚdÚeÚfÚgZabcdefgr Úobject)Úindexr( Zdtype) r ÚdatetimeÚsetr r, Únextr. r+ r0 )r r ÚexpectedÚ_Úresultr r r Útest_iterrows_cornerA s$ ùÿ üz"TestIteration.test_iterrows_cornerc C sh t | ¡ D ]J\}}t |dd
¡}|d |_|j|d d
f jdd}t ||¡ q tt j
|
d¡t ddddgd }|jd
|
d D ]}t |d tstqtdd d gdddgdd}|ddg }t| ¡ dddgksàtt s tr tt|jd ddks tt|jdd}|jdks,t|j|j|jf|ksFtt|jdksZtddg|_t|jdd}|d kst|jd ksttd d t d D } t| ¡ }
|
t |
|
t sÄtt |
|
d!sÔttd"d t d#D g} t| jd
|
d } t | t s tt | d!s ttd$d t d%D g} t| jd
|
d }t |t sTtt |d!sdtd S )&Nr r T)Zdropr )ÚfloatsÚintsrE rF r F©r= r r 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)]ZTestName)ÚIndexr r ÚdefÚreturn)r r r )rI Z_1Z_2c S s i | ]}d t | |gqS )r: )Ústr©Ú.0Úir r r Ú
|
<dictcomp>{ s
|
z1TestIteration.test_itertuples.<locals>.<dictcomp>i Ú_fieldsc S s i | ]}d | d| qS ©Zfoo_Zbar_r rM r r r rP s éþ c S s i | ]}d | d| qS rR r rM r r r rP s éÿ ) Ú enumerateÚ
|
itertuplesr r r( ZilocZ reset_indexr+ r0 ÚnpÚrandomZrandnÚranger Úintr r, r r Úreprr@ rQ rI r r ÚtypeÚ__name__r ÚtupleÚhasattr)r r rO ÚtupZserrA r ZdfaaZtup2Zdf3Ztup3Zdf_254_columnsZresult_254_columnsZdf_255_columnsZresult_255_columnsr r r Útest_itertuplesW sN
|
ÿ ÿÿ
|
z TestIteration.test_itertuplesc C s² t ddddddgdddddd gd
|
}t|d |d < t|jj}t |jj¡ ¡ }t ||¡ |j
|
d dD ] }t | ql| ¡ D ]\}}t | q| ¡ D ]\}}t | qd S )Nr r r r r r r r r9 )ÚidÚ raw_graderc ÚgradeFrG )r r r, rd ÚvaluesrW ÚarrayÚtolistr+ Zassert_almost_equalrV rL r. r ) r r rC rA ÚtÚrowr5 r# Úcolr r r Ú#test_sequence_like_with_categorical s ÿ
|
|
z1TestIteration.test_sequence_like_with_categoricalN) r] Ú
|
__module__Ú __qualname__r r" r' r* r- r2 r6 rD ra rk r r r r r s 5r )r> ÚnumpyrW Z pandas.compatr r Zpandasr r r r Zpandas._testingZ_testingr+ r r r r r Ú<module> s
|
|
