U
|
ý°d/. ã @ sf d dl Z d dlZd dlmZmZ d dlZd dlZd dlm Z d dl
|
Z d dl
|
m Z m Z mZmZmZmZmZ d dlmZmZ d dlmZ ejjZ dd Z d d
|
Z d d Z d d Z dd Z!dd Z"ej#dd Z$dd Z%ej &ddej'gdej(gdej)gdej*gdej'gdej(gdej)gdej*gdej+gdej+gg
|
¡dd Z,d d Z-d d Z.ej &d ddg¡d!d" Z/dS )#é N)Úcatch_warningsÚ simplefilter)Ú Timestamp)Ú DataFrameÚHDFStoreÚIndexÚ
|
MultiIndexÚSeriesÚ_testingÚconcat)Ú _maybe_removeÚensure_clean_store)Ú_test_decoratorsc C st t dddgi}t| | N}|jd|dd |jd|dd | d¡jdksRt| d¡jdksftW 5 Q R X d S ) NÚAé é ÚaÚfixed©ÚformatÚbÚtable)r r ÚputÚ
|
get_storerZ format_typeÚAssertionError)Útmp_pathÚ
|
setup_pathÚdfÚstore© r úXd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/io/pytables/test_put.pyÚtest_format_type s r! c C s2 d}t jt|d t| | dd W 5 Q R X d S )Nz-format is not a defined argument for HDFStore©Úmatchr r )ÚpytestÚraisesÚ
|
ValueErrorr )r r Úmsgr r r Ú test_format_kwarg_in_constructor* s r( c C sð t |Ú}t ¡ }t dd¡V t|d | d|¡ | d¡jrFt d}t
|
j t |d | d|¡ W 5 Q R X W 5 Q R X t dd¡R t|d | d|¡ | d¡js®t t|d | d|¡ | d¡jsÔt W 5 Q R X W 5 Q R X | | }t ¡ }t dd¡b | |d¡ t|}| d¡jr.t W 5 Q R X t
|
j t |d |j|ddd W 5 Q R X W 5 Q R X t dd¡n | |d
|
¡ t|}| d
|
¡js¢t W 5 Q R X |j|d dd t|}| d ¡jsØt W 5 Q R X W 5 Q R X d S ) Nzio.hdf.default_formatr r úCan only append to Tablesr" Údf2r T©ÚappendÚdf3Zdf4)r ÚtmZ makeDataFrameÚpdZoption_contextr r r Zis_tabler r$ r% r& r, Úto_hdfr )r r r r r' Úpathr r r Útest_api_default_format3 s>
|
|
|
|
$
|
$
|
|
r2 c
|
C s` t | L}t ¡ }t ¡ }||d< |d d
|d< |d d
|d< |d d
|d< |d d
|d< |jd|d d
dd d
|
}tjt|d |jd|dd
d d W 5 Q R X t|d tjt|d |jd|dd
d d W 5 Q R X tjt|d |jd|dd
d d W 5 Q R X |jd|d d
ddd t |d d
|d ¡ W 5 Q R X d S )Nr é
|
r z foo/bar/bahÚfooz/fooÚcr r r) r" Tr+ ÚfF)r r, )
|
r r. ZmakeTimeSeriesÚmakeTimeDataFramer r$ r% r& r Úassert_frame_equal)r r Útsr r' r r r Útest_put] s& "
|
""r: c C sê t | Ø}tdd tdD }tt d¡|d}t||d}||d< t |d |¡ ||d< t |d |¡ tdgd d tdD }tt d
|
¡|d}t||d}||d< t |d |¡ ||d< t |d |¡ W 5 Q R X d S ) Nc S s g | ]}d | qS ©z I am a very long string index: r ©Ú.0Úir r r Ú
|
<listcomp>~ s z)test_put_string_index.<locals>.<listcomp>é ©Úindex)r ÚBr r Z$abcdefghijklmnopqrstuvwxyz1234567890c S s g | ]}d | qS r; r r< r r r r? s é )
|
r r Úranger ÚnpÚaranger r. Úassert_series_equalr8 )r r rB Úsr r r r Útest_put_string_index| s&
|
ÿÿrJ c
|
C sr t | `}t ¡ }|jd|ddd t |d |¡ d}tjt|d |jd|ddd W 5 Q R X W 5 Q R X d S ) Nr5 r Úzlib©r Zcomplibú0Compression not supported on Fixed format storesr" r r )r r. r7 r r8 r$ r% r& )r r r r' r r r Útest_put_compression s
|
rN c
|
C sr t ¡ }t| X}d}tjt|d |jd|ddd W 5 Q R X |jd|ddd t |d |¡ W 5 Q R X d S ) NrM r" r r ZbloscrL r5 r )r. r7 r r$ r% r& r r8 )r r r r' r r r Útest_put_compression_blosc£ s
|
rO c
|
C s t ¡ }d|d< d|d< |d dk|d< |d dk|d < d
|
|d < d |d < d|d< td|d< td|d< t dd ddd¡|d< t dd ddd¡|d< tj|j|jdd
dgf< | ¡ }t | V}t
|
|d t d
|
d t dt jj | d|¡ W 5 Q R X | d¡}t ||¡ W 5 Q R X d S ) Nr4 Zobj1ÚbarÚobj2r r Zbool1rC Zbool2TZbool3r Zint1r Zint2Z20010102Z
|
timestamp1Z20010103Z
|
timestamp2iÑ Z datetime1é Z datetime2é r )ÚrecordÚignore)r. r7 r ÚdatetimerF ÚnanÚlocrB Z _consolidater r r r r/ ÚerrorsZPerformanceWarningr Úgetr8 )r r r Úexpectedr r r Útest_put_mixed_type± s*
|
|
|
r\ z format, indexr r c C sj t | X}ttj dd¡tdd}|t||_t|d |j d||d t
|
||d ¡ W 5 Q R X d S )Nr3 r ZAB©Úcolumnsr r ) r r rF ÚrandomÚrandnÚlistÚlenrB r r r. r8 )r r rB r r r r r Útest_store_index_typesÍ s
|
|
rc c
|
C sÆ t jddddgddgd}tt d¡ d d
|
¡|d }| |j ¡ ¡}t | ¼}|
|
d |¡ t j |d |d d d |j
|
d|dd t j |d |d d d t d¡}tjt|d |j
|
d|ddgd W 5 Q R X t d¡}tjt|d |j
|
d|dd d W 5 Q R X W 5 Q R X t | 6}| d|¡ | d|¡ t |d t||f¡ W 5 Q R X tt d¡ d d
|
¡ttdddd }| |j ¡ ¡}t | ,}|j
|
d|dd t j |d |d d d W 5 Q R X d S ) N)r r )r r )rC r )rC r ÚfirstÚsecond©Únamesé rR é r] r T)Zcheck_index_typeZcheck_column_typeZdf1r r z<cannot use a multi-index on axis [1] with data_columns ['A']r" r* r )r Z data_columnsz;cannot use a multi-index on axis [1] with data_columns Truer- ÚABCDr4 )Úname)r Ú from_tuplesr rF rG ZreshapeZset_axisrB Zto_numpyr r r. r8 ÚreÚescaper$ r% r& r, r r ra )r rB r r[ r r' r r r Útest_column_multiindexé sR
|
ÿ
|
ÿ ÿ
|
|
&
|
"&
|
ÿro c
|
C s t | î}ddd}t|d tt d¡ddg| d}| d|¡ t | d¡|¡ t|d tt d¡ddg|dd d gd}| d|¡ t | d¡|¡ t|d t t d
|
¡|dd d gd }| d |¡ t t d
|
¡|dd d gd }t
|
| d ¡|¡ t|d tt d¡ddg|dddgd}d}t j t |d | d|¡ W 5 Q R X t|d tt d¡ddg|dddgd}t j t |d | d|¡ W 5 Q R X t|d tt d¡ddg|dd dgd}| d|¡ t | d¡|¡ W 5 Q R X d S )Nc S s t jdd tddD | dS )Nc S s< g | ]4}t d D ]&}t dD ]}t dd|¡||fq qqS )r rR iÝ rh )rE rV )r= ÚdrI Útr r r r? s
|
|
ýz=test_store_multiindex.<locals>.make_index.<locals>.<listcomp>r rR rf )r rl rE rf r r r Ú
|
make_index s þùz)test_store_multiindex.<locals>.make_indexr )rh r r r )r^ rB ÚdaterI rh rA Zlevel_1Zlevel_2rq zBduplicate names/columns in the multi-index when storing as a tabler" )N)r r r rF Úzerosr, r. r8 Úselectr rH r$ r% r& )r r rr r rI Zxpr' r r r Útest_store_multiindex sX
|
|
|
ý
|
|
ý
|
ý
|
ý rv r c C sX t tj dd¡tjddddd}| | }|j|dd|d t |d¡}t ||¡ d S )
|
Né r Z20220101ÚM)ÚfreqZperiodsrA r Úw)Úmoder )
|
r rF r_ r` r/ Z period_ranger0 Zread_hdfr. r8 )r r r r r1 r[ r r r Útest_store_periodindex^ s ÿ r| )0rV rm Úwarningsr r ÚnumpyrF r$ Zpandas._libs.tslibsr Zpandasr/ r r r r r r
|
r. r Z pandas.tests.io.pytables.commonr r Z pandas.utilr ÚtdÚmarkZ
|
single_cpuZ
|
pytestmarkr! r( r2 r: rJ rN Zskip_if_windowsrO r\ Z parametrizeZmakeFloatIndexZmakeStringIndexZ makeIntIndexZ makeDateIndexZmakePeriodIndexrc ro rv r| r r r r Ú<module> sL $
|
*
|
öþ
|
.G
|
