U
|
ý°dG ã @ s¤ d dl mZ d dlZd dlZd dlZd dlmZmZ d dl m
|
Z d dl m Z d dlmZ dd Zdd Zd
|
d Zd d Zdd Zdd Zdd Zdd Zdd ZdS )é )Úcatch_warningsN)Ú DataFrameÚSeries)Úensure_clean_store)Úread_hdfc C s¨ t tj dd¡ tj¡tdtdd}| | }| |d¡ t|d}t
|
||¡ t tj dd¡ tj ¡tdtdd}| | }| |d¡ t|d}t
|
||¡ d S )Né é ÚabcdÚABCDE©ÚindexÚcolumnsÚdf© r ÚnpÚrandomZrandÚastypeÚ complex64ÚlistÚto_hdfr ÚtmÚassert_frame_equalÚ
|
complex128©Útmp_pathÚ
|
setup_pathr ÚpathÚreread© r ú\d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/io/pytables/test_complex.pyÚtest_complex_fixed s$ ý
|
ý
|
r c C s² t tj dd¡ tj¡tdtdd}| | }|j|ddd t|d}t
|
||¡ t tj dd¡ tj ¡tdtdd}| | }|j|ddd d
|
t|d}t
|
||¡ d S ) Nr r r r
|
r r Útable©ÚformatÚw)r# Úmoder r r r r Útest_complex_table( s$ ý
|
ý
|
r& c C s t jddddgt jd}t jddddgt jd}tddddgddd d
|
g||d d d dgdtdd}| | }| |d¡ t|d}t ||¡ d S )Nù ð? ð?©Zdtypeé é é r ÚaÚbÚcÚdç ð?ç @ç @ç @©ÚAÚBÚCÚDÚEr ©r r )
|
r Úarrayr r r r r r r r )r r r r r r r r r r Útest_complex_mixed_fixed@ s*
|
ÿ
|
ÿ
|
|
|
ûø
|
|
r< c C sä t jddddgt jd}t jddddgt jd}tddddgddd d
|
g||d d d dgdtdd}t|@}|jd|ddgd |jddd}t
|
|j |j dk |¡ W 5 Q R X | | }|j |ddd t|d}t
|
||¡ d S )Nr' r( r) r* r+ r r, r- r. r/ r0 r1 r2 r3 r4 r r: r r5 r6 ©Z data_columnszA>2)Úwherer! r" )r r; r r r r r ÚappendÚselectr r Úlocr5 r r ) r r r r r ÚstoreÚresultr r r r r Útest_complex_mixed_tableW s2
|
ÿ
|
ÿ
|
|
|
ûø
|
"
|
rD c C s t dd t ddddg¡}t|tdd}t||d}||g}tjtjg}t ||D ]4\}}| | } |j
|
| ddd t | d}
|
|||
|
qZW 5 Q R X d S )
|
NT©Úrecordr' r r: ©r5 r6 ÚobjZfixedr" ) r r r; r r r r Úassert_series_equalr Úzipr r © r r r Úsr ÚobjsÚcompsrH Úcompr r r r r Ú$test_complex_across_dimensions_fixedt s
|
rP c C s t ddddg¡}t|tdd}t||d}tddT |g}tjg}t||D ]4\}}| | } |j | ddd t
|
| d}
|
|||
|
qTW 5 Q R X d S )
|
Nr' r r: rG TrE rH r! r" ) r r; r r r r r r rJ r r rK r r r Ú test_complex_across_dimensions s
|
rQ c
|
C s t jddddgt jd}tddddgddd d
|
g|d td d }d}t| 2}tjt|d |j d|dgd W 5 Q R X W 5 Q R X d S )Nr' r( r) r* r+ r r, r- r. r/ )r5 r6 r7 r r: úíColumns containing complex values can be stored but cannot be indexed when using table format. Either use fixed format, set index=False, or do not include the columns containing complex values to data_columns when initializing the table.©Úmatchr r7 r= )
|
r r; r r r r ÚpytestÚraisesÚ TypeErrorr? )r r r ÚmsgrB r r r Útest_complex_indexing_error s
|
ÿþÿ
|
rY c C s t ddddg¡}t|tdd}d}| | }tjt|d |j|ddd W 5 Q R X | | }|j|ddd d
|
t|d}t
|
||¡ d S ) Nr' r r: rR rS rH Útr" F)r# r ) r r; r r rU rV rW r r r rI )r r r rL rX r r r r r Útest_complex_series_error¨ s ÿ
|
r[ c C s t tj d¡ tj¡tj d¡d}t| H}|jd|dgd | d|¡ | d¡}t
|
t j ||gdd|¡ W 5 Q R X d S )Néd )r, r- r r- r= r )Zaxis) r r r Zrandnr r r r? r@ r r ÚpdÚconcat)r r rB rC r r r Útest_complex_append¾ s ÿ
|
|
r_ )Úwarningsr Únumpyr rU Zpandasr] r r Zpandas._testingZ_testingr Z pandas.tests.io.pytables.commonr Zpandas.io.pytablesr r r& r< rD rP rQ rY r[ r_ r r r r Ú<module> s
|
