U
|
ý°dJ6 ã @ sX d dl Z d dlZd dlZd dlZd dlmZ d dlm Z d dl
|
m Z G dd de Z dS )é N)ÚExtensionArray)Ú EABackedBlock)ÚBaseExtensionTestsc @ sp e Zd ZdZej dddg¡dd Zej dddg¡dd Zd d
|
Z d d Z
|
d d Z dd Z dd Z dd Zdd Zdd Zdd Zdd Zd d Zd d Zd!d" Zej d#d$d%gejjd&d'gd(d)gd*g¡d+d, Zej d-ejjd$d%gd.d/ggd.d/gd*ej d&d'd0g¡ej d1d2d3g¡ej d4d5d6d7d8g¡g¡ej d9d:d;g¡d<d= Zd>d? Z d@dA Z dBdC Z dDS )EÚBaseReshapingTestsz&Tests for reshaping and concatenation.Úin_frameTFc C s¢ t |¡}|rt |¡}t j||gdd}t|t|d ksBt|rR|jd }n|j}||jksftt|j drt
|
|j j d t stt
|
|j j d tstd S )NT©Z ignore_indexé r Úblocks)ÚpdÚSeriesÚ DataFrameÚconcatÚlenÚAssertionErrorÚdtypesÚdtypeÚhasattrZ_mgrÚ
|
isinstancer r Zarraysr )ÚselfÚdatar ÚwrappedÚresultr © r ú\d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/extension/base/reshaping.pyÚ test_concat s
|
|
z BaseReshapingTests.test_concatc C s¼ t j| ddg¡ddgd}t j| ddg¡ddgd}|rXt d|i¡}t d|i¡}t ||g¡}|rt d| ddddg¡i¡}| ||¡ n$t | ddddg¡¡}| ||¡ d S )Né r ©Úindexr é Úa)r
|
r Útaker r Úassert_frame_equalÚassert_series_equal)r Z data_missingr Z valid_blockZna_blockr Úexpectedr r r Útest_concat_all_na_block" s z+BaseReshapingTests.test_concat_all_na_blockc C s4 t d|d d
i¡}t ddddgi¡}t ddddgi¡ d¡}|||g}t |¡}t d d
|
|D ¡}| ||¡ t d d
|
|D ¡}t d d
|
|D ¡}| ||¡ t || t¡g¡}t | d ¡| d ¡g¡}| ||¡ t |d |d t¡g¡}t |d d ¡|d d ¡g¡}| ||¡ d S )NÚAr r r r ÚbÚcÚcategoryc S s g | ]}| t¡qS r ©ÚastypeÚobject©Ú.0Úxr r r Ú
|
<listcomp>: s z?BaseReshapingTests.test_concat_mixed_dtypes.<locals>.<listcomp>c S s g | ] }|d qS ©r% r r, r r r r/ > s c S s g | ]}|d t¡qS r0 r) r, r r r r/ ? s r+ )r
|
r r* r r! r" r+ )r r Údf1Údf2Zdf3Údfsr r# r r r Útest_concat_mixed_dtypes1 s
|
|
"z+BaseReshapingTests.test_concat_mixed_dtypesc C s* t d|d d
i¡}t ddddgi¡}t |d d
dddgd¡}t j||gdd}| ||¡ t j|d |d gdd}| ||¡ t jddddgidddgd}t |jt|d d
|g |jd tjdddgd¡}t j||gdd}| ||¡ t j|d |d gdd}| ||¡ d S )
|
Nr% r ÚBr r ©r% r5 ©Úaxisr ©r ) r
|
r r r! Ú_from_sequenceÚlistr ÚnpÚnan©r r Úna_valuer1 r2 r# r r r r Útest_concat_columnsK s" þÿ z&BaseReshapingTests.test_concat_columnsc C s t d|d d
i¡}t d|dd
i¡}t |jt|d d
|g |jd|dd
d¡}t j||gddd }| ||¡ d S )
|
Nr% r r5 é r9 r6 r F)r8 Úcopy)r
|
r r: r; r r r! r> r r r Ú'test_concat_extension_arrays_copy_falsec s
|
þÿz:BaseReshapingTests.test_concat_extension_arrays_copy_falsec C s t d|d d
i¡}t d|d d
i¡}t j||gdd}t |jttddgd dd|jdgd ttd ddd¡}| ||¡ d S ) Nr é r& Tr éÿÿÿÿ)Z
|
allow_fill)r r& )r
|
r r r r; Úranger! )r r r r& r r# r r r Útest_concat_with_reindexp s þÿz+BaseReshapingTests.test_concat_with_reindexc C s |d d
}|dd
}t |¡ t j|dddgd¡\}}t |jt||g |jd¡}t |j|gt| |jd¡}| ||¡ | ||¡ d S )Nr r rD r r r9 )r
|
r Úalignr: r; r r" © r r r? r r& Úr1Úr2Úe1Úe2r r r Ú
|
test_align} s $ z BaseReshapingTests.test_alignc C s¨ |d d
}|dd
}t d|i¡ t jd|idddgd¡\}}t d|jt||g |jdi¡}t d|j|gt| |jdi¡}| ||¡ | ||¡ d S )Nr r rD r% r r r9 )r
|
r rH r: r; r r! rI r r r Útest_align_frame s , ÿ ÿ z#BaseReshapingTests.test_align_framec C sz t j|dd}t dt t|d ¡i¡}| |¡\}}t j|jt||g |j d|j
|
d}| ||¡ | ||¡ d S )Nr )ÚnameÚcolr r9 ) r
|
r r r< Zaranger rH r: r; r rP r" r! )r r r? ÚserÚdfrJ rK rL r r r Útest_align_series_frame s þ z*BaseReshapingTests.test_align_series_framec C sJ t ddgt| i¡}||d< t dgt| |d¡}| ||¡ d S ©Nr% r r5 r6 ©r
|
r r r! ©r r rS r# r r r Ú,test_set_frame_expand_regular_with_extension¥ s z?BaseReshapingTests.test_set_frame_expand_regular_with_extensionc C sJ t d|i¡}dgt| |d< t |dgt| d¡}| ||¡ d S rU rV rW r r r Ú,test_set_frame_expand_extension_with_regular« s z?BaseReshapingTests.test_set_frame_expand_extension_with_regularc C s< t jddgt| itd}||d< |jd |jks8td S )Nr% r r9 )r
|
r r r+ r r r )r r rS r r r Ú test_set_frame_overwrite_object± s z2BaseReshapingTests.test_set_frame_overwrite_objectc C s6 t |d d
dddgdddgd¡}t ddddgddddgd¡}t ||¡}t dddgdddgdddg|j|d |d |d g|jdd ¡}| ||d
|
d d d g ¡ t j||dd}t ddddtjgdddtjdgdddddg|j|d |d |d |d |g|jdd ¡}| ||d
|
d d d g ¡ d S )Nr r r r )ÚextÚint1Úkeyé )Úint2r] r9 )r\ r_ r] r[ r[ r\ r] r_ Úouter)Úhow)r
|
r Úmerger: r r! r< r= )r r r? r1 r2 ÚresÚexpr r r Ú
|
test_merge· s4 & ÿüÿ
|
ÿüÿ
|
z BaseReshapingTests.test_mergec C s² |d d
\}}t |j||g|jd}t |ddgd¡}tj||dd}t |ddgddgd¡}| ||¡ tj|jddg |dd}|jddg jd d
|
}| ||¡ d S ) Nr r9 r ©r] Úvalr] ©Úon©r] Zval_xZval_yr T)Zdrop) Útyper: r r
|
r rb r! ÚilocZ reset_index)r r r r& r] rS r r# r r r Ú test_merge_on_extension_arrayÖ s z0BaseReshapingTests.test_merge_on_extension_arrayc C s¬ |d d
\}}t |j|||g|jd}t |dddgd¡}t |dddgd¡}tj||dd}t | dddddg¡dddddgdddddgd ¡}| ||¡ d S )
|
Nr r9 r r rf r] rh r rj )rk r: r r
|
r rb r r! ) r r r r& r] r1 r2 r r# r r r Ú(test_merge_on_extension_array_duplicateså s ýÿz;BaseReshapingTests.test_merge_on_extension_array_duplicatesÚcolumnsr% r5 )r% r )r% r& r` Úinner)Únamesc C s® t |d d
|d d
d¡}||_| ¡ }| t¡ ¡ }| t¡}t|t jrr|j|j d d
df jkst
|
n"t |j |j d d
df jkst
|
| t¡}| ||¡ d S )NrD r6 r )r
|
r ro Ústackr* r+ r r r rl r Úallr Z assert_equal)r r ro rS r r# r r r Ú
|
test_stackö s
|
|
"
|
z BaseReshapingTests.test_stackr r r& )r5 r& r6 )r r& r' )r r r )r% r r )r% r& r )r% r r )r5 r r )r5 r' r ÚobjÚseriesÚframec s( d t |
|dkr(tj |d}ntj d|d}|j}tt|tj fddtd|D ¡}|D ]°}|j
|
|dt fddj D s t |dkrÒ| ¡ }|j
|
|djd dd
|
} | | ¡ | t¡}
|
|
|
j
|
| jjd } |dkr | jtk ¡ s t t¡| | ¡ qrd S ) Nrv r r6 c 3 s | ]}t |¡V qd S ©N)Ú itertoolsÚ permutations)r- Úi)Úlevelsr r Ú <genexpr>0 s z2BaseReshapingTests.test_unstack.<locals>.<genexpr>r )Úlevelc 3 s" | ]}t | jt V qd S rx )r Úarrayrk )r- rQ )r r r r r} 6 s r r7 )r~ Z
|
fill_value)r r
|
r r Znlevelsr; rF ry ÚchainÚ from_iterableZunstackrs ro r Zto_frameZ droplevelr! r* r+ r r? r ) r r r ru rR ÚnÚ combinationsr~ rS ZaltZobj_serr# r )r r| r r Ú test_unstack s0 ÿ ÿ
|
|
|
z BaseReshapingTests.test_unstackc C s@ | ¡ }t|t|ks t|d |d< |d |d ks<td S )Nr r )Zravelrk r ©r r r r r r Ú
|
test_ravelJ s z BaseReshapingTests.test_ravelc C sf | ¡ }t|t|ks t||k s(t|j|jd d d
ksBt|d |d< |d |d ksbtd S )NrE r r )Ú transposerk r Úshaper
r r r Útest_transposeS s z!BaseReshapingTests.test_transposec C s t j|d d
|d d
dddddgd}|j}t jt|j|d gd |jd
|
t|j|d gd |jd
|
t|j|d gd |jd
|
t|j|d gd |jd
|
d ddgd}| ||¡ | t t |¡¡|¡ | t t |dg ¡¡|dg ¡ d S )Nr^ r6 r r& r' Údr r r r9 r r )r r& r' r r% r5 ) r
|
r ÚTrk r: r r! r< r )r r rS r r# r r r Útest_transpose_framea s , üù z'BaseReshapingTests.test_transpose_frameN) Ú__name__Ú
|
__module__Ú __qualname__Ú__doc__ÚpytestÚmarkZ parametrizer r$ r4 r@ rC rG rN rO rT rX rY rZ re rm rn r
|
Z
|
MultiIndexÚ from_tuplesrt Z from_productr r r r r r r r r s`
|
|
ÿþþ
|
ûÿùþ % r ) ry Únumpyr< r Zpandasr
|
Zpandas.api.extensionsr Z pandas.core.internals.blocksr Z pandas.tests.extension.base.baser r r r r r Ú<module> s
|
