U
|
ý°dõI ã @ sn d dl mZ d dlZd dlmZ d dlZd dlZd dlm Z m
|
Z
|
m Z m Z m Z mZ d dlmZ G dd dZdS )é )ÚabcN)ÚParser)ÚCategoricalDtypeÚ DataFrameÚ
|
MultiIndexÚSeriesÚ TimestampÚ
|
date_rangec @ s e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dd Z ej di ejjddgdej dfd ej dfd ej d fd gd fdd!iejjddgdej dfd ej dfd ej d fd gd fd"ej d#iejjd$d%gdej dfd ej d#fd ej d#fd&ej d#fgd fd'ej d(iejjd)d*gdej d(fd ej dfd ej d fd gd fd"eiejjd$d%gdej dfd ej d+fd ej d+fd&ej d+fgd fd"e d,¡iejjd$d%gdej dfd ej d+fd ej d+fd&ej d+fgd fd"ejejej d-d.iejjd$d%gdej dfd/d ej d0fd&ej d-fgd fd'd1d2iiejjddgd3d ej dfd ej d fd gd fd4ej d-d5ejjd6d7gd ej dfd ej d fd gd fd'd2d8d9iejjddgd3d ej dfd ej d fd gd fd"ejejd:iejjd$d%gdej dfd/d ej d0fd gd fd"e d;¡e d8¡d:iejjd$d%gdej dfd/d ej d0fd gd fejejd:ej d-d<ejjd$d%gdej d-fd/d ej d0fd gd fd4g d=ed>ffd4d?d@d:d=edAffd4d?edBdCgd:d=edDffd4d?dEd:d=e dFffg¡dGdH Z ej dIe dJdKdLgdMd@dNgdOdPdQgge dRdS dBdCg¡dTd?d;dUd<ejjdVdWdXgdBej dYfdZd[ej d fgd fe dJdKdLgdMd@dNgdOdPdQgge! "d\d]d^g¡dSej d(d8d_d8d<ejjd`dadbgdej d0fdcej d(fddej dfdeej d0fgd fe dJdKdLgdMd@dNgdOdPdQgge!j"d\d]d^ge dfdge!j"dhdidjge dkdgdldTej d-d;dUd<ejjdmdndogd[ej d-fdpdcej d fddej d fdeej d fgd fg¡dqdr Z#dsdt Z$ej dudvdwdxg¡dydz Z%d{S )|ÚTestDataFrameToRecordsc C sT t ddd}ttj dd¡|dddgd}| ¡ }|d jd
|
ksDt|jd d }d S ) Nz1/1/2000é
|
©Úperiodsé ÚaÚbÚc)ÚindexÚcolumnsr úM8[ns]F©r )r r ÚnpÚrandomÚrandnÚ
|
to_recordsÚdtypeÚAssertionError)Úselfr ÚdfÚresult© r úad:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/methods/test_to_records.pyÚtest_to_records_timeseries s
|
z1TestDataFrameToRecords.test_to_records_timeseriesc C sN t dddgdddggtddd }|jjd
|
}| ¡ d d
|
}||ksJtd S ) NÚoneÚtwoZthreeZfourZfiveÚsixú
|
2012-01-01ú
|
2012-01-02r r r )r r r Úvaluesr r ©r r Úexpectedr r r r Útest_to_records_dt64 s þ z+TestDataFrameToRecords.test_to_records_dt64c C s` t dtddddi}| ¡ }|jd tks0t|d d }t|tsJt||jd ks\td S ) NÚAr% r& ú
|
US/Eastern)Útzr é )r r+ ) r r r r Úobjectr Ú
|
isinstancer Úloc)r r r Úvalr r r Ú test_to_records_dt64tz_column' s z4TestDataFrameToRecords.test_to_records_dt64tz_columnc C sj ddddddddgddddddddgg}t d¡}t||d}|jd dd
|
}d|ksZtd|ksftd S ) NÚbarZbazÚfooZquxr" r# )é é r TZlevel_0)r Úzerosr r r )r r Údatar Úrr r r Ú test_to_records_with_multindex2 s þ
|
z5TestDataFrameToRecords.test_to_records_with_multindexc sB t j tjj¡ t d¡}t |g¡ t
|
fdddD d S )NzbFrom: <user@example.com>
|
To: <someone_else@example.com>
|
Subject: Test message
|
|
Body would go here
|
c 3 s | ]}| kV qd S ©Nr )Ú.0Úx©Úframer r Ú <genexpr>J s zKTestDataFrameToRecords.test_to_records_with_Mapping_type.<locals>.<genexpr>)ÚTypeÚSubjectÚFrom) r ÚMappingÚregisterÚemailÚmessageÚMessager Úparsestrr Z from_recordsÚall)r Úheadersr r? r Ú!test_to_records_with_Mapping_type> s ÿ z8TestDataFrameToRecords.test_to_records_with_Mapping_typec C s t tj dd¡}| ¡ d S )Nr )r r r Zrandr )r r r r r Útest_to_records_floatsL s z-TestDataFrameToRecords.test_to_records_floatsc s t tj dd¡ d j_ ¡ }d|jjks2t t tj dd¡ ¡ }d|jjks\t t
|
dddg¡ _dd g j_ ¡ }tj jt ddd g¡t d
|
d d g¡g fd dtdD dddddgddtj dtj dtj dgdd}t ||¡ d S )Nr ÚXr )r r> )r Úy)r Úzr+ r r r> rP rQ c s$ g | ] }t jd d
|f ¡qS r< )r ZasarrayZiloc)r= Úi©r r r Ú
|
<listcomp>_ s zETestDataFrameToRecords.test_to_records_index_name.<locals>.<listcomp>Zlevel_1Ú0Ú1Ú2ÚOÚf8©ÚnamesÚformats©r )r r r r r Únamer r Úfieldsr r Ú from_tuplesr[ ÚrecZ
|
fromarraysÚarrayÚrangeÚtmÚENDIANÚassert_numpy_array_equal)r Úrsr r) r rS r Útest_to_records_index_nameP s0 ÿ
|
|
|
ûþýz1TestDataFrameToRecords.test_to_records_index_namec C s@ t dddg d¡ ¡ }tjjdgddgd}t ||¡ d S ) Nr> rP )r r r )r> rP )r rX )r rX r] )r Ú set_indexr r ra rb rd Úassert_almost_equal©r r r) r r r Ú"test_to_records_with_unicode_indexm s z9TestDataFrameToRecords.test_to_records_with_unicode_indexc C s t tdddtdddtdddd}tjjddgdtj dfd tj dfd
|
tj dfgd }|jd d }t ||¡ | d¡jdd }t ||¡ | ddg¡jdd }t ||¡ d S )Nú
|
2022-01-01é r )r. rn r )rm rm rm )ú
|
2022-01-02ro ro rV r rW Ú3r] Fr r. T)
|
r r r ra rb rd re r rj ri r( r r r Útest_to_records_index_dtypet s*
|
|
|
ýÿþýû z2TestDataFrameToRecords.test_to_records_index_dtypec C sD t ddgid ¡ }tjjdgddgddgdd }t ||¡ d S )
|
Nu accented_name_éç ð?)r9 )r rr r ú=i8z=f8rZ r] )r r r ra rb rd rj rk r r r Ú)test_to_records_with_unicode_column_names s þz@TestDataFrameToRecords.test_to_records_with_unicode_column_namesc C s t dtdidd}ttdddd}t |d |¡ t tddd}ttdddd}t |d |¡ | ¡ }tjjddd gd
|
d gd}t ||¡ d S ) Nr+ r Úcategoryr] )r r^ r )r r )r. r )rn r )r rs )rU rX )
|
r Úlistr rd Zassert_series_equalr r ra rb rj r( r r r Ú test_to_records_with_categorical¡ s ÿz7TestDataFrameToRecords.test_to_records_with_categoricalzkwargs,expected)r r. çÉ?r )r. rn ç ø?Úbcr Úi8r+ ÚBrY ©ÚCrX r] TÚ column_dtypesZU4©rU rV z0.2r ©rV rW z1.5rz r~ Ú index_dtypesZU1)rU r. rx r )rV rn ry rz ÚUÚunicodeÚU2©r+ r| r~ ©r+ Úi1Úf4r Úint16)r Úi2F)r r )r. rx r )rn ry rz Úfloat32)r z not-there©r+ r| Úint8©r r )r r z)Invalid dtype \[\] specified for column AÚint32é z&Invalid dtype 5 specified for column Br r z-Invalid dtype category specified for column Br5 z$data type ["']foo["'] not understoodc C st t ddgddgddgd}t|tjsXtj|d |d d |jf | W 5 Q R X n|jf |}t ||¡ d S )
|
Nr. rn rx ry r rz r r )Úmatch) r r0 r ZrecarrayÚpytestZraisesr rd rj )r Úkwargsr) r r r r r Útest_to_records_dtype¶ s O z,TestDataFrameToRecords.test_to_records_dtypezdf,kwargs,expectedr. rn r r7 é é r6 é r )r Úfloat64)r r. )r. rn ç @)r7 r ç @)r r6 ç "@Úi4)r r r )r Úd)r Úe)r Úf)r rn )g rV rn r )rr Ú4r r )ç @Ú7r6 r z
|
('a', 'd')z
|
('b', 'e')z
|
('c', 'f')Úab)r[ )r éüÿÿÿ)r éûÿÿÿ)r éúÿÿÿZcd)r r )r r¥ rr r¢ r )r r¦ g @g @r )r r§ r r6 r )r r c C s |j f |}t ||¡ d S r< )r rd rj )r r r r) r r r r Útest_to_records_dtype_mi s J z/TestDataFrameToRecords.test_to_records_dtype_mic C s G dd d}t ddgddgddgd }|f tjtjd
|
tj d d }|jf |}tjjd dgdtj d fddtj dfdgd}t ||¡ d S )Nc @ s8 e Zd ZddddZdd ZedddZd d
|
ZdS ) zBTestDataFrameToRecords.test_to_records_dict_like.<locals>.DictLikeN)Úreturnc [ s | ¡ | _d S r< )Úcopyr )r r r r r Ú__init__Ý s zKTestDataFrameToRecords.test_to_records_dict_like.<locals>.DictLike.__init__c S s | j |¡S r< )r Ú __getitem__©r Úkeyr r r r¬ à s zNTestDataFrameToRecords.test_to_records_dict_like.<locals>.DictLike.__getitem__c S s
|
|| j kS r< )r r r r r Ú __contains__ã s zOTestDataFrameToRecords.test_to_records_dict_like.<locals>.DictLike.__contains__c S s
|
| j ¡ S r< )r Úkeys)r r r r r° æ s zGTestDataFrameToRecords.test_to_records_dict_like.<locals>.DictLike.keys)Ú__name__Ú
|
__module__Ú __qualname__r« r¬ Úboolr¯ r° r r r r ÚDictLikeÜ s rµ r. rn rx ry r rz r r r
r r r r r r| r r} r] )
|
r r r r rd re r ra rb rj )r rµ r Zdtype_mappingsr r) r r r Útest_to_records_dict_likeÚ s
|
þ üþ z0TestDataFrameToRecords.test_to_records_dict_liker- ÚUTCÚGMTr, c C sF t ddd|d}td|i|d}| ¡ }| d¡ ¡ }t ||¡ d S )Nz
|
2016-01-01r ÚS)r Úfreqr- Údatetimer r· )r r r Z
|
tz_convertrd rf )r r- Zdrr r) r r r r Ú%test_to_records_datetimeindex_with_tzü s
|
z<TestDataFrameToRecords.test_to_records_datetimeindex_with_tzN)&r± r² r³ r! r* r3 r; rM rN rh rl rq rt rw r ÚmarkZ parametrizer ra rb rd re Ústrr r r Ú
|
ValueErrorr Ú TypeErrorr r rv ri r r` r¨ r¶ r¼ r r r r r
|
sØ
|
üþþüþþüþþüþþüþþ üþþ
|
ýÿüþø
|
üþþýþþ üþþüþþüþþ
|
þüþûþþþû þ ¼þ M
|
ÿþýþûÿþþüþõ ÿ ÿû
|
þýûúòÙþH
|
"r
|
)Ú collectionsr rG Ú email.parserr Únumpyr r Zpandasr r r r r r Zpandas._testingZ_testingrd r
|
r r r r Ú<module> s
|
