U
|
ý°dbE ã @ sr d dl mZmZ d dlmZ d dlZd dlZd dlZd dlm Z m
|
Z
|
m Z m Z m Z mZ d dlmZ G dd dZdS )é )Ú OrderedDictÚ defaultdict)ÚdatetimeN)ÚNAÚ DataFrameÚIndexÚ
|
MultiIndexÚSeriesÚ Timestampc @ s$ e Zd Zdd Zdd Zdd Zej ddd d
|
d d d g¡dd Z ej de
|
e e e g¡dd Zej de e g g¡dd Zdd Zej dddd fdd d fd d d fd d d fd!d"d fg¡d#d$ Zd%d& Zej d'e
|
d(d)d*d+d,d*d-d.d*d/fe e d0d(d)d*fd(d+d,d*fd+d-d.d*fgfe e
|
e e
|
d(d)d*d+d,d*d-d.d*d/fg¡d1d2 Zd3d4 Zd5d6 Zej d7d8d8d9gefed:d(d(ed;d+d+ed<d-d-gegd)d,d.gegd(d+d-gegd=d>d?gegf¡d@dA Z ej dBe d+¡ege dC¡ege dD¡ege !d8¡ege "dE¡egf¡dFdG Z#dHdI Z$ej d!dJe%dKdLge%dKdLgdMdNe& 'dOdPg¡e&j'dOdPgdQdRgdSg¡ej dTdUdVge%dUdVge%dUdVgdWdNe& 'dXdYg¡e&j'dXdYgdZd[gdSg¡d\d] Z(ej ddd d dd!d^g¡ej d_e d(¡d(e d-¡ge d)¡d,e d.¡ge d)¡d+e d-¡ge d)¡d`e d-¡ge d)¡d`ge d-¡ge d)¡dae d-¡gdbeeegeeegeeegeeegee egee)egdbfd(d+d-gdDdcddgdOeeegeeegdOfd(ded-gdDdfddgdOeeegeeegdOff¡dgdh Z*ej ddd didd!g¡djdk Z+ej dld dmdngd(d-gd+doggdpfd^dmdngd(d-gd+doggdJgdqfg¡drds Z,ej dldd`d(dJdtifd d`d(dJgifd d0d(gd`gd(gdJggdufd^d0d(gd`gd(gdJggdJgdJgdvfdd`d(id`dJigfd!d`d(id`dJidtfg¡dwdx Z-dydz Z.dJS ){ÚTestDataFrameToDictc
|
C s\ t d}t||g||gd}t||gddgd}||d||dg}|dd|ddg}|jdd|kspt|jdd|kstt||gddt||gd dd}t||gddtddgd dd}t |jd
|
d|¡ t |jd
|
d|¡ d dg||g||ggdd gd }d dg|dg|dggdd gd } t |jd d|¡ t |jd d| ¡ d S )NZ20130101©ÚAÚBé é Úrecords©Úorientr ©Únamer Úseriesr )ÚindexÚdataÚcolumnsÚsplit)r
|
r Úto_dictÚAssertionErrorr ÚtmÚassert_dict_equal)
|
ÚselfZtsmpÚ test_dataZtest_data_mixedÚexpected_recordsZexpected_records_mixedZexpected_seriesZexpected_series_mixedÚexpected_splitZexpected_split_mixed© r# ú^d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/methods/test_to_dict.pyÚtest_to_dict_timestamp s> þþ
|
ÿýý
|
ÿz*TestDataFrameToDict.test_to_dict_timestampc C sL t ddgddgdddgd}d}tjt|d |jd
|
d W 5 Q R X d S ) Nr r g à?g è?©ÚaÚbr ©r z1DataFrame index must be unique for orient='index'©Úmatchr r ©r ÚpytestÚraisesÚ
|
ValueErrorr ©r ÚdfÚmsgr# r# r$ Ú/test_to_dict_index_not_unique_with_index_orientD s zCTestDataFrameToDict.test_to_dict_index_not_unique_with_index_orientc C s> t dddgi}d}tjt|d |jdd W 5 Q R X d S )Nr r r z orient 'xinvalid' not understoodr* Zxinvalidr r, r0 r# r# r$ Útest_to_dict_invalid_orientL s z/TestDataFrameToDict.test_to_dict_invalid_orientr ÚdÚlÚrÚspÚsÚic C s: t dddgi}tjtdd |j|d W 5 Q R X d S )Nr r r znot understoodr* r r, )r r r1 r# r# r$ Ú test_to_dict_short_orient_raisesR s z4TestDataFrameToDict.test_to_dict_short_orient_raisesÚmappingc C s| dddddddd}t |j|d }| ¡ D ].\}}| ¡ D ] \}}||| | ks@tq@q0t | d
|
|¡}| ¡ D ]6\}}| ¡ D ]$\}}||| t|d kstqqxt | d |¡}| ¡ D ].\}}| ¡ D ] \}}||| | ksØtqØqÈt | d |¡}d dgdddgddgddgtjdggd}t ||¡ t | d|¡}ddddddtjddg} t |t
|
s|tt |dkstt || D ]\}
|
} t |
|
| ¡ qt | d¡}| ¡ D ]4\}}| ¡ D ] \}}||| | ksÖtqÖqÆt |} | | j d | d< | d¡}| ¡ } | | j d | d< | ¡ D ]4\}}| ¡ D ] \}}||| | ksRtqRqBd S )Nr r )Ú1Ú2r= r> Ú3)r= r> r? r ©ÚintoÚlistr r r r ç ð?ç @)r r r r é r r Zduped)r r Úitemsr ÚintÚnpÚnanr r Ú
|
isinstancerB ÚlenÚzipr Úcopy)r r< r Z recons_dataÚkÚvZk2Zv2r" r! ÚleftÚrightr1 Z comp_datar# r# r$ Ú test_to_dictY sP ý
|
ý
|
z TestDataFrameToDict.test_to_dictc C sJ t tj dd¡}d ddg¡}tjt|d |j|d W 5 Q R X d S )NrE ú|z unsupported type: <class 'list'>z1to_dict\(\) only accepts initialized defaultdictsr* r@ ) r rH ÚrandomZrandnÚjoinr- r. Ú TypeErrorr )r r< r1 r2 r# r# r$ Útest_to_dict_errors s þÿz'TestDataFrameToDict.test_to_dict_errorsc C s< t dddggdddgd}t t¡ | ¡ W 5 Q R X d S )Nr r rE r' r( ©r )r r Zassert_produces_warningÚ UserWarningr )r r1 r# r# r$ Ú test_to_dict_not_unique_warning¢ s z3TestDataFrameToDict.test_to_dict_not_unique_warningzorient,item_getterÚdictc C s | | | S ©Nr# ©r5 ÚcolÚidxr# r# r$ Ú<lambda>¯ ó z TestDataFrameToDict.<lambda>r c C s | | | S r\ r# r] r# r# r$ r` ° ra rB c C s | | | S r\ r# r] r# r# r$ r` ± ra r c C s | d | | d |¡ S )Nr r r) r] r# r# r$ r` ² ra r c C s | | | S r\ r# r] r# r# r$ r` ³ ra c C sR t ddgddgd}|j|d}t||ddts8tt||d dtsNtd S )
|
Nr r g¹?gÉ?r& r r' r r( )r r rJ rG r Úfloat)r r Z item_getterr1 Úresultr# r# r$ Útest_to_dict_box_scalars¬ s z,TestDataFrameToDict.test_to_dict_box_scalarsc C s¦ t dddddddtjdft dddd d
|
d d tjdfg}tt|d gd}|jdd}d tdtjdid tdtjdig}t |d |d ¡ t |d |d ¡ d S )Niá é é é é5 r iYX )Útzinfoé é é irñ r5 rX r r z 2017-11-18 21:53:00.219225+0000)Útzz 2017-11-18 22:06:30.061810+0000r ) r ÚpytzÚutcr rB r r
|
r r )r r r1 rc Úexpectedr# r# r$ Útest_to_dict_tz¾ s þ þz#TestDataFrameToDict.test_to_dict_tzzinto, expectedr rC ©Úint_colÚ float_colr rD rE ç @)r r r r c C sd t dddgdddgd}|jd|d }d
|
d g}t j|dd | }t j|dd | }t ||¡ d S ) Nr r rE rC rD ru rr r )r rA rs rt r )r r Ú from_dictr Úassert_frame_equal)r rA rp r1 rc Úcolsr# r# r$ Útest_to_dict_index_dtypesÏ s 'z-TestDataFrameToDict.test_to_dict_index_dtypesc C sF t dd tdD }t| d¡d ¡ }t|j}||ksBtd S )Nc S s i | ]}t ||gqS r# )Ústr©Ú.0r: r# r# r$ Ú
|
<dictcomp> s zBTestDataFrameToDict.test_to_dict_numeric_names.<locals>.<dictcomp>é r r )r ÚrangeÚsetr Úkeysr r ©r r1 rc rp r# r# r$ Útest_to_dict_numeric_namesþ s
|
z.TestDataFrameToDict.test_to_dict_numeric_namesc C sF t dd tdD }| d¡d }dd tdD }||ksBtd S )Nc S s i | ]}d |d|gqS ©ZA_r5 r# r{ r# r# r$ r} s
|
z9TestDataFrameToDict.test_to_dict_wide.<locals>.<dictcomp>é r r c S s i | ]}d |d|qS r r# r{ r# r# r$ r} s
|
)r r r r r r# r# r$ Útest_to_dict_wide s z%TestDataFrameToDict.test_to_dict_widez
|
data,dtypeTFiâ iã iä ÚXÚYÚZc s6 t d|i}|jdd}t fdd|D s2td S )Nr' r r c 3 s | ]}t |d kV qdS )r' N)Útype)r| Úrecord©Údtyper# r$ Ú <genexpr>" s z@TestDataFrameToDict.test_to_dict_orient_dtype.<locals>.<genexpr>)r r Úallr )r r r r1 r5 r# r r$ Útest_to_dict_orient_dtype s z-TestDataFrameToDict.test_to_dict_orient_dtypezdata,expected_dtypei÷ÿÿÿgñ?z
|
2005-02-25c C s> t d|idgd}|jdd}t|d d }||ks:td S )Nr' r r) r r )r r r r )r r Zexpected_dtyper1 r5 rc r# r# r$ Ú,test_to_dict_scalar_constructor_orient_dtype$ s z@TestDataFrameToDict.test_to_dict_scalar_constructor_orient_dtypec C s> t dgdgd}| ¡ d¡}ddddg}||ks:td S )NrC g "@r& r r )r r' r( )r Z reset_indexr r r r# r# r$ Ú test_to_dict_mixed_numeric_frame6 s z4TestDataFrameToDict.test_to_dict_mixed_numeric_frameNZaaZbbÚccr r& )r' ÚcZn1Zn2)Únamesr ÚxÚyÚz)r r )r r Zz1Zz2c C sB t jddgddgg||d}t j|jdddd}t ||¡ d S )Nr rE r é )r r Útightr )r Z from_recordsrv r r rw )r r r r1 Z roundtripr# r# r$ Útest_to_dict_orient_tight= s ýz-TestDataFrameToDict.test_to_dict_orient_tightr zdata,expected_typesr' )r' )r' r( r r5 ÚeÚfg@gffffff
|
@ZhelloZworldc sà t |}| |¡ |dkr.dd ¡ D }nr|dkrJdd ¡ D }nV|dkrj fdd d D }n6|d krd
|
d t D }n|dkr d d ¡ D }|D ]6\}}}||| | ksÂtt||| | ks¤tq¤d S ) Nr[ c s s. | ]&\}}| ¡ D ]\}}|||fV qqd S r\ ©rF )r| ÚkeyZindex_value_mapr: Úvaluer# r# r$ r s
|
þzHTestDataFrameToDict.test_to_dict_returns_native_types.<locals>.<genexpr>rB c s s. | ]&\}}t |D ]\}}|||fV qqd S r\ ©Ú enumerate)r| r Úvaluesr: r r# r# r$ r s
|
þ> r r c 3 s: | ]2}t d D ] \}}|| d | | fV qqdS )r r Nr¡ )r| r: Újr ©rc r# r$ r s þr r c s s. | ]&\}}| ¡ D ]\}}|||fV qqd S r\ r ©r| r: r r r r# r# r$ r ¢ s
|
þc s s. | ]&\}}| ¡ D ]\}}|||fV qqd S r\ r r¦ r# r# r$ r ¨ s
|
þ)r r rF r¢ r r ) r r r Zexpected_typesr1 Zassertion_iteratorr: r r r# r¥ r$ Ú!test_to_dict_returns_native_types[ s2 2
|
þþ
|
þþþz5TestDataFrameToDict.test_to_dict_returns_native_typesr c C sN t ddgddgdddgd}d }tjt|d
|
|j|d d W 5 Q R X d S ) Nr r rE r ©Úcol1Úcol2Úrow1Úrow2r) z?'index=False' is only valid when 'orient' is 'split' or 'tight'r* F©r r r, )r r r1 r2 r# r# r$ Ú test_to_dict_index_false_error² s z2TestDataFrameToDict.test_to_dict_index_false_errorzorient, expectedr© rª r )r r )r r Ú column_namesc C s< t ddgddgdddgd}|j|d d
|
}t ||¡ d S ) Nr r rE r r¨ r« r¬ r) Fr )r r r r ©r r rp r1 rc r# r# r$ Útest_to_dict_index_falseº s z,TestDataFrameToDict.test_to_dict_index_false)r r )r r r )r r r Z index_namesr¯ c C s0 t ddtgidd}|j|d}||ks,td S )Nr' r ÚInt64r r )r r r r r° r# r# r$ Útest_to_dict_na_to_noneÎ s z+TestDataFrameToDict.test_to_dict_na_to_nonec C s t tddgdddd}|jdd}t|d d tks>tt tdtgdddd}|jdd}t|d d tks|td S )
|
Nr r r² r )r' r r r r r' )r r r r rG r r )r r1 rc r# r# r$ Ú!test_to_dict_masked_native_pythonè s z5TestDataFrameToDict.test_to_dict_masked_native_python)/Ú__name__Ú
|
__module__Ú __qualname__r% r3 r4 r- ÚmarkZ parametrizer; r[ r rB r rR rW rZ rd rq ry r r Úboolr r
|
rb rG rz r rH Zuint64Úint64Úfloat64Zbool_Z
|
datetime64r r r r Ú from_tuplesr Útupler§ r® r± r³ r´ r# r# r# r$ r sN -
|
|
;
|
|
|
|
|
|
|
ûþ
|
|
ýþ ýÿþ ýþþíþ"
|
|
|
|
ýú ôþ
|
ûþ
|
|
|
ûþ
|
|
ûþ
|
|
þú ú÷þþû þþûãþ+ (
|
ýþþþ
|
|
ûþ
|
ñþ
|
r )Ú collectionsr r r ÚnumpyrH r- rn Zpandasr r r r r r
|
Zpandas._testingZ_testingr r r# r# r# r$ Ú<module> s
|
