U
|
ý°dP2 ã @ s d dl Z d dlZd dlZd dlmZ d dlmZmZ d dl m
|
Z
|
m Z m Z m ZmZ d dlmZmZmZ d dlmZ d dlmZ ejjZej dd d
|
d d g¡d d Z dd Z dd Z dd Z dd Z dd Z!dd Z"dd Z#ejj$e d d d d Z%ej d!d"d#d$d%d&gd"d$d%d&gd'd#d$d%d&gd"d(d$d%d&gd#d$d%d&gd(d$d%d&gd)d#d$d%d&gej&d#d%d&gd)ej&d#d%d&gg ¡ej d*d+e'g¡d,d- Z(d.d/ Z)d0d1 Z*dS )2é N)Úis_platform_little_endian)ÚClosedFileErrorÚPossibleDataLossError)Ú DataFrameÚHDFStoreÚSeriesÚ_testingÚread_hdf)Ú _maybe_removeÚensure_clean_storeÚtables)Úpytables)ÚTermÚmodeÚrúr+ÚaÚwc
|
C s t ¡ }d}|| }|dkrDtjt|d t||d W 5 Q R X n(t||d}|jj|ksbtW 5 Q R X || }|dkr°tjt|d t||d}W 5 Q R X W 5 Q R X n(t||d}|jj|ksÎtW 5 Q R X || }|dkr&tjt|d |j |d|d W 5 Q R X |j |ddd n|j |d|d |dkrnd}tjt
|
|d t |d|d W 5 Q R X nt |d|d}t ||¡ d S ) Nz[\S]* does not exist)r r ©Úmatch©r Údfr )r zNmode w is not allowed while performing a read. Allowed modes are r, r\+ and a.) ÚtmÚmakeTimeDataFrameÚpytestÚraisesÚOSErrorr Ú_handler ÚAssertionErrorÚto_hdfÚ
|
ValueErrorr Úassert_frame_equal)Ú
|
setup_pathÚtmp_pathr r ÚmsgÚpathÚstoreÚresult© r( úbd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/io/pytables/test_file_handling.pyÚ test_mode s8
|
|
ÿr* c C s: t ¡ }| | }|j|ddd t|d}t ||¡ d S )Nr r r )r r r r r! )r# r" r r% r' r( r( r) Útest_default_modeP s
|
|
r+ c C s | | }t |dd}t ¡ |d< d}tjt|d | d¡ W 5 Q R X | ¡ |jrZt | d¡ |jsnt t
|
|dks~t | ¡ |jrt t |dd}t ¡ |d< | d¡ |js¼t t
|
|dksÌt |j dksÚt | ¡ |jrìt | d¡ |jst t
|
|dkst |j dks$t | ¡ |jr8t | d¡ |jsNt t
|
|dks`t |j dkspt | ¡ |jrt d S ) Nr r zKRe-opening the file \[[\S]*\] with mode \[a\] will delete the current file!r r r r é ) r r ÚmakeTimeSeriesr r r ÚopenÚcloseÚis_openr ÚlenÚ_mode)r# r" r% r& r$ r( r( r) Útest_reopen_handleY sD ÿ
|
|
|
|
|
|
|
|
|
r3 c C s~ t | ¡Z}t ¡ }t|dddd}||d< | d|¡ t |d |¡ t |d |¡ | ¡ W 5 Q R X tj |¡rzt
|
d S )Nr Z H5FD_COREr )r ZdriverZdriver_core_backing_storer Údf2) r Ú ensure_cleanÚ makeDataFramer Úappendr! r/ Úosr% Úexistsr )r" r% r r& r( r( r) Útest_open_args s ÿ r: c C s8 t | &}t ¡ |d< | ¡ |jdd W 5 Q R X d S )Nr T)Úfsync)r r r- Úflush)r" r& r( r( r) Ú
|
test_flush¡ s
|
r= c C s t ¡ }| | }|j|ddd t|d}t ||¡ tj|dd:}|jdddD ]$}|jj dksht
|
|jj d ksTt
|
qTW 5 Q R X | | }|j|dd d
|
t|d}t ||¡ tj|dd:}|jdddD ]$}|jj d ksät
|
|jj d ksÐt
|
qÐW 5 Q R X | | }| |d¡ t|d}t ||¡ tj|dd@}|jdddD ]*}|jj d ks^t
|
|jj d ksHt
|
qHW 5 Q R X d S ) Nr é )Ú complevelr r ú/dfÚLeaf©ÚwhereÚ classnameÚzlib)Úcomplibr ) r r6 r r r! r Ú open_fileÚ
|
walk_nodesÚfiltersr? r rF )r# r" r Útmpfiler' Úh5fileÚnoder( r( r) Ú test_complibs_default_settings¨ s2
|
|
|
rM c C sÄ t ¡ }| | }t|}|jd|ddd | d|¡ | ¡ tj|ddn}|jdd d
|
D ]$}|jj d kspt
|
|jj d ks\t
|
q\|jd d d
|
D ]$}|jj dks¤t
|
|jj dkst
|
qW 5 Q R X d S ) NZdfcr> Zblosc)r? rF r r r r@ rA rB r z/dfc) r r6 r r7 r/ r rG rH rI r? r rF )r# r" r rJ r& rK rL r( r( r) Ú'test_complibs_default_settings_overrideÏ s rN c
|
s t ¡ }tjj}t d¡s$| d¡ t d¡s8| d¡ tdd fdd|D }|D ]¨\}}| | }d}|j||||d t ||} t
|
| |¡ tj |d d
|
X}
|
|
|
j d | d d D ]>} | jj |ksÊt|dkrä| jjd ksôtq¶| jj|ks¶tq¶W 5 Q R X qXd S )NZlzoÚbzip2r é
|
c s g | ]} D ] }||fq qS r( r( )Ú.0ÚlibÚlvl©Z
|
all_levelsr( r) Ú
|
<listcomp>ï s z!test_complibs.<locals>.<listcomp>Úfoo)rF r? r r ú/rA rB )r r6 r rI Ú all_complibsZwhich_lib_versionÚremoveÚranger r r! rG rH r? r rF ) r# r" r rX Z all_testsrR rS rJ Úgnamer' Zh5tablerL r( rT r) Ú test_complibsá s(
|
|
|
|
|
|
r\ z$reason platform is not little endian)Úreasonc C s t | }tdddtdd}tj|jd< tj|jd< t|d |jd|d d
|
t |d |¡ |j
|
d gd }| dt d d d
|
¡}t ||¡ W 5 Q R X d S )NrV Úbar)ÚAÚBé )Úindex)é r_ )é r` r Úascii)Úencodingr_ )Úcolumnsz columns=A) r r rZ ÚnpÚnanÚlocr
|
r7 r r! ZreindexÚselectr )r" r& r Úexpectedr' r( r( r) Ú test_encoding s
|
|
rm Úvals EÉ, 17ó ó aó bó cs EE, 17s øüs AøüÚdtypeÚcategoryc
|
sn d d}d} fdd|D }t ||d}| | }|j||d |d t||}| |tj¡} t | |¡ d S ) Nzlatin-1Ú Údatac s$ g | ] }t |tr | ¡n|qS r( )Ú
|
isinstanceÚbytesÚdecode)rQ Úx©Úencr( r) rU ) s z'test_latin_encoding.<locals>.<listcomp>)rs Útable)Úformatrf Únan_rep)r r r Úreplacerh ri r Zassert_series_equal)
|
r# r" rs rn r ÚkeyZserr& ZretrZs_nanr( r{ r) Útest_latin_encoding s
|
r c C sD | | }t ¡ }|j|dddd t|}d| ¡ ks:t|jsDt| ¡ d| ¡ ks\t|jrft| | }tj r®t|}d}t
|
j t |d t| W 5 Q R X | ¡ n°t|}t|}d| ¡ ksÎtd| ¡ ksÞt|jsèt|jsòt| ¡ d| ¡ ks t|jrtd| ¡ ks*t|js6t| ¡ d| ¡ ksPtd| ¡ ksbt|jrnt|jrztt|dd}| d|¡ t|}| d |¡ | ¡ d| ¡ ksÀt|jrÌt| ¡ d| ¡ ksæt|jròtt|dd}| d|¡ t|}| ¡ d| ¡ ks,t|jr8t| ¡ d| ¡ ksRt|jr^t| | }t ¡ }|j|dddd t|}| ¡ d
|
}t
|
j t|d | ¡ W 5 Q R X t
|
j t|d d|k W 5 Q R X t
|
j t|d t| W 5 Q R X t
|
j t|d |d W 5 Q R X t
|
j t|d | d¡ W 5 Q R X t
|
j t|d | d¡ W 5 Q R X t
|
j t|d | d |¡ W 5 Q R X t
|
j t|d | d |¡ W 5 Q R X t
|
j t|d | d ¡ W 5 Q R X t
|
j t|d | d ¡ W 5 Q R X t
|
j t|d | d¡ W 5 Q R X d }t
|
j t|d |j W 5 Q R X d S ) Nr r r} )r r~ ZCLOSEDzTThe file [\S]* is already opened\. Please close it before reopening in write mode\.r r r4 z[\S]* file is not open!Zdf3z''HDFStore' object has no attribute 'df')r r6 r r Úinfor r0 r/ r Z!_table_file_open_policy_is_strictr r r r7 r Úkeysr1 rk ÚgetÚputZ
|
get_storerrY ÚAttributeErrorr )r# r" r% r r& Zstore1r$ Zstore2r( r( r) Útest_multiple_open_close5 s¦
|
|
ÿ
|
|
r c
|
C sD t d¡0} t| }t |¡t| ks,tW 5 Q R X W 5 Q R X d S )Nzfoo.h5)r r5 r r8 ÚfspathÚstrr )r% r& r( r( r) Ú test_fspath´ s
|
r )+r8 Únumpyrh r Z pandas.compatr Z pandas.errorsr r Zpandasr r r r r r Z pandas.tests.io.pytables.commonr
|
r r Z pandas.ior Zpandas.io.pytablesr ÚmarkZ
|
single_cpuZ
|
pytestmarkZ parametrizer* r+ r3 r: r= rM rN r\ Zskipifrm ri Úobjectr r r r( r( r( r) Ú<module> sP
|
0 4'# ÿ
|
|
|
|
÷þ
|
