U
|
ý°dÀE ã @ s d dl m Z d dlmZ d dlmZ d dlZd dlZd dlZd dl m
|
Z
|
d dl m Z m Z mZmZmZmZmZ d dlmZ G dd dZdS ) é )Údatetime)ÚDecimal)ÚIteratorN)Úis_platform_little_endian)ÚCategoricalIndexÚ DataFrameÚIndexÚIntervalÚ
|
RangeIndexÚSeriesÚ
|
date_rangec @ sÔ e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dd Z dd Zdd Zdd Zd d Zd d Zd!d" Zd#d$ Zd%d& Zd'd( Zd)d* Zd+d, Zd-d. Zd/d0 Zd1d2 Zd3S )4ÚTestFromRecordsc sB t dddd t fddtdD }t |¡}t ||¡ d S )Nz
|
2016-01-01é
|
z
|
US/Pacific)ZperiodsÚtzc s i | ]
|
}| qS © r ©Ú.0Úi©Zdtir úhd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/constructors/test_from_records.pyÚ
|
<dictcomp> s zBTestFromRecords.test_from_records_dt64tz_frame.<locals>.<dictcomp>é )r r ÚrangeÚ from_recordsÚtmÚassert_frame_equal)ÚselfÚdfÚresr r r Ú test_from_records_dt64tz_frame s
|
z.TestFromRecords.test_from_records_dt64tz_framec C sÖ t st d¡ tdtdddddd gi}t tdddddd g¡g}dg}tjjj ||d}t
|
|¡}t ||¡ t tdddddd g¡g}d g}tjjj ||d}t
|
|¡}|d d
|
¡|d< t ||¡ d S ) Nz*known failure of test on non-little endianÚEXPIRYiÕ é é r )r z<M8[ns]©Údtype)r z<M8[m]zM8[s])r ÚpytestÚskipr r ÚnpÚarrayÚcoreÚrecordsZ
|
fromarraysr r r Zastype)r ÚexpectedZarrdataÚdtypesÚrecarrayÚresultr r r Ú test_from_records_with_datetimes s
|
|
|
z0TestFromRecords.test_from_records_with_datetimesc s t tjtj d¡tjdtjtj d¡tjdtjt d¡tjddgd tjddgd tdtjtj d¡tj dtjtj d¡tj dtjt d¡tj
|
dd}| ¡ }g }g g }| ¡ D ].\} |j¡ | fdd |jD ¡ qÂtt|jD ]>}g }| ¡ D ] \}}| |j| j¡ q| t|¡ q tj||d tj¡} | ¡ }
|
d
|
d |D } t j| d j|jd } t j| d j|jd } t j|
|
d j|jd }t j| d j|jd }tj | |dd t | |¡ t ||¡ tj ||dd t |¡} t | jt d ¡ d¡ d¡g}t t td t | }t j||d} fdd t!|D | _t "| d |d ¡ t "| d |d ¡ d S )Né r# ÚfooTFr! ©ÚAÚA1ÚBÚCÚDÚEÚE1ÚFc s$ g | ] }|t ¡jd d fqS )r r" )r' r$ Údescr)r Úcr# r r Ú
|
<listcomp>P s zBTestFromRecords.test_from_records_sequencelike.<locals>.<listcomp>c S s g | ] }t |qS r )Úlist)r Úxr r r r= Y s ©Úcolumns©Z check_dtypeé r6 r9 )Úexcludec s g | ] } | qS r r r r@ r r r= | s )#r r' r( ÚrandomÚrandnÚfloat64ÚarangeÚint64ÚboolÚfloat32Úint32Ú_to_dict_of_blocksÚitemsÚextendrA r ÚlenÚindexÚilocÚvaluesÚappendÚtupleÚviewr- Ú
|
to_recordsr Úreindexr r Úassert_index_equalr
|
r> ÚsetÚsortedÚassert_series_equal)r r ÚblocksÚtuplesr, Úbr ÚtupÚ_r- Z recarray2Úlistsr. Zresult2Zresult3Zresult4Zcolumns_to_testrD r )rA r$ r Ú test_from_records_sequencelike: sd øÿ ÿÿÿÿ
|
z.TestFromRecords.test_from_records_sequencelikec C sl t jg dddgd}t|dks$tt |jtdddg¡ t g ¡}t|dksVtt|jdkshtd S )Nr1 ÚbarZbazr@ r )r r rP ÚAssertionErrorr rY rA r )r r. r r r Ú$test_from_records_sequencelike_empty s
|
z4TestFromRecords.test_from_records_sequencelike_emptyc C s` t tjtj d¡tjdtjtj d¡tjdtjt d¡tjddgd tjddgd tdtjtj d¡tj dtjtj d¡tj dtjt d¡tj
|
dd}| ¡ }g }| ¡ D ]}| |j¡ qºt| ¡ }dd | ¡ D }g }| t |¡j|jd
|
¡ | t j||d
|
j|jd
|
¡ | t j||d
|
j|jd
|
¡ |D ]}t ||¡ qHd S ) Nr0 r# r1 TFr! r2 c S s i | ]\}}||j qS r )rS )r r? Úyr r r r ¡ s z>TestFromRecords.test_from_records_dictlike.<locals>.<dictcomp>r@ )r r' r( rE rF rG rH rI rJ rK rL rM rS rO rA ÚdictrN rT r rX r r ) r r r] rA r_ ZasdictZasdict2ÚresultsÚrr r r Útest_from_records_dictlike s6 øÿ ÿÿz*TestFromRecords.test_from_records_dictlikec C sL t tj dd¡dddgd}tj d¡}t j||d}t |jt|¡ d S )Nr r! r3 r5 r6 r@ ©rQ ) r r' rE rF r r rY rQ r )r r ÚdataÚdf1r r r Ú!test_from_records_with_index_data° s z1TestFromRecords.test_from_records_with_index_datac C sÄ t tj dd¡dddgd}t j|dgd}t |jt|j ¡ t j|dd}t |jt|j ¡ d
|
d g¡}t j t |d
|
t j|d gd W 5 Q R X t j t |d
|
t j|d d W 5 Q R X d S ) Nr r! r3 r5 r6 r@ rl ú|z"'None of \[2\] are in the columns'©Úmatché )r r' rE rF r r rY rQ r r6 Újoinr% ÚraisesÚKeyError)r r rn Úmsgr r r Ú"test_from_records_bad_index_column· s ÿÿz2TestFromRecords.test_from_records_bad_index_columnc C sb G dd d}|ddd|ddd|d d
|
d g}d d |D }t |¡}t |¡}t ||¡ d S )Nc @ s0 e Zd ZddddZdd ZedddZdS ) z;TestFromRecords.test_from_records_non_tuple.<locals>.RecordN)Úreturnc W s
|
|| _ d S ©N©Úargs)r r| r r r Ú__init__Î s zDTestFromRecords.test_from_records_non_tuple.<locals>.Record.__init__c S s
|
| j | S rz r{ )r r r r r Ú __getitem__Ñ s zGTestFromRecords.test_from_records_non_tuple.<locals>.Record.__getitem__c S s
|
t | jS rz )Úiterr| )r r r r Ú__iter__Ô s zDTestFromRecords.test_from_records_non_tuple.<locals>.Record.__iter__)Ú__name__Ú
|
__module__Ú __qualname__r} r~ r r r r r r ÚRecordÍ s r r" rs r! r é r0 é rC é c S s g | ] }t |qS r )rU )r Úrecr r r r= Ø s z?TestFromRecords.test_from_records_non_tuple.<locals>.<listcomp>©r r r r )r r ZrecsZtupsr. r+ r r r Útest_from_records_non_tupleÌ s
|
"
|
|
z+TestFromRecords.test_from_records_non_tuplec C sP t jg dddgd}tdg}t|dks.t|jjdks>tt |j |¡ d S )Nr1 rd ©rQ rA r )
|
r r r rP re rQ Únamer rY rA ©r r. r+ r r r Ú#test_from_records_len0_with_columnsÞ s
|
|
z3TestFromRecords.test_from_records_len0_with_columnsc C sR t ddddddggj}tdddgdddgg}t |¡}t ||¡ d S )Nr" rs ©Úar_ r! r )r ÚTr r r r )r r+ rm r. r r r Ú"test_from_records_series_list_dictç s
|
z2TestFromRecords.test_from_records_series_list_dictc C s t tddtddtddg}tddiddidd ig|d
|
}tj||d
|
}tddtjgtjtjd gd |d
|
}t ||¡ d S ) Niìÿÿÿiöÿÿÿr r r r" rs r_ r! rl r ) r r r r r r' ÚNaNr r )r rQ Zseries_of_dictsÚframer+ r r r Ú*test_from_records_series_categorical_indexî s ÿ ÿz:TestFromRecords.test_from_records_series_categorical_indexc C s, dt dddtjdd}tj|gdd d S ) Ng ø?iÖ r é )Útzinfo)ZdatumÚ
|
begin_timer rl )r ÚpytzÚutcr r )r r r r r Útest_frame_from_records_utcú s z+TestFromRecords.test_frame_from_records_utcc C s t jddd}ddg|d d
< t |¡}tt t|¡d d d
}tj||d}t |j |¡ t d¡}t
|
t |¡t|¡ d d
|
g¡}t j t|d tj||d d
d W 5 Q R X tj|d d}| ¡ }t|jjd ksæt|jdd}t|jjdkstd|jjkstd S )N)rs z i4,f4,a10r# )r" g @ZHello)rs ç @ZWorldéÿÿÿÿrl )rs r! rp z;Length of values \(2\) does not match length of index \(1\)rq Úf1r! Frs rQ )r' Úzerosr r r rH rP r rY rQ r rt r% ru Ú
|
ValueErrorrW r$ Únamesre )r Úarrr rQ Z indexed_frameZarr2rw r* r r r Ú test_from_records_to_records s(
|
|
ÿÿ z,TestFromRecords.test_from_records_to_recordsc C s: dddg}t j|ddddgd}t |d d ¡s6td S ) N)r" rs Nr! )Nrs r
r! r r_ r< Údr@ r )r r r' Úisnanre ©r r^ r r r r Útest_from_records_nones# s
|
z'TestFromRecords.test_from_records_nonesc C sð t jddddgdt jfdt jfdt jfdt jfgd }tjt|d
|
d }tt jd d gt jd t jd d gt jd t jd
|
dgt jd t jd
|
dgt jd d}t
|
| |¡|¡ ddddg}tjt|ddgd
|
d}t j
|
||j ddgddd d S )N)ç ð?r¨ rs rs )r r r r )ç @r© r0 r0 )ç @rª rC rC r? Úurg Úzr# rs )Únrowsr¨ r r )r? r« rg r¬ )r¨ rs )r r )r© r0 )rª rC )rA r r@ FrB ) r' r( rG rK rI rL r r r r r Z reindex_likerX )r r¢ r Zxpr r r Útest_from_records_iterator) s(
|
üþ üÿ z*TestFromRecords.test_from_records_iteratorc sv dd dddg} fddt t|D }|d |d |d
|
d }t||d } d }tj||d }t ||¡ d S )Nc s s2 t | D ]$}d}|||t| || fV qd S ©NÚABCDEFGHIJKLMNOPQRSTUVWXYZ©r rP ©Úlengthr Úlettersr r r Útuple_generatorD s zKTestFromRecords.test_from_records_tuples_generator.<locals>.tuple_generatorÚIntegerÚStringÚFloatc s" g | ] fd ddD qS )c s g | ] }| qS r r r ©Újr r r= K s zQTestFromRecords.test_from_records_tuples_generator.<locals>.<listcomp>.<listcomp>r r ©r ©rµ r¹ r r= J s zFTestFromRecords.test_from_records_tuples_generator.<locals>.<listcomp>r r" rs ©r¶ r· r¸ r@ r ©r rP r r r r ©r Z columns_namesrA rm r+ Ú generatorr. r r¼ r Ú"test_from_records_tuples_generatorC s
|
|
|
ÿ z2TestFromRecords.test_from_records_tuples_generatorc sv dd dddg} fddt t|D }|d |d |d
|
d }t||d } d }tj||d }t ||¡ d S )Nc s s2 t | D ]$}d}|||t| || gV qd S r¯ r± r² r r r Úlist_generatorU s zITestFromRecords.test_from_records_lists_generator.<locals>.list_generatorr¶ r· r¸ c s" g | ] fd ddD qS )c s g | ] }| qS r r r r¹ r r r= \ s zPTestFromRecords.test_from_records_lists_generator.<locals>.<listcomp>.<listcomp>r r r» ©r r¹ r r= [ s zETestFromRecords.test_from_records_lists_generator.<locals>.<listcomp>r r" rs r½ r@ r r¾ r¿ r rà r Ú!test_from_records_lists_generatorT s
|
|
|
ÿ z1TestFromRecords.test_from_records_lists_generatorc C s< dddg}dddg}t |}tj||dd}||ks8td S )N©r" rs r! )rs r
r! r r_ r< )rA rQ )r> r r re )r r^ rA Zoriginal_columnsr r r r Ú&test_from_records_columns_not_modifiede s
|
|
|
z6TestFromRecords.test_from_records_columns_not_modifiedc C sz t dft dfdg}tj|dgd}|d jtks8ttj|dgdd}|d jtjks^tt |d j d ¡svtd S ) Nz1.5z2.5rz r r@ T)rA Z coerce_floatr )
|
r r r r$ Úobjectre r' rG r¥ rS r¦ r r r Útest_from_records_decimalo s z)TestFromRecords.test_from_records_decimalc C s> t jddgdddgd}t ddgdddgd}t ||¡ d S )NrÅ )r r
r0 r r_ r@ r r r r r Ú test_from_records_duplicatesy s z,TestFromRecords.test_from_records_duplicatesc sr dd fddt dD }| ddd¡ tj|dd }|jjdksLttj|dd
|
gd }|jjdksntd S ) Nc S s | t j dd¡t j dd¡dS )Nr" r )Úorder_idÚquantityZprice)r' rE Úrandint)rÊ r r r Ú create_dict s ýzETestFromRecords.test_from_records_set_index_name.<locals>.create_dictc s g | ] } |qS r r r ©rÍ r r r= s zDTestFromRecords.test_from_records_set_index_name.<locals>.<listcomp>r r
)rÊ rË rÊ rl rË )r rT r r rQ r re r¡ )r Z documentsr. r rÎ r Ú test_from_records_set_index_name s z0TestFromRecords.test_from_records_set_index_namec C sp dgdgd}t j|ddgd}t |ddgd}t ||¡ ddd gd
|
d d gd }t j|dddgd}t |dddgd}t ||¡ g }| tddddg¡ | tddddg¡ t j|ddgd}|j}tt d¡t d¡gddgd}t
|
||¡ g }| tddddg¡ | tddddg¡ t j|ddgd}|j}tt d¡t d¡gddgd}t
|
||¡ d S )Nr1 rd ©r" rs r r_ r@ r" rs r! r r
r0 r r< rl iÚ ÚhiÚdateÚtestzdatetime64[ns]rÇ rI ) r r r r rT r r, r r' r$ r\ )r rm r. ÚexpÚrowsZdf2_objr+ r r r Ú!test_from_records_misc_brokenness s8 ÿ ÿz1TestFromRecords.test_from_records_misc_brokennessc C sd t jg dddgd}t dddgd}t ||¡ t jg dddgd}t dddgd}t ||¡ d S )Nr r_ r< r@ r r r r r Útest_from_records_empty¹ s z'TestFromRecords.test_from_records_emptyc C s t jdgdt jfdt jfgd}tj|dd}tdgdd}tddgi|dgd }t ||¡ |d d
|
}tj|dd}t ||jd d
|
¡ d S ) NrÐ ÚidÚvaluer# rl r" )r rs r r ) r' r( rI r r r r r rR )r r r Zex_indexr+ r_ Zdf2r r r Ú3test_from_records_empty_with_nonempty_fields_gh3682Ã s zCTestFromRecords.test_from_records_empty_with_nonempty_fields_gh3682c C sh dt fg}dt|f}tj||d}t |¡}tdtjg t di}t ||¡ t|}t ||¡ d S )NÚpropr r# ) ÚintrP r' Úemptyr r r( r r )r r$ Úshaper¢ r. r+ Zaltr r r Útest_from_records_empty2Ï s
|
|
z(TestFromRecords.test_from_records_empty2N) r r r r r/ rc rf rk ro rx r r r r r r£ r§ r® rÁ rÄ rÆ rÈ rÉ rÏ rÖ r× rÚ rß r r r r r s2 F
|
& #
|
|
&
|
r )r Údecimalr Útypingr Únumpyr' r% r Z pandas.compatr Zpandasr r r r r
|
r r Zpandas._testingZ_testingr r r r r r Ú<module> s $
|
