U
|
ý°d# ã
|
@ sÄ d dl Zd dlZd dlmZmZmZmZm Z m
|
Z
|
d dl m Z m Z ejjgZdd Zdd Zdd Zej d
|
d ed d ddgied g ifded d ddgied d gifg¡eeedddZdS )é N)Ú CategoricalÚ DataFrameÚSeriesÚ_testingÚconcatÚread_hdf)Ú _maybe_removeÚensure_clean_storec
|
|
C s t |
|
}t|d ttddddddgddddgdd}|jd|dd | d¡}t ||¡ t|d
|
ttddddddgddddgd d}|jd
|
|dd | d
|
¡}t ||¡ t|d t|d dddddgd}|jd |dd | d ¡}t ||¡ t|d td d dddddg
|
d¡}| d|¡ | d¡}t ||¡ t|d td d t j ddddg
|
d¡}| d|¡ | d¡}t ||¡ t|d | ¡ }ttd
|
d¡|d< | d|¡ | d¡}t ||¡ | ¡ }d|ksôtd|kstd |kstt|d ttddddddgddddgdd}|jd|dd | d¡}t ||¡ t|d |jd |dgd ||j ddg¡ }|jd d gd }t ||¡ ||j ddg¡ }|jd d!gd }t ||¡ ||j dg¡ }|jd d"gd }t ||¡ ||j d#g¡ }|jd d$gd }t ||¡ | d |¡ t||g}||j ddg¡ }|jd d gd }t ||¡ | ¡ }|d j ¡ |d< d%} tjt| d& | d |¡ W 5 Q R X | d'¡}|d k sât| d ¡ tjtd(d& | d'¡ W 5 Q R X W 5 Q R X d S ))NÚsÚaÚbÚcÚdF)Ú
|
categoriesZorderedÚtable)ÚformatZ s_orderedTÚdfé é é é é é )r
|
ÚvalsÚsiÚcategoryZsi2Údf2ZabcdefgÚs2z/df2 z /df2/meta/values_block_0/metaz /df2/meta/values_block_2/metaÚdf3)Ú data_columnszs in ["b","c"]©Úwherez s = ["b","c"]z
|
s in ["d"]Úfz
|
s in ["f"]zEcannot append a categorical with different categories to the existing)Úmatchzdf3/meta/s/metaz-'No object named df3/meta/s/meta in the file')r r r r ÚappendÚselectÚtmZassert_series_equalr Úassert_frame_equalÚastypeÚnpÚnanÚcopyÚlistÚinfoÚAssertionErrorr
|
Úisinr ÚcatZremove_unused_categoriesÚpytestZraisesÚ
|
ValueErrorÚremoveÚKeyError)
|
Ú
|
setup_pathÚstorer
|
Úresultr r r- Úexpectedr Úmsg© r: ú`d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/io/pytables/test_categorical.pyÚtest_categorical s¶
|
|
ýÿ
|
|
|
ýÿ
|
|
|
|
|
|
|
|
|
|
|
ýÿ
|
|
|
|
ÿr< c C sÔ ddg}ddg}ddg}t |||d}|jg d d
f }| | }|j|dd d
|
d t|dd d }t ||¡ |j d¡|_|j d¡|_|jg d d
f }| | }|j|dd d
|
d t|dd d }t ||¡ d S )NZESP_012345_6789ZESP_987654_3210Z
|
APF00006npZ
|
APF0001immg333333@g#@)ÚobsidsÚimgidsÚdatar r T©r r zobsids=Br r ) r ZilocÚto_hdfr r& r' r= r( r> ) Útmp_pathr5 r= r> r? r r8 Úpathr7 r: r: r; Útest_categorical_conversion s rD c C sª t dddtjgtjtjtjtjgddddgtd gd tdd }|j d
|
¡|d< |j d
|
¡|d< |j d
|
¡|d < |}| | }|j|d d dd t |d }t
|
||¡ d S )Nr r r r r r r )Zdtype)r r r r r r r r Tr@ ) r r) r* r Úobjectr r( r rA r r& r' )rB r5 r r8 rC r7 r: r: r; Ú!test_categorical_nan_only_columns s
|
üÿ
|
rF zwhere, df, expectedzcol=="q"Úcolr r r
|
zcol=="a")r! r r8 c C sx |j d¡|_ ddi}t|j ¡ }|j d¡|_ |j j |¡|_ | | }|j|dd|d t||d}t ||¡ d S )Nr rG r r r )r Z min_itemsizer )
|
rG r( ÚsortedÚuniquer0 Zset_categoriesrA r r& r' ) rB r5 r! r r8 Z
|
max_widthsZcategorical_valuesrC r7 r: r: r; Útest_convert_valueà s rJ )Únumpyr) r1 Zpandasr r r r r& r r Z pandas.tests.io.pytables.commonr r ÚmarkZ
|
single_cpuZ
|
pytestmarkr< rD rF Z parametrizeÚstrrJ r: r: r: r; Ú<module> s$ ÿy þþ ÿ
|
