U
|
ý°dÁY ã @ s d dl Z d dlZd dlZd dlZd dlmZ d dlmZ d dl m
|
Z
|
d dl Z d dl mZ d dlmZ d dlmZ G dd deZdS ) é N)ÚDtype)Ú is_bool_dtype)Úna_value_for_dtype)Únargsort)ÚBaseExtensionTestsc @ sÖ e Zd ZU dZdd Zej dddg¡dd Zd d
|
Z d d Z
|
d d Z dd Z dd Z dd Zdd Zdd Zej dddg¡d d Zej dddg¡d d Zej d d!d"d#d$d%dejfd&dejfd'd(g¡d)d* Zd+d, Zej d-d.ejd/d0d1ge d2¡d3fd4ejd1d/d0ge d2¡d3fg¡d5d6 Zej d7ddg¡d8d9 Zej d7ddg¡d:d; Zej d7ddg¡d<d= Z ej d>e j d?d@ g¡ej ddAd@ e j g¡dBdC Z dDdE Z!dFdG Z"dHdI Z#dJdK Z$dLdM Z%dNdO Z&e e'¡Z(e)e*dP< dQdR Z+dSdT Z,dUdV Z-ej dWddg¡ej dXdYd/dZd[d\d\gfd0d0d1d/dZd[gfd/d\d\d0d1d/gfg¡d]d^ Z.d_d` Z/ej dad1dYg¡dbdc Z0ej dXddd\d\ggd\d1d\ggd0d0d1ggd1d\d0ggd[d\d\ggg¡dedf Z1ej daddd\d0d1d[g¡dgdh Z2didj Z3dkdl Z4dmdn Z5dodp Z6dqdr Z7dsdt Z8ej dud0d1d/d1d/dZgg¡dvdw Z9ej dxd/dyd1ie:dyfd\i e:dzfd1d/gi e:d{fd/d|d}ie;d~fg¡dd Z<dd Z=dd Z>d
d Z?dd Z@ej d>e je j e jAg¡dd ZBdS )ÚBaseMethodsTestsz%Various Series and DataFrame methods.c C sF t |ds t dt| ¡ t |j¡}|jd }|jdksBt d S )NÚ value_countsz$value_counts is not implemented for ÚdropnaT)
|
ÚhasattrÚpytestÚskipÚtypeÚinspectÚ signaturer Ú
|
parametersÚdefaultÚAssertionError)ÚselfÚdataÚsigÚkwarg© r úZd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/extension/base/methods.pyÚ test_value_counts_default_dropna s
|
|
|
z1BaseMethodsTests.test_value_counts_default_dropnar TFc C s` |d d
}|r || ¡ }n|}t |¡j|d ¡ }t |¡j|d ¡ }| ||¡ d S )Né
|
)r )ÚisnaÚpdÚSeriesr Ú
|
sort_indexÚassert_series_equal)r Zall_datar ÚotherÚresultÚexpectedr r r Útest_value_counts s z"BaseMethodsTests.test_value_countsc C sÆ |d d
¡ }t || ¡ ¡}tj||jd}|jdd ¡ }t |tj
|
svtjdt | gt | |j dd}n&tjd|j dd}dt | ||d k< t |jtjkr¶| d
|
¡}| ||¡ d S ) Nr ©ÚdtypeT)Ú normalizeé Z
|
proportion)ÚindexÚnameg r ZFloat64)ÚuniqueÚnpÚarrayr r r r% r r Ú
|
isinstanceZ CategoricalÚlenr( r ZNAZastyper )r r ÚvaluesÚserr! r" r r r Ú test_value_counts_with_normalize* s ÿ
|
z1BaseMethodsTests.test_value_counts_with_normalizec C s8 t d|i¡}|jdd}t ddg¡}| ||¡ d S )NÚAÚcolumns©Úaxisr r' )r Ú DataFrameÚcountr r )r Ú data_missingÚdfr! r" r r r Ú
|
test_count? s zBaseMethodsTests.test_countc C s& t |¡}| ¡ }d}||ks"td S )Nr' )r r r7 r )r r8 r0 r! r" r r r Útest_series_countE s
|
z"BaseMethodsTests.test_series_countc C s$ t |¡ t¡}t|t js td S ©N)r r ÚapplyÚidr- r ©r r r! r r r Útest_apply_simple_seriesL s z)BaseMethodsTests.test_apply_simple_seriesc C s: t |¡ ¡ }t tjdddgtjd¡}| ||¡ d S ©Né r r' r$ ©r r Úargsortr+ r, Úintpr )r Údata_for_sortingr! r" r r r Ú test_argsortP s z BaseMethodsTests.test_argsortc C s. | ¡ }tjdddgtjd}t ||¡ d S rA )rD r+ r, rE ÚtmÚassert_numpy_array_equal©r Údata_missing_for_sortingr! r" r r r Útest_argsort_missing_arrayV s z+BaseMethodsTests.test_argsort_missing_arrayc C s: t |¡ ¡ }t tjdddgtjd¡}| ||¡ d S )Nr' éÿÿÿÿr r$ rC rJ r r r Útest_argsort_missing\ s z%BaseMethodsTests.test_argsort_missingc C sz | ¡ dkst| ¡ dks t| ddddddg¡}| ¡ dksFt| ¡ dksVt| ¡ dksft| ¡ dksvtd S )Nr' rB r é )Úargmaxr ÚargminÚtake)r rF rK Úna_valuer r r r Útest_argmin_argmaxa s z#BaseMethodsTests.test_argmin_argmaxÚmethodrP rQ c C s6 d}t jt|d t|d d
| W 5 Q R X d S )Núattempt to get©Úmatchr )r ÚraisesÚ
|
ValueErrorÚgetattr)r rU r Úerr_msgr r r Ú test_argmin_argmax_empty_arrayr s z/BaseMethodsTests.test_argmin_argmax_empty_arrayc C sF d}t |j||g|jd}tjt|d t|| W 5 Q R X d S )NrV r$ rW )r Ú_from_sequencer% r rY rZ r[ )r rU r rS r\ Údata_nar r r Útest_argmin_argmax_all_nay s z*BaseMethodsTests.test_argmin_argmax_all_nazop_name, skipna, expected)ÚidxmaxTr )ÚidxminTrB )rP Tr )rQ TrB ra rb )rP FrM )rQ FrM c C s* t |¡}t|||d}t ||¡ d S )N©Úskipna)r r r[ rH Zassert_almost_equal)r rK Zop_namerd r" r0 r! r r r Útest_argreduce_series s
|
z&BaseMethodsTests.test_argreduce_seriesc C sT |}t jtdd |jdd W 5 Q R X t jtdd |jdd W 5 Q R X d S )NÚ rW Frc )r rY ÚNotImplementedErrorrQ rP )r rK r r r r Ú+test_argmax_argmin_no_skipna_notimplemented s
|
z<BaseMethodsTests.test_argmax_argmin_no_skipna_notimplementedzna_position, expectedÚlastrB r r' rE r$ Úfirstc C s t ||d}t ||¡ d S )N)Ú na_position)r rH rI )r rK rk r" r! r r r Ú test_nargsort s z BaseMethodsTests.test_nargsortÚ ascendingc C sj t |¡}|j||d}|jdddg }|sZ| ¡ dkrJ|jdddg }n|jdddg }| ||¡ d S ©N)rm ÚkeyrB r r' )r r Ú sort_valuesÚilocZnuniquer )r rF rm Ú sort_by_keyr0 r! r" r r r Útest_sort_values¬ s
|
z!BaseMethodsTests.test_sort_valuesc C sN t |¡}|j||d}|r.|jdddg }n|jdddg }| ||¡ d S rn )r r rp rq r )r rK rm rr r0 r! r" r r r Útest_sort_values_missingº s
|
z)BaseMethodsTests.test_sort_values_missingc C s` t dddg|d¡}| ddg¡}t jdddg| dddg¡ddddgd}| ||¡ d S )Nr' rB ©r2 ÚBr2 rv r ©r( )r r6 rp rR Úassert_frame_equal)r rF rm r9 r! r" r r r Útest_sort_values_frameÆ s ÿz'BaseMethodsTests.test_sort_values_frameÚboxc C s | S r< r ©Úxr r r Ú<lambda>Ï ó zBaseMethodsTests.<lambda>c C s | ¡ S r< )r* r{ r r r r} Ð r~ c C s\ || |d |d g¡}||}t|dks2tt|t|sDt|d |d ksXtd S ©Nr r' )r^ r. r r- r )r r rz rU Z
|
duplicatedr! r r r Ú test_uniqueÏ s
|
z BaseMethodsTests.test_uniquec C s^ t j|dd\}}tjddddddddgtjd}| ddd g¡}t ||¡ | ||¡ d S )
|
NT©Zuse_na_sentinelr rM r' rB r$ é é ) r Ú factorizer+ r, rE rR rH rI Ú assert_extension_array_equal)r Údata_for_groupingÚcodesÚuniquesÚexpected_codesÚexpected_uniquesr r r Útest_factorizeÚ s
|
z BaseMethodsTests.test_factorizec C sh t j|dd\}}|jdd\}}t ||¡ | ||¡ t|tt |¡ksTt|j|jksdtd S )NTr ) r r rH rI r
r. r* r r% )r r Zcodes_1Z uniques_1Zcodes_2Z uniques_2r r r Útest_factorize_equivalenceâ s z+BaseMethodsTests.test_factorize_equivalencec C s^ t |d d
¡\}}tjg tjd}t|jg |d d
jd}t ||¡ |
|
||¡ d S )Nr r$ ) r r r+ r, rE r r^ r% rH rI r
)r r r r r r r r r Útest_factorize_emptyë s
|
z%BaseMethodsTests.test_factorize_emptyc C sR | ddg¡}t d|i¡}| ¡ }|jd }| |¡}||jd< | ||¡ d S )Nr' r2 )r r )rR r r6 Úcopyrq Úfillnarx )r r8 Úarrr9 Zdf_origÚ
|
filled_valr! r r r Útest_fillna_copy_frameó s
|
|
|
z'BaseMethodsTests.test_fillna_copy_framec C sP | ddg¡}tj|dd}| ¡ }|d }| |¡}||jd< | ||¡ d S )Nr' F)r r )rR r r r r rq r )r r8 r r0 Zser_origr r! r r r Útest_fillna_copy_seriesÿ s
|
|
z(BaseMethodsTests.test_fillna_copy_seriesc C s4 d}t jt|d | | dg¡¡ W 5 Q R X d S )Nz!Length of 'value' does not match.rW r' )r rY rZ r rR )r r8 Úmsgr r r Útest_fillna_length_mismatch
|
s z,BaseMethodsTests.test_fillna_length_mismatchÚ_combine_le_expected_dtypec s° |d\}}t |¡}t |¡}| |dd ¡}t jdd tt|t|D | jd}| ||¡ |jd | dd ¡}t j fd dt|D | jd}| ||¡ d S )
|
NrB c S s | |kS r< r ©Úx1Zx2r r r r} r~ z2BaseMethodsTests.test_combine_le.<locals>.<lambda>c S s g | ]\}}||kqS r r ©Ú.0ÚaÚbr r r Ú
|
<listcomp> s z4BaseMethodsTests.test_combine_le.<locals>.<listcomp>r$ r c S s | |kS r< r r r r r r} r~ c s g | ] }| kqS r r ©r r ©Úvalr r r " s )r r ÚcombineÚzipÚlistr r rq ©r Z data_repeatedZ
|
orig_data1Z
|
orig_data2Ús1Ús2r! r" r r r Útest_combine_le s
|
|
þ
|
þz BaseMethodsTests.test_combine_lec
|
sÈ |d\}}t |¡}t |¡}| |dd ¡}tjdd. t | dd tt|t|D ¡¡}W 5 Q R X | ||¡ |j d | d d ¡}t | fd
|
dt|D ¡¡}| ||¡ d S ) NrB c S s | | S r< r r r r r r} , r~ z3BaseMethodsTests.test_combine_add.<locals>.<lambda>Úignore)Zoverc S s g | ]\}}|| qS r r r r r r r 0 s z5BaseMethodsTests.test_combine_add.<locals>.<listcomp>r c S s | | S r< r r r r r r} 6 r~ c s g | ] }| qS r r r r r r r 8 s )
|
r r r¡ r+ Zerrstater^ r¢ r£ r rq r¤ r r r Útest_combine_add' s"
|
|
ÿÿ
|
ÿz!BaseMethodsTests.test_combine_addc C sZ t |d d
¡}t j|dd
dddgd}| |¡}t |d d
¡}| ||¡ d S )NrO rB é r rw )r r Z combine_firstr )r r r r r! r" r r r Útest_combine_first< s
|
|
z#BaseMethodsTests.test_combine_firstÚframezperiods, indiceséþÿÿÿrO r rM c C s |d d
}t j|dd}t j|j|dddd}|r|jddjdd |¡}t j|t jdgd dd |¡gdd }| j}n| |¡}| j}||| d S )
|
Nrª r2 ©r) T©Z
|
allow_fillr' )rv rv r4 ) r r rR Úto_frameZassignÚshiftÚconcatrx r ) r r r¬ ÚperiodsÚindicesÚsubsetr" r! Úcomparer r r Útest_container_shiftD s ÿ
|
z%BaseMethodsTests.test_container_shiftc C sB | d¡}|d |d ks t|d |d< |d |d ks>td S r ©r± r r? r r r Útest_shift_0_periods[ s
|
z%BaseMethodsTests.test_shift_0_periodsr³ c C sþ |d d
}t |jr tj}ntj}z||| W n( tk
|
rZ t t| d¡ Y nX t
|
|¡}| |¡}t
|
||| |¡¡}| ||¡ t |dgd d¡}| |¡}|dkrÌtjddddg}ndddtjtjg}t ||d¡}| ||¡ d S )Nrª z does not support diffg ð?ru r' r )r r% ÚoperatorÚxorÚsubÚ Exceptionr r r r r Zdiffr± r r6 r+ Únanrx ) r r r³ ÚopÚsr! r" r9 r r r r Ú test_diffb s&
|
|
|
|
zBaseMethodsTests.test_difféüÿÿÿc C s4 |d d
}| |¡}|j|dd}| ||¡ d S )NrB Tr¯ ©r± rR r
)r r r³ r´ rµ r! r" r r r Útest_shift_non_empty_array| s
|
z+BaseMethodsTests.test_shift_non_empty_arrayc C s* |d d
}| |¡}|}| ||¡ d S )Nr )r± r
)r r r³ Úemptyr! r" r r r Útest_shift_empty_array s
|
z'BaseMethodsTests.test_shift_empty_arrayc C s8 | d¡}||k st|d d
d¡}||k s4td S )Nr rB r¸ r? r r r Útest_shift_zero_copies s
|
z'BaseMethodsTests.test_shift_zero_copiesc C sp |d d
}|d }|j d|d}| ddddg¡}| ||¡ |j d|d}| ddddg¡}| ||¡ d S )Nr r r' )Ú
|
fill_valuerB r rO rà )r r r rÈ r! r" r r r Útest_shift_fill_value s z&BaseMethodsTests.test_shift_fill_valuec C s& t jtdd t| W 5 Q R X d S )Nzunhashable typerW )r rY Ú TypeErrorÚhash)r r r r r Útest_not_hashable¢ s z"BaseMethodsTests.test_not_hashablec C s> t |¡}|r| ¡ }t j |¡}t j |¡}| ||¡ d S r< )r r r° ÚutilZhash_pandas_objectÚ assert_equal)r r Úas_framer r r r r Ú test_hash_pandas_object_works§ s
|
z.BaseMethodsTests.test_hash_pandas_object_worksc
|
C sþ |\}}}| dddg¡}|r(t |¡}| |¡dks:t|j|dddksPt| |¡dksbt|j|dddksxt| |¡dkst|j|dddks t| | ddg¡¡}tjddgtjd}t ||¡ t dddg¡} |j|| ddksútd S ) NrB r r' Úright)ZsiderO r$ )Úsorter)
|
rR r r Z searchsortedr r+ r, rE rH rI )
|
r rF Ú as_seriesr Úcr r r! r" rÒ r r r Útest_searchsorted° s
|
|
z"BaseMethodsTests.test_searchsortedc C s® |d |d kst t|}|d d
\}}t |j||||g|jd¡}| ¡ }t ddddg¡} |r|j dd}|
|
d d¡} | | ¡}
|
t |j||||g|jd¡} |r¸| j dd} | |
|
| ¡ |j | dd
|
| || ¡ | ¡ }|rø|j dd}t ddddg¡} |j||||g|jd} |rDt d| i¡} t d| i¡} | | | ¡}
|
t |j||||g|jd¡} |r| j dd} | |
|
| ¡ |j | | dd
|
| || ¡ d S ) Nr r' rB r$ TFr r® rM )Zinplace)r r r r r^ r% r r+ r, r° ZreshapeÚwhererÎ Úmaskr6 ) r r rS rÏ Úclsr r Úorigr0 Zcondr! r" r r r r Útest_where_seriesÈ sB
|
ÿ z"BaseMethodsTests.test_where_seriesÚrepeatsc C s® t |j|d d
|jd}|r*t |¡}|r:t ||¡n| |¡}t|trX|gd n|}dd t ||D }t |j||jd}|rtj||j
|
|¡d}| ||¡ d S )NrO r$ c S s" g | ]\}}t |D ]}|qqS r )Úrange)r r| ÚnÚ_r r r r ü s
|
z0BaseMethodsTests.test_repeat.<locals>.<listcomp>rw ) r r^ r% r r r+ Úrepeatr- Úintr¢ r( rÎ )r r rÛ rÓ Ú use_numpyr r! r" r r r Ú test_repeató s
|
z BaseMethodsTests.test_repeatzrepeats, kwargs, error, msgr5 ÚnegativeÚshapeZfooÚbarz'foo'c C sB t j||d* |r&tj||f| n|j|f| W 5 Q R X d S )NrW )r rY r+ rß )r r rÛ ÚkwargsÚerrorr rá r r r Útest_repeat_raises s
|
z#BaseMethodsTests.test_repeat_raisesc C sd | d¡}|dd
}| ||¡ | ddg¡}| |dg |dg |dd
g¡}| ||¡ d S )Nr r' rO rB r )Údeleter
Z_concat_same_type)r r r! r" r r r Ú test_delete s
|
$z BaseMethodsTests.test_deletec C sº |dd
d|d ¡}| ||¡ |dd
t|dd
|d ¡}| ||¡ |d d
d|d ¡}t t|¡}|dd
|dd
< t|d |d< | |¡}| ||¡ d S )Nr' r rM r rª )Úinsertr
r. r+ ZarangerR )r r r! Ztakerr" r r r Ú test_insert s &
|
z BaseMethodsTests.test_insertc C s |}t ttf¡ | d|¡ W 5 Q R X t ttf¡ | d|¡ W 5 Q R X t ttf¡ | t|d |¡ W 5 Q R X d S )Nr r r' )r rY rÊ rZ rë r. )r r Zinvalid_scalarÚitemr r r Útest_insert_invalid- s z$BaseMethodsTests.test_insert_invalidc C s t |}t t¡ | |d |d ¡ W 5 Q R X t t¡ | | d |d ¡ W 5 Q R X t t¡ | d|d ¡ W 5 Q R X d S )Nr' r g ø?)r. r rY Ú
|
IndexErrorrë rÊ )r r Zubr r r Útest_insert_invalid_loc9 s z(BaseMethodsTests.test_insert_invalid_locc C s8 t |j|d gt| |jd}t |j|gt| |jd}tj||dd}tj||dd}tj||dd}| |¡dkst| | ¡ ¡dkst| |¡dksªt| |¡dks¼t|d d
|d d
¡dksÞt|d d
|d d
¡dkst| d ¡dkst|dg |d ¡dks4td S )Nr r$ F)Z transposeTrB rO ) r r^ r. r% rH Z box_expectedÚequalsr r )r r rS rÓ rz Zdata2r_ r r r Ú test_equalsF s " "$z BaseMethodsTests.test_equalsN)CÚ__name__Ú
|
__module__Ú __qualname__Ú__doc__r r ÚmarkZ parametrizer# r1 r: r; r@ rG rL rN rT r] r` r+ r¾ re rh r, r% rl rs rt ry r r r* r r r r r r r Úboolr r Ú__annotations__r§ r© r« r· r¹ rÁ rÄ rÆ rÇ rÉ rÌ rÐ rÕ rÚ râ rZ rÊ rè rê rì rî rð r6 rò r r r r r s¾
|
|
|
|
|
|
øþ
|
|
þþ
|
|
|
|
2þ
|
4þ
|
|
+
|
|
üþ
|
r )r rº Únumpyr+ r Zpandas._typingr Zpandas.core.dtypes.commonr Zpandas.core.dtypes.missingr Zpandasr Zpandas._testingZ_testingrH Zpandas.core.sortingr Z pandas.tests.extension.base.baser r r r r r Ú<module> s
|
