U
|
ý°d4E ã @ sà d dl Z d dlZd dlmZmZ d dlZd dlZd dlm Z d dl
|
m Z d dl Z d dl mZmZmZmZmZmZ d dlmZmZ d dlmZ dZej j Z d d
|
Z d d Z d d Z!dd Z"dd Z#dd Z$dd Z%dd Z&dd Z'ej (d¡d d Z)d d Z*d d! Z+ej (d¡d"d# Z,d$d% Z-ej .d&d'ej/d(ej0d)g¡d*d+ Z1d,d- Z2ej .d.ej3ej4e5d/d0g¡d1d2 Z6d3d4 Z7d5d6 Z8ej .d&d'ej/d(ej0d)g¡d7d8 Z9dEd9d:Z:dFd;d<Z;d=d> Z<d?d@ Z=dAdB Z>dCdD Z?dS )Gé N)Úcatch_warningsÚ simplefilter)Ú Timestamp)Úis_platform_windows)Ú DataFrameÚIndexÚSeriesÚ_testingÚ bdate_rangeÚread_hdf)Ú _maybe_removeÚensure_clean_store)Ú_test_decoratorsZbloscc s¾ t ¡ ¬ fdd} t ¡ }t || d|¡ t ¡ }t || d|¡ t ¡ }t || d|¡ ttdtdd}|j dd d
|
t
|
dd gd }t ||j d k |¡ W 5 Q R X d S )Nc s |j | f| t | S )N)Úto_hdfr )ÚkeyÚobjÚkwargs©Úpath© ú_d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/io/pytables/test_round_trip.pyÚ roundtrip& s z'test_conv_read_write.<locals>.roundtripZseriesZ string_seriesÚframeé ©ÚAÚBÚtableT©Úappendzindex>2)Úwhereé ) ÚtmZ ensure_cleanÚmakeTimeSeriesÚassert_series_equalÚmakeStringSeriesÚ makeDataFrameÚassert_frame_equalr Úranger r Úindex)r ÚoÚdfÚresultr r r Útest_conv_read_write# s
|
r- c C sd t dtjddditjdddd}t| .}|jd|dgd | d¡}t ||¡ W 5 Q R X d S )NÚaéd é
|
)Úsize©r) r+ )Z data_columns)r r" Z rands_arrayr r Úselectr' )Ú
|
setup_pathr+ Ústorer, r r r Útest_long_strings: s ÿ
|
|
r6 c C s¨ | | }t ¡ }|jd d
j|dddd |jdd
j|dddd t t|d|¡ |jd d
j|dddd |jdd
j|dddd t t|d|¡ d S )Nr0 r+ Tr ©r ÚformatF©r" r& Úilocr r' r ©Útmp_pathr4 r r+ r r r Útest_apiG s r= c C s¤ | | }t ¡ }|jd d
j|ddd |jdd
j|dddd t t|d|¡ |jd d
j|dddd |jdd
j|ddd t t|d|¡ d S )Nr0 r+ Tr r r7 Fr9 r; r r r Útest_api_appendW s r> c C s
|
| | }t ¡ }|j|dddd t t|d|¡ |j|dddd t t|d|¡ |j|ddd t t|d|¡ | |d¡ t t|d|¡ t|^}t ¡ }t|d |jd|jd d
dd d |jd|jdd
dd d t | d¡|¡ t|d |jd|jd d
dd d |jd|jdd
dd d t | d¡|¡ t|d |jd|jd d
dd d |jd|jdd
dd d t | d¡|¡ t|d |jd|jd d
dd d |jd|jdd
dd d t | d¡|¡ W 5 Q R X d S )
|
Nr+ FÚfixedr7 Úfr r0 Tr )
|
r" r& r r' r r r r r: r3 )r< r4 r r+ r5 r r r Ú
|
test_api_2e s8
|
|
|
|
rA c C s | | }t ¡ }d}tjt|d |j|dddd W 5 Q R X tjt|d |j|dddd W 5 Q R X d}tjt|d |j|ddd d W 5 Q R X tjt|d |j|dd
|
d d W 5 Q R X d }d | d }tjt|d t|d W 5 Q R X d S )NzCan only append to Tables©Úmatchr+ Tr@ r7 r? z)invalid HDFStore format specified \[foo\]ÚfooFÚ zFile z does not exist) r" r& ÚpytestÚraisesÚ
|
ValueErrorr Ú TypeErrorÚFileNotFoundErrorr )r< r4 r r+ Úmsgr r r Útest_api_invalid s rL c
|
C s t | r}t ¡ |d< | d¡}|d }t ||¡ | d¡}|d }t ||¡ tjtdd | d¡ W 5 Q R X W 5 Q R X d S )Nr. z/az 'No object named b in the file'rB Úb)r r" r# Úgetr$ rF rG ÚKeyError)r4 r5 ÚleftÚrightr r r Útest_get¬ s
|
|
|
rR c C s$ t tj dd¡}t|tj| d S )Né2 r/ )r ÚnpÚrandomÚrandnÚ_check_roundtripr" r' )r4 r+ r r r Útest_put_integer» s rX c C sÒ t | ¾}tddddgidd}| d|¡ t |j|d j¡ tddddgidd}| d |¡ t |j|d j¡ t d
|
¡}tj t
|
|d | d |¡ W 5 Q R X tt j dgdgdggd dd gd}| d|¡ t |j|d j¡ |jd dksþt tdd dD }d|d< d|d< |d d¡|d< |d dk|d< td|d< td |d < | d |¡ | d ¡j ¡ }d d |jD |_tdddddddddd! d"d#}| ¡ }| ¡ }t ||¡ W 5 Q R X d S )$Nr. é r! é Zf8©ÚdtypeZdf_f8Úi8Zdf_i8zñinvalid combination of [values_axes] on appending data [name->values_block_0,cname->values_block_0,dtype->float64,kind->float,shape->(1, 3)] vs current table [name->values_block_0,cname->values_block_0,dtype->int64,kind->integer,shape->None]rB Zf4r ©ÚcolumnsZdf_f4r Úfloat32c S s" i | ]}|t tj d ¡|dqS )r r[ )r rT rU Úrandint)Ú.0Úcr r r Ú
|
<dictcomp>à s ÿ z6test_table_values_dtypes_roundtrip.<locals>.<dictcomp>)r` Úfloat64Úint32Úint64Úint16Úint8rD Ústringç ð?Zfloat322ÚboolZ20130101Ztime1Z20130102Ztime2Zdf_mixed_dtypes1c S s g | ] }t |qS r )Ústr)rb Úir r r Ú
|
<listcomp>î s z6test_table_values_dtypes_roundtrip.<locals>.<listcomp>) r` re rf rl rh ri rg Úobjectzdatetime64[ns]Úcount)Úname)r r r r" r$ ZdtypesÚreÚescaperF rG rH rT ÚarrayÚAssertionErrorZastyper r3 Z value_countsr) r Z
|
sort_index)r4 r5 Údf1Údf2rK r, Úexpectedr r r Ú"test_table_values_dtypes_roundtripÁ s\ ÿ$ þÿ ÷ ôrz z(ignore::pandas.errors.PerformanceWarningc C s t ¡ }t|t j| d t ¡ }t|t j| d t|jt|jtd}t|t j| d t|j tt
|
j |jtdtd}t|t j| dd d S )Nr r[ F)r Úcheck_index_type) r" r% rW r$ r# r r) r rp ÚvaluesrT Zasarray)r4 ÚsÚtsZts2Zts3r r r Ú test_series s ÿr c C s4 t j d¡}tt j d¡|d}t|tj| d d S )Nr0 r2 r )rT rU rV r rW r" r$ )r4 r) r} r r r Útest_float_index s r c C sl t d¡}dddg}t j d¡ d¡}t|||d}tdd $ td
|
tj j
|
t |t j | d W 5 Q R X d S ) Nr0 )g rk )ç @ç @)g @g @é )rZ r0 )r) r_ T©ÚrecordÚignorer )rT ZarangerU rV Zreshaper r r ÚpdÚerrorsÚPerformanceWarningrW r" r' )r4 ÚcolÚidxÚdataZDFr r r Útest_tuple_index s
|
|
r c C s t dd tj d¡}dd }W 5 Q R X t dd" t|ddg}t||| d W 5 Q R X t dd( t|tj ¡ dg}t||| d W 5 Q R X t dd" t|ddg}t||| d W 5 Q R X t dd( t|tj ¡ d g}t||| d W 5 Q R X t dd t|ddg}t||| d t|tj ¡ dg}t||| d t|ddg}t||| d t|tj ¡ d g}t||| d t|d
|
d g}t||| d t|d d g}t||| d t|d dg}t||| d t|t dd d ¡t dd d¡g}t||| d W 5 Q R X d S )NTr r! c S s t j| |ddS )NT)r{ )r" r$ )ÚlhsÚrhsr r r Ú<lambda>+ ó z"test_index_types.<locals>.<lambda>r Úyr r. g®Gáz®ó?rM rY g{®Gázø?r iÜ ) r rT rU rV r rW ÚdatetimeÚtodayÚdate)r4 r| ÚfuncZserr r r Útest_index_types& sF ÿr c C sl t dd}ttj t|¡|d}zt|tj| d W n0 t k
|
rf t
|
r`|j t j d¡¡ Y nX d S )Nz1/1/1940z1/1/1960r2 r z'known failure on some windows platforms)r
|
r rT rU rV ÚlenrW r" r$ Ú OverflowErrorr ÚnodeZ
|
add_markerrF ÚmarkZxfail)r4 ÚrequestZdrr~ r r r Útest_timeseries_preepochY s
|
|
ÿr Ú compressionFT)Zmarksc C sÀ t ¡ }tj|jd< tj|jd< t|t j|| d t|t j|| d t ¡ }t|t j|| d t |8}tj
|
t |¡|d< ||d< |d }|j ¡ stW 5 Q R X t|d d
t j|d d S )N)r r )r rZ ©r r rD r+ r r )r" r& rT Únanr: Ú_check_roundtrip_tabler' rW ZmakeTimeDataFramer rU rV r Z_mgrZis_consolidatedrv )r r4 r+ Ztdfr5 Úreconsr r r Ú
|
test_framef s8 ÿ ÿ ÿ
|
r£ c C s t td}t dtd}t }tdddgd}tdd d
|
gd }t|tj| d t|tj| d t|tj| d t|tj| d t|tj| d d S ) Nr[ Zmyseries)rr r\ r. rM rc r2 ÚdÚer@ r^ r )r rp r rW r" r$ r' )r4 Ús0Ús1Zdf0rw rx r r r Útest_empty_series_frame s
|
r¨ r\ zm8[ns]zM8[ns]c C s t | d}t|tj|d d S )Nr[ r )r rW r" r$ )r\ r4 r} r r r Útest_empty_series s
|
r© c C sB dd t ddD }ttj t|d¡|d}t|tj| d d S )Nc S s g | ] }| ¡ qS r )r )rb Úxr r r ro s z,test_can_serialize_dates.<locals>.<listcomp>z1/1/2000z 1/30/2000é r2 r ) r
|
r rT rU rV r rW r" r' )r4 Úrngr r r r Útest_can_serialize_dates s r c C sn |}t |tj| d t |jtj| d t |d tj| d t| "}||d< |d }t ||¡ W 5 Q R X d S )Nr r r )rW r" r' ÚTr$ r )r4 Z multiindex_dataframe_random_datar r5 r¢ r r r Útest_store_hierarchical¢ s
|
r¯ c C s¾ dd }| }| }t |tj|d t |tj|d t|6}||d< t |d |¡ ||d< t |d |¡ W 5 Q R X t |d tj|| d t |d tj|| d t |d tj|| d d S ) Nc S sP t ¡ } d| d< d| d< | d dk| d< | d dk| d < d
|
| d < d | d < | ¡ S )NrD Úobj1ÚbarÚobj2r r Úbool1r Zbool2rY Úint1r! Zint2)r" r& Z _consolidate)r+ r r r Ú _make_one´ s z#test_store_mixed.<locals>._make_oner r r° r r³ r´ )rW r" r' r r$ )r r4 rµ rw rx r5 r r r Útest_store_mixed° s8
|
|
üüür¶ c K sL i }|rt |d< t|df|$}| |d< |d }||| f| W 5 Q R X d S )NÚcomplibÚwr )Ú_default_compressorr )r Ú
|
comparatorr r r Úoptionsr5 Ú retrievedr r r rW ß s rW c C sP i }|rt |d< t|df|(}|jd| dd |d }||| W 5 Q R X d S )Nr· r¸ r r ©r8 )r¹ r Úput)r rº r r r» r5 r¼ r r r r¡ ê s r¡ c C sV ddg}t dd: tdtjj ttj t |¡|}t
|
|t j | d W 5 Q R X d S )Nu Ïu ÏÏTr r r ) r r r r r r rT rU rV r rW r" r$ )r4 Zunicode_valuesr} r r r Útest_unicode_indexö s
|
r¿ c C s¤ d}t d|gi}t| .}|jd|ddd | d¡}t ||¡ W 5 Q R X t d|gddgd }t| .}|jd|ddd | d¡}t ||¡ W 5 Q R X d S )
|
Nu Îr r+ r zutf-8)r8 Úencodingr. rM r )r r r¾ rN r" r' )r4 Úcharr+ r5 r, r r r Útest_unicode_longer_encoded s
|
|
|
|
r c C sP t dddgdddgddd gd
|
}t ¡ }|jd d
|d < t|tj| d d S ) NrY r! rZ rk r r r. rM rc )r. rM rc r¤ r )r r" r# r) rW r' )r4 r+ r~ r r r Útest_store_datetime_mixed s "rà c C sd t ddgddgd}| | }|j|ddd t|d}t ||¡ | |¡sRt| |¡s`td S ) NrY r! rª r )r r r+ r r½ )r r r r" r' Úequalsrv )r< r4 r+ r Úotherr r r Útest_round_trip_equals s
|
rÆ )F)F)@r rs Úwarningsr r ÚnumpyrT rF Zpandas._libs.tslibsr Z pandas.compatr Zpandasr r r r r r" r
|
r Z pandas.tests.io.pytables.commonr r Z pandas.utilr Útdr¹ r Z
|
single_cpuZ
|
pytestmarkr- r6 r= r> rA rL rR rX rz Úfilterwarningsr r r r r Z parametrizeÚparamZskip_if_windowsr£ r¨ rg re rp r© r r¯ r¶ rW r¡ r¿ r rà rÆ r r r r Ú<module> sb * A
|
|
|
|
2 ÿ
|
|
ÿ
|
,
|
|
|
|
