o
    ýòTiÕe  ã                   @   sv   d dl Z d dlZd dlZd dlZd dlmZ d dlmZ d dl	m
Z
 d dlZd dlmZ d dlmZ G dd„ dƒZdS )é    N)ÚDtype)Úis_bool_dtype)Úna_value_for_dtype)Únargsortc                   @   sê  e Zd ZU dZdd„ Zdd„ Zej dddg¡d	d
„ ƒZ	dd„ Z
dd„ Zdd„ Zdd„ Zej dddg¡dd„ ƒZdd„ Zdd„ Zdd„ Zdd„ Zej d d!d"g¡d#d$„ ƒZej d d!d"g¡d%d&„ ƒZej d'd(d)d*d+d,dejfd-dejfd.d/g¡d0d1„ ƒZd2d3„ Zej d4d5ejg d6¢e d7¡d8fd9ejg d:¢e d7¡d8fg¡d;d<„ ƒZej d=ddg¡d>d?„ ƒZej d=ddg¡d@dA„ ƒZej d=ddg¡dBdC„ ƒZej dDej dEdF„ g¡ej d dGdF„ ej!g¡dHdI„ ƒƒZ"dJdK„ Z#dLdM„ Z$dNdO„ Z%dPdQ„ Z&dRdS„ Z'dTdU„ Z(e e)¡Z*e+e,dV< dWdX„ Z-dYdZ„ Z.d[d\„ Z/ej d]ddg¡ej d^d_g d`¢fdag db¢fdcg dd¢fg¡dedf„ ƒƒZ0dgdh„ Z1ej didjd_g¡dkdl„ ƒZ2ej d^dmdndnggdndjdnggdadadjggdjdndaggdodndnggg¡dpdq„ ƒZ3ej dig dr¢¡dsdt„ ƒZ4dudv„ Z5dwdx„ Z6dydz„ Z7d{d|„ Z8d}d~„ Z9dd€„ Z:dd‚„ Z;ej dƒdadjdcg d„¢g¡d…d†„ ƒZ<ej d‡dcdˆdjie=dˆfdni e=d‰fdjdcgi e=dŠfdcd‹dŒie>dfg¡dŽd„ ƒZ?dd‘„ Z@d’d“„ ZAd”d•„ ZBd–d—„ ZCej dDejej ejDg¡d˜d™„ ƒZEdšd›„ ZFdS )œÚBaseMethodsTestsz%Various Series and DataFrame methods.c                 C   s@   ddl m} |jd|dd}|jtjksJ ‚|j|jksJ ‚d S )Nr   )Ú_default_hash_keyzutf-8F)ÚencodingZhash_keyZ
categorize)Zpandas.core.util.hashingr   Z_hash_pandas_objectÚdtypeÚnpZuint64Úshape)ÚselfÚdatar   Úres© r   úc/var/www/html/evchargy.com/venv/lib/python3.10/site-packages/pandas/tests/extension/base/methods.pyÚtest_hash_pandas_object   s   ÿz(BaseMethodsTests.test_hash_pandas_objectc                 C   sF   t |dƒst dt|ƒ› ¡ t |j¡}|jd }|jdu s!J ‚d S )NÚvalue_countsz$value_counts is not implemented for ÚdropnaT)	ÚhasattrÚpytestÚskipÚtypeÚinspectÚ	signaturer   Ú
parametersÚdefault)r   r   ÚsigÚkwargr   r   r   Ú 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 ¡ }t ||¡ d S )Né
   )r   )ÚisnaÚpdÚSeriesr   Ú
sort_indexÚtmÚassert_series_equal)r   Zall_datar   ÚotherÚresultÚexpectedr   r   r   Útest_value_counts'   s   z"BaseMethodsTests.test_value_countsc                 C   s  |d d…   ¡ }t || ¡   ¡}tj||jd}|jdd ¡ }t	|tj
ƒs;tjdt|ƒ gt|ƒ |jdd}ntjd|jdd}dt|ƒ ||d	k< t|jd
dƒdks^t	|jtjƒrd| d¡}nt|jd
dƒdkrs| d¡}nt|jƒtju r€| d¡}t ||¡ d S )Nr   ©r	   T)Ú	normalizeé   Z
proportion)ÚindexÚnameg        r   ZstorageÚ Zpyarrowzdouble[pyarrow]Zpyarrow_numpyZfloat64ZFloat64)Úuniquer
   Úarrayr    r!   r"   r	   r   r#   Ú
isinstanceZCategoricalÚlenr-   ÚgetattrZ
ArrowDtypeZastyper   ZNAr$   r%   )r   r   ÚvaluesÚserr'   r(   r   r   r   Ú test_value_counts_with_normalize4   s&   ÿÿ
z1BaseMethodsTests.test_value_counts_with_normalizec                 C   s8   t  d|i¡}|jdd}t  ddg¡}t ||¡ d S )NÚAÚcolumns©Úaxisr   r,   )r!   Ú	DataFrameÚcountr"   r$   r%   )r   Údata_missingÚdfr'   r(   r   r   r   Ú
test_countR   s   zBaseMethodsTests.test_countc                 C   s&   t  |¡}| ¡ }d}||ksJ ‚d S )Nr,   )r!   r"   r=   )r   r>   r6   r'   r(   r   r   r   Útest_series_countX   s   
z"BaseMethodsTests.test_series_countc                 C   s$   t  |¡ t¡}t|t jƒsJ ‚d S ©N)r!   r"   ÚapplyÚidr2   ©r   r   r'   r   r   r   Útest_apply_simple_series_   s   z)BaseMethodsTests.test_apply_simple_seriesÚ	na_actionNÚignorec                 C   s*   |j dd„ |d}| ¡ }t ||¡ d S )Nc                 S   ó   | S rB   r   ©Úxr   r   r   Ú<lambda>e   ó    z+BaseMethodsTests.test_map.<locals>.<lambda>)rG   )ÚmapZto_numpyr$   Úassert_numpy_array_equal)r   r>   rG   r'   r(   r   r   r   Útest_mapc   s   zBaseMethodsTests.test_mapc                 C   s8   t  |¡ ¡ }t  tjg d¢tjd¡}t ||¡ d S ©N©é   r   r,   r*   )r!   r"   Úargsortr
   r1   Úintpr$   r%   )r   Údata_for_sortingr'   r(   r   r   r   Útest_argsorti   s   zBaseMethodsTests.test_argsortc                 C   s,   |  ¡ }tjg d¢tjd}t ||¡ d S rQ   )rT   r
   r1   rU   r$   rO   )r   Údata_missing_for_sortingr'   r(   r   r   r   Útest_argsort_missing_arrayo   s   z+BaseMethodsTests.test_argsort_missing_arrayc                 C   sj   d}t jt|d t |¡ ¡ }W d   ƒ n1 sw   Y  t tjg d¢tjd¡}t  	||¡ d S )Nz;The behavior of Series.argsort in the presence of NA values©Úmatch)r,   éÿÿÿÿr   r*   )
r$   Úassert_produces_warningÚFutureWarningr!   r"   rT   r
   r1   rU   r%   )r   rX   Úmsgr'   r(   r   r   r   Útest_argsort_missingu   s   ÿz%BaseMethodsTests.test_argsort_missingc                 C   sŽ   |j j}d}d}|rd}d}| ¡ |ksJ ‚| ¡ dksJ ‚| g d¢¡}| ¡ |ks-J ‚| ¡ dks5J ‚| ¡ dks=J ‚| ¡ dksEJ ‚d S )Nr,   é   r   rS   )rS   r   r   r,   r,   rS   )r	   Ú_is_booleanÚargmaxÚargminÚtake)r   rV   rX   Úna_valueÚis_boolZ
exp_argmaxZexp_argmax_repeatedr   r   r   r   Útest_argmin_argmax|   s   z#BaseMethodsTests.test_argmin_argmaxÚmethodrc   rd   c                 C   sL   d}t jt|d t|d d… |ƒƒ  W d   ƒ d S 1 sw   Y  d S )Núattempt to getrZ   r   )r   ÚraisesÚ
ValueErrorr4   )r   ri   r   Úerr_msgr   r   r   Útest_argmin_argmax_empty_array•   s   "ÿz/BaseMethodsTests.test_argmin_argmax_empty_arrayc                 C   s\   d}t |ƒj||g|jd}tjt|d t||ƒƒ  W d   ƒ d S 1 s'w   Y  d S )Nrj   r*   rZ   )r   Ú_from_sequencer	   r   rk   rl   r4   )r   ri   r   rf   rm   Údata_nar   r   r   Útest_argmin_argmax_all_naœ   s
   "ÿz*BaseMethodsTests.test_argmin_argmax_all_nazop_name, skipna, expected)ÚidxmaxTr   )ÚidxminTrS   )rc   Tr   )rd   TrS   rr   rs   )rc   Fr\   )rd   Fr\   c           	      C   s˜   d }d}|  d¡r|dkrt}|  d¡r t |¡r t}d|› }t |¡}tj||d t||ƒ|d}W d   ƒ n1 s?w   Y  t 	||¡ d S )Nz$The behavior of Series.argmax/argminÚargr\   ÚidxzThe behavior of Series.rZ   ©Úskipna)
Ú
startswithr^   r
   Úisnanr!   r"   r$   r]   r4   Zassert_almost_equal)	r   rX   Zop_namerw   r(   Úwarnr_   r6   r'   r   r   r   Útest_argreduce_series¤   s   

ÿz&BaseMethodsTests.test_argreduce_seriesc                 C   s~   |}t jtdd |jdd W d   ƒ n1 sw   Y  t jtdd |jdd W d   ƒ d S 1 s8w   Y  d S )Nr/   rZ   Frv   )r   rk   ÚNotImplementedErrorrd   rc   )r   rX   r   r   r   r   Ú+test_argmax_argmin_no_skipna_notimplementedÁ   s   ÿ"ÿz<BaseMethodsTests.test_argmax_argmin_no_skipna_notimplementedzna_position, expectedÚlastrR   rU   r*   Úfirst©r,   rS   r   c                 C   s   t ||d}t ||¡ d S )N)Úna_position)r   r$   rO   )r   rX   r   r(   r'   r   r   r   Útest_nargsortË   s   	zBaseMethodsTests.test_nargsortÚ	ascendingc                 C   sd   t  |¡}|j||d}|jg d¢ }|s*| ¡ dkr#|jg d¢ }n|jg d¢ }t ||¡ d S )N©rƒ   ÚkeyrR   rS   )r   r,   rS   )r,   r   rS   )r!   r"   Úsort_valuesÚilocZnuniquer$   r%   )r   rV   rƒ   Úsort_by_keyr6   r'   r(   r   r   r   Útest_sort_values×   s   
z!BaseMethodsTests.test_sort_valuesc                 C   sJ   t  |¡}|j||d}|r|jg d¢ }n|jg d¢ }t ||¡ d S )Nr„   rR   )r   rS   r,   )r!   r"   r†   r‡   r$   r%   )r   rX   rƒ   rˆ   r6   r'   r(   r   r   r   Útest_sort_values_missingå   s   
z)BaseMethodsTests.test_sort_values_missingc                 C   sX   t  g d¢|dœ¡}| ddg¡}t jg d¢| g d¢¡dœg d¢d}t ||¡ d S )N)r,   rS   r,   ©r8   ÚBr8   rŒ   )r,   r,   rS   rR   ©r-   )r!   r<   r†   re   r$   Úassert_frame_equal)r   rV   rƒ   r?   r'   r(   r   r   r   Útest_sort_values_frameñ   s   ÿz'BaseMethodsTests.test_sort_values_frameÚboxc                 C   rI   rB   r   rJ   r   r   r   rL   ú   rM   zBaseMethodsTests.<lambda>c                 C   s   |   ¡ S rB   )r0   rJ   r   r   r   rL   û   ó    c                 C   s\   ||  |d |d g¡ƒ}||ƒ}t|ƒdksJ ‚t|t|ƒƒs"J ‚|d |d ks,J ‚d S ©Nr   r,   )ro   r3   r2   r   )r   r   r   ri   Z
duplicatedr'   r   r   r   Útest_uniqueú   s
   zBaseMethodsTests.test_uniquec                 C   s€   t j|dd\}}|jj}|r!tjg d¢tjd}| ddg¡}ntjg d¢tjd}| g d¢¡}t 	||¡ t 
||¡ d S )	NT©Zuse_na_sentinel)r   r   r\   r\   r,   r,   r   r   r*   r   é   )r   r   r\   r\   r,   r,   r   rS   )r   r•   é   )r!   Ú	factorizer	   rb   r
   r1   rU   re   r$   rO   Úassert_extension_array_equal)r   Údata_for_groupingÚcodesÚuniquesrg   Úexpected_codesÚexpected_uniquesr   r   r   Útest_factorize  s   zBaseMethodsTests.test_factorizec                 C   sh   t j|dd\}}|jdd\}}t ||¡ t ||¡ t|ƒtt  |¡ƒks*J ‚|j|jks2J ‚d S )NTr”   )r!   r—   r$   rO   r˜   r3   r0   r	   )r   r™   Zcodes_1Z	uniques_1Zcodes_2Z	uniques_2r   r   r   Útest_factorize_equivalence  s   z+BaseMethodsTests.test_factorize_equivalencec                 C   s^   t  |d d… ¡\}}tjg tjd}t|ƒjg |d d… jd}t 	||¡ t 
||¡ d S )Nr   r*   )r!   r—   r
   r1   rU   r   ro   r	   r$   rO   r˜   )r   r   rš   r›   rœ   r   r   r   r   Útest_factorize_empty  s
   z%BaseMethodsTests.test_factorize_emptyc                 C   sR   |  ddg¡}t d|i¡}| ¡ }|jd }| |¡}||jd< t ||¡ d S )Nr,   r8   )r   r   )re   r!   r<   Úcopyr‡   Úfillnar$   rŽ   )r   r>   Úarrr?   Zdf_origÚ
filled_valr'   r   r   r   Útest_fillna_copy_frame%  s   


z'BaseMethodsTests.test_fillna_copy_framec                 C   sP   |  ddg¡}tj|dd}| ¡ }|d }| |¡}||jd< t ||¡ d S )Nr,   F)r¡   r   )re   r!   r"   r¡   r¢   r‡   r$   r%   )r   r>   r£   r6   Zser_origr¤   r'   r   r   r   Útest_fillna_copy_series1  s   

z(BaseMethodsTests.test_fillna_copy_seriesc                 C   sJ   d}t jt|d | | dg¡¡ W d   ƒ d S 1 sw   Y  d S )Nz!Length of 'value' does not match.rZ   r,   )r   rk   rl   r¢   re   )r   r>   r_   r   r   r   Útest_fillna_length_mismatch<  s   "ÿz,BaseMethodsTests.test_fillna_length_mismatchÚ_combine_le_expected_dtypec                    s°   |dƒ\}}t  |¡}t  |¡}| |dd„ ¡}t jdd„ tt|ƒt|ƒƒD ƒ| jd}t ||¡ |jd ‰ | ˆ dd„ ¡}t j‡ fd	d„t|ƒD ƒ| jd}t ||¡ d S )
NrS   c                 S   ó   | |kS rB   r   ©Úx1Zx2r   r   r   rL   J  r‘   z2BaseMethodsTests.test_combine_le.<locals>.<lambda>c                 S   s   g | ]\}}||k‘qS r   r   ©Ú.0ÚaÚbr   r   r   Ú
<listcomp>L  ó    z4BaseMethodsTests.test_combine_le.<locals>.<listcomp>r*   r   c                 S   r©   rB   r   rª   r   r   r   rL   R  r‘   c                    s   g | ]}|ˆ k‘qS r   r   ©r­   r®   ©Úvalr   r   r°   T  ó    )	r!   r"   ÚcombineÚzipÚlistr¨   r$   r%   r‡   )r   Údata_repeatedÚ
orig_data1Ú
orig_data2Ús1Ús2r'   r(   r   r³   r   Útest_combine_leD  s    

þ
þz BaseMethodsTests.test_combine_lec                    s4  |dƒ\}}t  |¡}t  |¡}z,tjdd t  | dd„ tt|ƒt|ƒƒD ƒ¡¡}W d   ƒ n1 s6w   Y  W n) tye   t 	t¡ | 
|dd„ ¡ W d   ƒ Y d S 1 s]w   Y  Y d S w | 
|dd„ ¡}t ||¡ |jd	 ‰ | 
ˆ d
d„ ¡}t  | ‡ fdd„t|ƒD ƒ¡¡}t ||¡ d S )NrS   rH   )Zoverc                 S   s   g | ]\}}|| ‘qS r   r   r¬   r   r   r   r°   e  r±   z5BaseMethodsTests.test_combine_add.<locals>.<listcomp>c                 S   ó   | | S rB   r   rª   r   r   r   rL   l  r‘   z3BaseMethodsTests.test_combine_add.<locals>.<lambda>c                 S   r¿   rB   r   rª   r   r   r   rL   o  r‘   r   c                 S   r¿   rB   r   rª   r   r   r   rL   s  r‘   c                    s   g | ]}|ˆ  ‘qS r   r   r²   r³   r   r   r°   u  rµ   )r!   r"   r
   Zerrstatero   r·   r¸   Ú	TypeErrorr   rk   r¶   r$   r%   r‡   )r   r¹   rº   r»   r¼   r½   r(   r'   r   r³   r   Útest_combine_addY  s8   

ÿÿÿ€
ÿþû
ÿz!BaseMethodsTests.test_combine_addc                 C   sX   t  |d d… ¡}t j|dd… g d¢d}| |¡}t  |d d… ¡}t ||¡ d S )Nra   rS   é   )rS   ra   r•   r   )r!   r"   Zcombine_firstr$   r%   )r   r   r®   r¯   r'   r(   r   r   r   Útest_combine_firsty  s
   
z#BaseMethodsTests.test_combine_firstÚframezperiods, indiceséþÿÿÿ)rS   ra   r•   r\   r\   r   )r   r,   rS   ra   r•   rS   )r\   r\   r   r,   rS   c           	      C   sž   |d d… }t j|dd}t j|j|dddd}|r@|jddjdd |¡}t j|t jdgd dd |¡gdd	}tj}n| |¡}tj	}|||ƒ d S )
NrÂ   r8   ©r.   T©Z
allow_fillr,   )rŒ   rŒ   r:   )
r!   r"   re   Úto_frameZassignÚshiftÚconcatr$   rŽ   r%   )	r   r   rÄ   ÚperiodsÚindicesÚsubsetr(   r'   Úcomparer   r   r   Útest_container_shift  s   ÿ
z%BaseMethodsTests.test_container_shiftc                 C   sB   |  d¡}|d |d ksJ ‚|d |d< |d |d ksJ ‚d S r’   ©rÉ   rE   r   r   r   Útest_shift_0_periods˜  s   
z%BaseMethodsTests.test_shift_0_periodsrË   r,   c           	      C   sü   |d d… }t |jƒrtj}ntj}z|||ƒ W n ty,   t t|ƒ› d¡ Y nw t	 
|¡}| |¡}t	 
||| |¡ƒ¡}t ||¡ t	 |dgd dœ¡}| |¡}|dkretjddddg}n	dddtjtjg}t	 ||dœ¡}t ||¡ d S )NrÂ   z does not support diffg      ð?r‹   r,   r   )r   r	   ÚoperatorÚxorÚsubÚ	Exceptionr   r   r   r!   r"   ÚdiffrÉ   r$   r%   r<   r
   ÚnanrŽ   )	r   r   rË   ÚopÚsr'   r(   r?   r¯   r   r   r   Ú	test_diffŸ  s(   
ÿ


zBaseMethodsTests.test_difféüÿÿÿr\   r•   c                 C   s4   |d d… }|  |¡}|j|dd}t ||¡ d S )NrS   TrÇ   ©rÉ   re   r$   r˜   )r   r   rË   rÌ   rÍ   r'   r(   r   r   r   Útest_shift_non_empty_array¹  s   
z+BaseMethodsTests.test_shift_non_empty_array)rÛ   r\   r   r,   r•   c                 C   s*   |d d… }|  |¡}|}t ||¡ d S )Nr   )rÉ   r$   r˜   )r   r   rË   Úemptyr'   r(   r   r   r   Útest_shift_empty_arrayÄ  s   
z'BaseMethodsTests.test_shift_empty_arrayc                 C   s8   |  d¡}||usJ ‚|d d…   d¡}||usJ ‚d S )Nr   rS   rÐ   rE   r   r   r   Útest_shift_zero_copiesÌ  s   
z'BaseMethodsTests.test_shift_zero_copiesc                 C   sh   |d d… }|d }|j d|d}| g d¢¡}t ||¡ |j d|d}| g d¢¡}t ||¡ d S )Nr•   r   r,   )Ú
fill_value)r   r   r,   rS   rÅ   )rS   ra   r   r   rÜ   )r   r   r£   rá   r'   r(   r   r   r   Útest_shift_fill_valueÔ  s   z&BaseMethodsTests.test_shift_fill_valuec                 C   s<   t jtdd t|ƒ W d   ƒ d S 1 sw   Y  d S )Nzunhashable typerZ   )r   rk   rÀ   Úhash©r   r   r   r   r   Útest_not_hashableß  s   
"ÿz"BaseMethodsTests.test_not_hashablec                 C   s>   t  |¡}|r| ¡ }t j |¡}t j |¡}t ||¡ d S rB   )r!   r"   rÈ   ÚutilZhash_pandas_objectr$   Ú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  |j jr
|  ||¡S |\}}}| g d¢¡}|rt |¡}| |¡dks&J ‚|j|dddks1J ‚| |¡dks:J ‚|j|dddksEJ ‚| |¡dksNJ ‚|j|dddksYJ ‚| | ddg¡¡}tjddgtj	d}t
 ||¡ t g d	¢¡}	|j||	d
dks…J ‚d S )NrR   r   Úright©Zsider,   rS   ra   r*   r€   ©Úsorter)r	   rb   Ú_test_searchsorted_bool_dtypesre   r!   r"   Úsearchsortedr
   r1   rU   r$   rO   )
r   rV   Ú	as_seriesr¯   Úcr®   r£   r'   r(   rí   r   r   r   Útest_searchsortedí  s"   

z"BaseMethodsTests.test_searchsortedc           
      C   sì   |j }tjddg|d}|\}}t|ƒ ||g¡}|r t |¡}| |¡dks)J ‚|j|dddks4J ‚| |¡dks=J ‚|j|dddksHJ ‚| | ddg¡¡}tjddgtj	d}t
 ||¡ t ddg¡}	|j||	d	dkstJ ‚d S )
NTFr*   r   rê   rë   r,   rS   rì   )r	   r!   r1   r   ro   r"   rï   re   r
   rU   r$   rO   )
r   rV   rð   r	   r¯   r®   r£   r'   r(   rí   r   r   r   rî     s   
z/BaseMethodsTests._test_searchsorted_bool_dtypesc                 C   s¢  |d |d ks
J ‚t |ƒ}|d d… \}}t |j||||g|jd¡}| ¡ }t g d¢¡}	|r>|jdd}|	 	dd¡}	| 
|	¡}
t |j||||g|jd¡}|rZ|jdd}t |
|¡ |j|	 d	d
 t ||¡ | ¡ }|rz|jdd}t g d¢¡}	|j||||g|jd}|rt d|i¡}t d|	i¡}	| 
|	|¡}
t |j||||g|jd¡}|rº|jdd}t |
|¡ |j|	 |d	d
 t ||¡ d S )Nr   r,   rS   r*   )TTFFr®   rÆ   r\   T)Zinplace)TFTT)r   r!   r"   ro   r	   r¡   r
   r1   rÈ   ZreshapeÚwherer$   rç   Úmaskr<   )r   r   rf   rè   Úclsr®   r¯   Úorigr6   Zcondr'   r(   r&   r   r   r   Útest_where_series"  sB   
ÿz"BaseMethodsTests.test_where_seriesÚrepeats)r,   rS   ra   c                 C   s®   t |ƒj|d d… |jd}|rt |¡}|rt ||¡n| |¡}t|tƒr,|gd n|}dd„ t	||ƒD ƒ}t |ƒj||jd}|rOtj||j
 |¡d}t ||¡ d S )Nra   r*   c                 S   s"   g | ]\}}t |ƒD ]}|‘q
qS r   )Úrange)r­   rK   ÚnÚ_r   r   r   r°   V  s   " z0BaseMethodsTests.test_repeat.<locals>.<listcomp>r   )r   ro   r	   r!   r"   r
   Úrepeatr2   Úintr·   r-   r$   rç   )r   r   rø   rð   Ú	use_numpyr£   r'   r(   r   r   r   Útest_repeatM  s   
zBaseMethodsTests.test_repeatzrepeats, kwargs, error, msgr;   Únegativer   ZfooÚbarz'foo'c                 C   sp   t j||d' |rtj||fi |¤Ž n|j|fi |¤Ž W d   ƒ d S W d   ƒ d S 1 s1w   Y  d S )NrZ   )r   rk   r
   rü   )r   r   rø   ÚkwargsÚerrorr_   rþ   r   r   r   Útest_repeat_raises]  s   
ü"þz#BaseMethodsTests.test_repeat_raisesc                 C   sd   |  d¡}|dd … }t ||¡ |  ddg¡}| |dg |dg |dd … g¡}t ||¡ d S )Nr   r,   ra   rS   r•   )Údeleter$   r˜   Z_concat_same_type)r   r   r'   r(   r   r   r   Útest_deletem  s   
$zBaseMethodsTests.test_deletec                 C   sº   |dd …   d|d ¡}t ||¡ |dd …   t|dd … ƒ |d ¡}t ||¡ |d d…   d|d ¡}t t|ƒ¡}|dd… |dd …< t|ƒd |d< | |¡}t ||¡ d S )Nr,   r   r\   r•   rÂ   )Úinsertr$   r˜   r3   r
   Zarangere   )r   r   r'   Ztakerr(   r   r   r   Útest_insertv  s   &
zBaseMethodsTests.test_insertc                 C   sÀ   |}t  ttf¡ | d|¡ W d   ƒ n1 sw   Y  t  ttf¡ | d|¡ W d   ƒ n1 s7w   Y  t  ttf¡ | t|ƒd |¡ W d   ƒ d S 1 sYw   Y  d S )Nr   r•   r,   )r   rk   rÀ   rl   r  r3   )r   r   Zinvalid_scalarÚitemr   r   r   Útest_insert_invalid‡  s   ÿÿ"ÿz$BaseMethodsTests.test_insert_invalidc                 C   sÆ   t |ƒ}t t¡ | |d |d ¡ W d   ƒ n1 sw   Y  t t¡ | | d |d ¡ W d   ƒ n1 s>w   Y  t t¡ | d|d ¡ W d   ƒ d S 1 s\w   Y  d S )Nr,   r   g      ø?)r3   r   rk   Ú
IndexErrorr  rÀ   )r   r   Zubr   r   r   Útest_insert_invalid_loc“  s   ÿÿ"þz(BaseMethodsTests.test_insert_invalid_locc                 C   s2  t |ƒj|d gt|ƒ |jd}t |ƒj|gt|ƒ |jd}tj||dd}tj||dd}tj||dd}| |¡du sAJ ‚| | ¡ ¡du sLJ ‚| |¡du sUJ ‚| |¡du s^J ‚|d d…  |d d… ¡du soJ ‚|d d…  |d d… ¡du s€J ‚| d ¡du s‰J ‚|dg  |d ¡du s—J ‚d S )Nr   r*   F)Z	transposeTrS   ra   )r   ro   r3   r	   r$   Zbox_expectedÚequalsr¡   )r   r   rf   rð   r   Zdata2rp   r   r   r   Útest_equals   s   """ zBaseMethodsTests.test_equalsc                 C   s   t  |¡ t  |¡¡sJ ‚d S rB   )r!   r"   r  rä   r   r   r   Ú&test_equals_same_data_different_object½  s   z7BaseMethodsTests.test_equals_same_data_different_object)GÚ__name__Ú
__module__Ú__qualname__Ú__doc__r   r   r   ÚmarkZparametrizer)   r7   r@   rA   rF   rP   rW   rY   r`   rh   rn   rq   r
   r×   r{   r}   r1   r	   r‚   r‰   rŠ   r   r!   r"   r0   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÷   rÿ   rl   rÀ   r  r  r  r
  r  r<   r  r  r   r   r   r   r      sÊ   
 





øþ

þþ



		  þ
4þ

	+

üþ
		
r   )r   rÒ   Únumpyr
   r   Zpandas._typingr   Zpandas.core.dtypes.commonr   Zpandas.core.dtypes.missingr   Zpandasr!   Zpandas._testingZ_testingr$   Zpandas.core.sortingr   r   r   r   r   r   Ú<module>   s    