U
|
ý°d© ã @ s d dl m Z d dlZd dlZd dlZd dlmZ d dlm m Z
|
d dl Z d dl mZ d dlmZ d dlmZmZ d dlmZ e jedd d d
|
e jedd d d
|
d Zd \ZZdd Z e dd Z e dd Z e dd Z e dd Z e dd Z!dddddej"giZ#ej$ %dd d g¡d d Z&d d! Z'd"d# Z(ej$ %de e e e e!g¡d$d% Z)d&d' Z*ej$ %dd(d)d*gd+d,d-gd.d/d0d gd+d,d-gd.e +d d d g¡e +d/d0d g¡e +d+d,d-g¡d1g¡d2d3 Z,d4d5 Z-d6d7 Z.d8d9 Z/d:d; Z0e
|
1d<¡d=d> Z2dS )?é )ÚdatetimeN)ÚiNaT)Ú PandasColumn)ÚColumnNullTypeÚ DtypeKind)Úfrom_dataframeZtestdataé T)ÚorderedF)r Ú unordered)éd éÈ c s fddt tD S )Nc s> i | ]6}d t |td t d fddttD qS )Úcolé é c s g | ]
|
} qS © r )Ú.0Ú_©Zmake_oner úYd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/interchange/test_impl.pyÚ
|
<listcomp> s z)_make_data.<locals>.<dictcomp>.<listcomp>)ÚintÚNCOLSÚrangeÚNROWS)r Úir r r Ú
|
<dictcomp> s ÿ z _make_data.<locals>.<dictcomp>)r r r r r r Ú
|
_make_data s
|
þr c C s t dd¡S )Niÿÿÿr ©ÚrandomÚrandintr r r r Ú<lambda>" ó r c C s t dd¡S )Nr r r r r r r r # r! c C s t ddg¡S )NTF)r Úchoicer r r r r $ r! c C s t ¡ S ©N)r r r r r r % r! c C s& t t dd¡t dd¡t dd¡dS )Nil i4 r é é )ÚyearÚmonthÚday)r r r r r r r r ' s
|
|
|
ýúseparator dataz abC|DeF,Hikz 234,3245.67z gSaf,qWer|Grez
|
asd3,4sad|Údata)r T)r
|
Fc
|
C sà t dt| d i¡}| ¡ d¡}|jd tjks8t|j dksFt|j
|
t j dfksZt| ¡ dksjt|j}|d | d kst|d dkstt|d ts¦tt |d jt d d
|
d d d g¡¡ t |t| ¡ ¡ d S )NÚAr éÿÿÿÿr Z
|
is_orderedZ is_dictionaryTÚ
|
categoriesÚaÚdÚeÚsÚt)ÚpdÚ DataFrameÚtest_data_categoricalÚ __dataframe__Úget_column_by_nameÚdtyper Z CATEGORICALÚAssertionErrorÚ
|
null_countÚ describe_nullr Ú USE_SENTINELZ
|
num_chunksZdescribe_categoricalÚ
|
isinstancer ÚtmZassert_series_equalZ_colZSeriesÚassert_frame_equalr )r* Údfr Zdesc_catr r r Útest_categorical_dtype9 s ÿrA c C s t dd¡} ddddddddd g }| d
|
| |¡ ¡ i¡}| ¡ }t|}tj|ddddddd gd }t d
|
|i¡}t
|
||¡ d S ) NÚpyarrowú11.0.0ÚMonÚTueÚWedÚThuÚFriÚSatÚSunÚweekday)r- ) ÚpytestÚ importorskipÚtableÚarrayZdictionary_encoder6 r r3 Ú Categoricalr4 r> r? )ÚpaÚarrrN Ú exchange_dfÚresultrK Úexpectedr r r Útest_categorical_pyarrowM s ÿrV c C sv t dd¡} ddg}| d| |d¡i¡}| ¡ }t|}t dddgi¡}t ||¡ | j
|
| j |¡|¡srt d S )NrB rC rD rE rK Z large_string)rL rM rN rO r6 r r3 r4 r> r? ZTableÚequalsZ interchanger9 )rQ rR rN rS rT rU r r r Útest_large_string_pyarrow\ s rX c s¼ t ¡}| ¡ }| ¡ tks"t| ¡ tks2tt| ¡ t
|
¡ ksNtd}t fdd|D }t | |¡}t | |¡}t ||¡ t|jd ts¤tt|jd ts¸td S )N)r r c 3 s | ]}t ¡ | V qd S r# )ÚlistÚkeys)r Úidx©r* r r Ú <genexpr>y s z!test_dataframe.<locals>.<genexpr>Z _INTERCHANGE_PROTOCOL_BUFFERS)r3 r4 r6 Z num_columnsr r9 Znum_rowsr rY Z column_namesrZ Útupler Zselect_columnsZselect_columns_by_namer> r? r= Úattrs)r* r@ Údf2ÚindicesÚnamesrT rU r r\ r Útest_dataframek s
|
rc c s t t dddddg¡t dddd dg¡t d
|
d d
|
d
|
d
|
g¡d ¡ ¡ } tj d ¡ fdd jD }| ¡ D ]6\}} j j
|
t t ¡|d d }d j ||f< qt ¡ } | d¡j|d ksÌt| d¡j|d ksät| d¡j|d ksütd S )Nr r é é r ç ø?ç @ç @g @TF©ÚxÚyÚzé* c s i | ]}|j d t dqS )r )ÚlowÚhigh)r Úlen)r r ©r@ Úrngr r r s z,test_missing_from_masked.<locals>.<dictcomp>)ÚsizeÚreplacerj rk rl )r3 r4 ÚnprO r6 r Z RandomStateÚcolumnsÚitemsÚindexr" Zarangerp Úlocr7 r: r9 )r` Z dict_nullr Z num_nullsZnull_idxr rq r Útest_missing_from_masked s$ ýÿ ÿrz rf rg rh çffffff"@ç %@ç'@)rj rk r r ri c C s6 t | ¡}| ¡ }|jD ]}| |¡jdkstqd S )Nr )r3 r4 r6 rv r7 r: r9 )r* r@ r` Úcol_namer r r Útest_mixed_data s
|
|
r c C sn t t dd dd dg¡t d dd ddg¡t ddd dd g¡d¡} | ¡ }| jD ]}| |¡jdksPtqPd S ) NTFr r r{ r| r} ri ) r3 r4 ru rO r6 rv r7 r: r9 )r@ r` r~ r r r Útest_mixed_missing± s ýÿ
|
r c C sÔ t d dg } t d| i¡}| ¡ d¡}| ¡ dks:t|jdksHt|jd t j
|
ks\t|j t j dfkspt|dd
}| ¡ d¡}| ¡ dkst|jdks¨t|jd t j
|
ks¼t|j t j dfksÐtd S )Nr) Ú r+ é r r é )Ú string_datar3 r4 r6 r7 rs r9 r: r8 r ÚSTRINGr; r Z USE_BYTEMASK)Z test_str_datar@ r Z df_slicedr r r Ú test_stringÀ s r c C sH t dddddgi¡} | ¡ d¡}tjtdd |j W 5 Q R X d S )Nr+ r. é
|
g ð?r znot supported yet)Úmatch)r3 r4 r6 r7 rL ZraisesÚNotImplementedErrorr8 ©r@ r r r r Útest_nonstring_objectÒ s r c C s t dt d¡t jgi¡} | ¡ d¡}| ¡ dks8t|jdksFt|j d t
|
j ksZt|j t jtfksntt | t| ¡ ¡ d S )Nr+ z
|
2022-01-01r r r )r3 r4 Ú TimestampZNaTr6 r7 rs r9 r: r8 r ZDATETIMEr; r r< r r> r? r r r r r Ú test_datetimeÙ s r z1.23c C sb t dt dddg¡i¡} | ¡ d¡}t | ¡ d d ¡}tjdddgdd}t
|
||¡ d S ) Nr+ r. Úbr* r r Zint8)r8 ) r3 r4 rP r6 r7 ru Z from_dlpackZ get_buffersrO r> Zassert_numpy_array_equal)r@ r rT rU r r r Ú test_categorical_to_numpy_dlpackå s
|
r )3r r Únumpyru rL Zpandas._libs.tslibsr Z pandas.util._test_decoratorsÚutilZ_test_decoratorsÚtdZpandasr3 Zpandas._testingZ_testingr> Z pandas.core.interchange.columnr Z*pandas.core.interchange.dataframe_protocolr r Z&pandas.core.interchange.from_dataframer rP rY r5 r r r Zint_dataZ uint_dataZ bool_dataZ
|
float_dataZ datetime_dataÚNaNr ÚmarkZ parametrizerA rV rX rc rz rO r r r r r Z skip_if_np_ltr r r r r Ú<module> sr þ ÿ ûÿ
|
ÿ
|
ýýþ
|
|
