U
|
ý°dU+ ã @ st d Z ddlZddlZddlmZ ddlmZmZ ddl Z
|
ddl m Z ddl mZ G dd deZG d d
|
d
|
eZdS ) z
|
Tests for 2D compatibility.
|
é N)Úis_matching_na)Ú is_bool_dtypeÚis_integer_dtype)ÚINT_STR_TO_DTYPE)ÚBaseExtensionTestsc @ sä e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dd Z ej dddg¡dd Zej dd d d d d d!g¡d"d# Zej dd d d d d d!g¡d$d% Zej dd d d d d d!g¡d&d' Zd(S ))ÚDim2CompatTestsc C sJ | d¡ dd¡}|j}|d |d ks,t|jj|d d d
ksFtd S )Né éÿÿÿÿr )ÚrepeatÚreshapeÚshapeÚAssertionErrorÚT)ÚselfÚdataÚarr2dr © r úWd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/extension/base/dim2.pyÚtest_transpose s z Dim2CompatTests.test_transposec C sT | d¡ dd¡}t |¡}t |d d
df |d d
df d¡}| ||¡ d S )Nr r r é )r r )r
|
r ÚpdZ DataFrameZassert_frame_equal)r r r ZdfÚexpectedr r r Útest_frame_from_2d_array s
|
(z(Dim2CompatTests.test_frame_from_2d_arrayc C s4 | d¡ dd¡}| dd¡}|j}| ||¡ d S )Nr r r r )r
|
r Zswapaxesr Ú assert_extension_array_equal©r r r Úresultr r r r Ú test_swapaxes% s z Dim2CompatTests.test_swapaxesc C st | d¡ dd¡}|jddd}| d¡ d¡ dd¡}| ||¡ |jddd}| d¡ dd¡}| ||¡ d S )Né r r r ©Úaxisr )r
|
r Údeleter r r r r Útest_delete_2d, s z Dim2CompatTests.test_delete_2dc C sH | dd¡}|jdddgdd}| dddg¡ dd¡}| ||¡ d S )Nr r r r )r Ztaker r r r r Ú test_take_2d9 s z Dim2CompatTests.test_take_2dc C s` t | dd¡}| dt|j ¡dks.tt | dd¡}| dt|j ¡dks\td S )Nr r ú<)Úreprr ÚcountÚtypeÚ__name__r )r r Úresr r r Ú test_repr_2dA s z Dim2CompatTests.test_repr_2dc C s´ | dd¡}|j|jdfks tt|t|ks4t| d¡}|j|jdfksRtt|t|ksftt t¡ | |jdf¡ W 5 Q R X t t¡ | |jd¡ W 5 Q R X d S )Nr r )r r r )r r Úsizer ÚlenÚpytestÚraisesÚ
|
ValueError)r r r r r r Ú test_reshapeI s
|
z Dim2CompatTests.test_reshapec C sÞ | dd¡}|d }| ||¡ t t¡ |d W 5 Q R X t t¡ |d W 5 Q R X |d d
}| ||¡ |d d
d d
f }| ||¡ |d d
df }|dg }| ||¡ |d d
tjf }| ||j¡ d S )Nr r r éþÿÿÿ)r r r, r- Ú
|
IndexErrorÚnpZnewaxisr r r r r Útest_getitem_2dW s
|
z Dim2CompatTests.test_getitem_2dc C s | dd¡}tt|}t||jd ks.t|D ]J}t|t|sHt|j|jksXt|j dksftt||jd ks2tq2d S )Nr r r )
|
r ÚlistÚiterr+ r r Ú
|
isinstancer& ÚdtypeÚndim)r r r ÚobjsÚobjr r r Ú test_iter_2dq s z Dim2CompatTests.test_iter_2dc C sR | dd¡}| ¡ }| ¡ g}t|ts,ttdd |D sBt||ksNtd S )Nr r c s s | ]}t |tV qd S )N)r6 r4 )Ú.0Úxr r r Ú <genexpr> s z1Dim2CompatTests.test_tolist_2d.<locals>.<genexpr>)r Útolistr6 r4 r Úallr r r r Útest_tolist_2d} s
|
z Dim2CompatTests.test_tolist_2dc C sì t | ||g¡ dd¡}| ¡ }|j||gdd}| |gd ¡ dd¡}| ||¡ |j||gdd}|jt|dfkst| |d d
d d
f |¡ | |d d
dd
f |¡ d}tj t
|
|d |j||gdd W 5 Q R X d S ) Nr r r r é r z0axis 2 is out of bounds for array of dimension 2©Úmatch) r& Z_concat_same_typer Úcopyr r r+ r r, r- r. )r r ÚleftÚrightr r Úmsgr r r Útest_concat_2d s z Dim2CompatTests.test_concat_2dÚmethodZbackfillÚpadc C sp | d¡ dd¡}|d ¡ ¡ s&t|d ¡ ¡ r:t|j|d}|j|d d¡ dd¡}| ||¡ d S )Nr r r )rJ )r
|
r Zisnar@ r ÚanyZfillnar )r Z data_missingrJ Zarrr r r r r Útest_fillna_2d_method s z%Dim2CompatTests.test_fillna_2d_methodÚmeanÚmedianÚvarÚstdÚsumÚprodc
|
C sÚ | dd¡}d }d }zt|| }W nb tk
|
r } zD|}zt||d d}W n& tk
|
rt } z| }W 5 d } ~ X Y nX W 5 d }~X Y nX t||d d}|d k s¨|d k rÀt|t|ks¼td S t||sÖ||ksÖtd S )Nr r r )r ÚgetattrÚ Exceptionr& r r )
|
r r rJ r Z err_expectedZ
|
err_resultr Úerrr Úerr2r r r Ú test_reductions_2d_axis_none§ s (z,Dim2CompatTests.test_reductions_2d_axis_nonec C s¨ | dd¡}i }|dkr d|d< z t||f ddi|}W n~ tk
|
rº } z`zt|| W nB tk
|
r } z$t|t|kstW Y ¢
|
W Y ¢ d S d }~X Y n
|
X tdW 5 d }~X Y nX dd }|d
|
krT|} |d krF|jjd krF||j}
|
| |
|
¡} |jjd kr|d krn t| t|ks6tt| |
|
| jksFt| || ¡ nP|dkr¤t |srt
|
|r|| d¡}|dkr| ||¡ n| ||| ¡ d S )Nr r )rQ rP r Zddofr ú'Both reductions should raise or neitherc S s> | j dkr| S | jdkr(tt t¡j S tt tj¡j S d S )Né Úib)ÚitemsizeÚkindr r2 r7 ÚintÚnameZuint)r7 r r r Úget_reduction_result_dtypeÔ s
|
|
|
zLDim2CompatTests.test_reductions_2d_axis0.<locals>.get_reduction_result_dtype)rO rR rS )rR rS ZiubÚb)rN rQ rP ZFloat64rN ) r rT rU r& r r7 r] Zastyper r r ) r r rJ r Úkwargsr rV rW r` r r7 r r r Útest_reductions_2d_axis0À s<
|
|
|
|
|
|
z(Dim2CompatTests.test_reductions_2d_axis0c C sÜ | dd¡}zt||dd}W n~ tk
|
r } z`zt|| W nB tk
|
r } z$t|t|kshtW Y ¢
|
W Y ¢ d S d }~X Y n
|
X tdW 5 d }~X Y nX |jdks®tt|| }|d }t||sØ||ksØtd S )Nr r r rY )r r )r rT rU r& r r r ) r r rJ r r rV rW Zexpected_scalarr( r r r Útest_reductions_2d_axis1ö s z(Dim2CompatTests.test_reductions_2d_axis1N)r' Ú
|
__module__Ú __qualname__r r r r! r" r) r/ r3 r; rA rI r, ÚmarkZ parametrizerM rX rc rd r r r r r s&
|
|
|
|
5r c @ s e Zd Zdd ZdS )ÚNDArrayBacked2DTestsc C sð | d¡ dd¡}|jjd s"t| ¡ }|jjd s:t|d d d
d d d
f ¡ }|jjd sft| d¡}|jjd rt|jjd st| d¡}|jjd sªt|j d¡}|jjd rÆt|jjd sÖtd}tjt |d | d ¡ W 5 Q R X |d d d
}|jjd r t|jjd r0t| ¡ jjd sFt| ¡ jjd r\t| d
|
¡jjd stt| d
|
¡jjd rt| d¡jjd r¤t| d¡jjd s¼t| d¡jjd sÔt| d¡jjd rìtd S ) Nr r Z C_CONTIGUOUSÚFZ F_CONTIGUOUSÚKz6order must be one of 'C', 'F', 'A', or 'K' \(got 'Q'\)rC ÚQÚC)
|
r
|
r Z_ndarrayÚflagsr rE r r, r- r. )r r r r( rH Zarr_ncr r r Útest_copy_order s8
|
|
z$NDArrayBacked2DTests.test_copy_orderN)r' re rf rn r r r r rh s rh )Ú__doc__Únumpyr2 r, Zpandas._libs.missingr Zpandas.core.dtypes.commonr r Zpandasr Zpandas.core.arrays.integerr Z pandas.tests.extension.base.baser r rh r r r r Ú<module> s z
|
