U
|
ý°d¨ ã @ sô d Z ddlZddlZddlZejjdddgddgdd d
|
d gd d dgdddddgddddgdgdddgddd Zdd Zd d Zd d Z d d! Z
|
d"d# Z ej d$d%gd&d'd(g¡d)d* Z ej d$d%gd&d'd(g¡d+d, Z d-d. Zd/d0 ZdS )1ze
|
A verbatim copy (vendored) of the spec tests.
|
Taken from https://github.com/data-apis/dataframe-api
|
é NÚ test_dataZfooÚbarZbazZqux©ÚaÚbç ø?ç @ç @çffffff"@ç %@ç'@é é é é )ÚAÚBZstr_dataZ
|
float_dataZint_data)Úidsc C s t | ¡ }|| }| ¡ }t| |d }|D ]T}| |¡j}|dksLtt|tsZt| |¡ ¡ |kspt| |¡j
|
dks0tq0d S )Nr ) ÚlistÚkeysÚ __dataframe__ÚlenÚget_column_by_nameÚ
|
null_countÚAssertionErrorÚ
|
isinstanceÚintÚsizeÚoffset)r Ú df_from_dictÚcolumnsÚdfÚdfXZ column_sizeÚcolumnr © r$ úed:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/interchange/test_spec_conformance.pyÚtest_only_one_dtype s
|
r& c C sÚ | dddgdddgdddgd d
|
d gd d d gdddgd}| ¡ }ddddddd}| ¡ D ]`\}}| |¡}|jdks|tt|jtst| ¡ dkst|jdksªt|j d |ks\tq\| d¡j d dksÖtd S )Nr r r r é r r r é é
|
é TFr Ú Úc)r r r, ÚdÚeÚfr é é é@ )
|
r Úitemsr r r r r r r Údtype)r r! r" r r# ÚkindÚcolXr$ r$ r% Útest_mixed_dtypes" s& úÿ
|
|
r7 c C sH | ddt jdgi}| ¡ }| d¡}|jdks4tt|jtsDtd S )Nr g ð?g @r )ÚmathÚnanr r r r r r ©r r! r" r6 r$ r$ r% Ú test_na_float? s
|
|
r; c C sH | ddddgi}| ¡ }| d¡}tjtdd |j W 5 Q R X d S )Nr r r r z.*categorical.*)Úmatch)r r ÚpytestZraisesÚ TypeErrorÚdescribe_categoricalr: r$ r$ r% Útest_noncategoricalG s
|
|
r@ c C s^ | ddddddddddg id d
|
}| ¡ d¡}|j}t|d tsHtt|d tsZtd S ) NÚweekdayÚMonÚTueÚWedÚThuÚFriÚSatÚSunT)Zis_categoricalZ
|
is_orderedZ is_dictionary)r r r? r Úboolr )r r! r6 Z categoricalr$ r$ r% Útest_categoricalO s þrJ c C s | dddgdddgdddgd }| ¡ }| ¡ d
|
ks:t| ¡ d
|
ksJt| ¡ dksZtt| ¡ d d d gksttt| d¡ ¡ t| d¡ ¡ kstd S )NTFr r r r
|
r r )ÚxÚyÚzr rK rL rM )r r )rK rM ) r Z num_columnsr Únum_rowsÚ
|
num_chunksr Z column_namesZselect_columnsZselect_columns_by_name)r r! r" r$ r$ r% Útest_dataframe[ s ÿ ÿrP r Ún_chunks)r) r )é r )rR r' c C sX |dt t| i}| ¡ }t | |¡}t||ks:ttdd |D | ksTtd S )NrK c s s | ]}| ¡ V qd S ©N)rN ©Ú.0Úchunkr$ r$ r% Ú <genexpr>p s z%test_df_get_chunks.<locals>.<genexpr>)r Úranger Ú
|
get_chunksr r Úsum©r rQ r r! r" Úchunksr$ r$ r% Útest_df_get_chunksj s
|
r] c C s^ |dt t| i}| ¡ }t | d¡ |¡}t||ks@ttdd |D | ksZtd S )NrK r c s s | ]}| ¡ V qd S rS )r rT r$ r$ r% rW y s z)test_column_get_chunks.<locals>.<genexpr>)r rX r Ú
|
get_columnrY r r rZ r[ r$ r$ r% Útest_column_get_chunkss s
|
r_ c C s | ddgddgd}| ¡ }| ¡ D ]$}| ¡ dks:t| ¡ dks&tq&| d¡jd dksdt| d¡jd dks|td S )Nr r r r r r )r Z get_columnsr r rO r^ r4 r: r$ r$ r% Útest_get_columns| s r` c C sæ dddg}| d|i}| ¡ }| d¡}| ¡ }|d \}}|jdksJt|jdksXt| ¡ \}} |d dkstt|dkrâ|d }
|
tjtj tj
|
tj d|
|
} t |D ]:\} } | |j| |
|
d ¡j}|| ks¦td| d q¦d S )
|
Nr r éÿÿÿÿr Údata)é é é r2 rc zBuffer at index z mismatch)r r^ Z get_buffersÚbufsizer ÚptrZ__dlpack_device__ÚctypesÚc_int8Úc_int16Úc_int32Úc_int64Ú enumerateÚ from_addressÚvalue)r Zarrr! r" r6 ZbufXZdataBufZ dataDtypeZdeviceÚ_ZbitwidthÚctypeÚidxÚtruthÚvalr$ r$ r% Ú test_buffer s,
|
|
üûru )Ú__doc__rh r8 r= ÚmarkZ parametrizer& r7 r; r@ rJ rP r] r_ r` ru r$ r$ r$ r% Ú<module> s. ýù
|
|
|
|
