
    aj>s                         S SK r S SKrS SKrS SKrS SKJr  S SKJrJ	r	J
r
  S SKJr  S SKJr  S SKrS SKJr  S SKJr   " S S5      rg)	    N)Dtype)is_bool_dtypeis_float_dtypeis_integer_dtype)NumpyEADtype)na_value_for_dtype)nargsortc                      \ rS rSr% SrS rS r\R                  R                  SSS/5      S 5       r
S	 rS
 rS rS r\R                  R                  SSS/5      S 5       rS rS rS rS r\R                  R                  SSS/5      S 5       r\R                  R                  SSS/5      S 5       r\R                  R                  S/ SQ5      S 5       rS r\R                  R                  SS\R2                  " / S Q\R4                  " S!5      S"94S#\R2                  " / S$Q\R4                  " S!5      S"94/5      S% 5       r\R                  R                  S&SS/5      S' 5       r\R                  R                  S&SS/5      S( 5       r\R                  R                  S&SS/5      S) 5       r\R                  R                  S&SS/5      S* 5       r\R                  R                  SS+S,/5      S- 5       r \R                  R                  S./ S/Q5      S0 5       r!\R                  R                  S1/ S2Q5      S3 5       r"\R                  R                  S4\#RH                  S5 /5      \R                  R                  SS6 \#RJ                  /5      S7 5       5       r&S8 r'S9 r(S: r)S; r*S< r+S= r,S> r-S? r.\/" S@5      r0\1\2SA'   SB r3SC r4SD r5SE r6\R                  R                  SFSS/5      \R                  R                  SGSH/ SIQ4SJ/ SKQ4SL/ SMQ4/5      SN 5       5       r7SO r8\R                  R                  SPSQSH/5      SR 5       r9\R                  R                  SGSSSTST//STSQST//SJSJSQ//SQSTSJ//SUSTST///5      SV 5       r:\R                  R                  SP/ SWQ5      SX 5       r;SY r<SZ r=S[ r>S\ r?S] r@S^ rAS_ rB\R                  R                  S`SJSQSL/ SaQ/5      Sb 5       rC\R                  R                  ScSLSdSQ0\DSd4ST0 \DSe4SQSL/0 \DSf4SLSgSh0\ESi4/5      Sj 5       rFSk rGSl rHSm rISn rJ\R                  R                  S4\#R2                  \#RH                  \#R                  /5      So 5       rLSp rMSqrNg)rBaseMethodsTests   z%Various Series and DataFrame methods.c                     SSK Jn  UR                  SUSS9nUR                  [        R
                  :X  d   eUR                  UR                  :X  d   eg )Nr   )_default_hash_keyzutf-8F)encodinghash_key
categorize)pandas.core.util.hashingr   _hash_pandas_objectdtypenpuint64shape)selfdatar   ress       f/var/www/html/Sattelite-Image/venv/lib/python3.13/site-packages/pandas/tests/extension/base/methods.pytest_hash_pandas_object(BaseMethodsTests.test_hash_pandas_object   sR     	?&&'8U ' 
 yyBII%%%yyDJJ&&&    c                     [        US5      (       d"  [        R                  " S[        U5       35        [        R
                  " UR                  5      nUR                  S   nUR                  SL d   eg )Nvalue_countsz$value_counts is not implemented for dropnaT)	hasattrpytestskiptypeinspect	signaturer    
parametersdefault)r   r   sigkwargs       r    test_value_counts_default_dropna1BaseMethodsTests.test_value_counts_default_dropna$   s]    t^,,KK>tDzlKL 1 12x(}}$$$r   r!   TFc                 .   U(       a  XR                  5       )    nOUn[        R                  " U5      R                  US9R	                  5       n[        R                  " U5      R                  US9R	                  5       n[
        R                  " XE5        g )N)r!   )isnapdSeriesr    
sort_indextmassert_series_equal)r   all_datar!   otherresultexpecteds         r   test_value_counts"BaseMethodsTests.test_value_counts,   ss    mmo-.EE8$111@KKM99U#000?JJL
v0r   c                 
   UR                  5       n[        R                  " XR                  5       )    5      n[        R
                  " XR                  S9nUR                  SS9R                  5       n[        U[        R                  5      (       d:  [        R
                  " S[        U5      -  /[        U5      -  UR                  SS9nO3[        R
                  " SUR                  SS9nS[        U5      -  XTS:  '   [        UR                  [        R                  5      (       a9  UR                  R                  [        R                  L a  UR!                  S	5      nO[#        UR                  S
S5      S:X  d)  [        UR                  [        R$                  5      (       a  UR!                  S5      nO7['        UR                  5      [        R(                  L a  UR!                  S5      n[*        R,                  " XE5        g )Nr   T)	normalize   
proportion)indexnameg        r   float64storage pyarrowzdouble[pyarrow]Float64)uniquer   arrayr/   r0   r1   r   r    r2   
isinstanceCategoricallenr@   StringDtypena_valuenanastypegetattr
ArrowDtyper   NAr3   r4   )r   r   valuesserr7   r8   s         r    test_value_counts_with_normalize1BaseMethodsTests.test_value_counts_with_normalize8   sh   {{}$		|,-iiJJ/!!D!1<<>$//yyS[!CK/v||,H yyFLL|LH#$s6{?HaZ djj"..11djj6I6IRVV6Sy1HTZZB/9<
JJA
 A
  '89H

+ruu4y1H
v0r   c                     [         R                  " SU05      nUR                  SS9n[         R                  " SS/5      n[        R
                  " X45        g )NAcolumnsaxisr   r>   )r0   	DataFramecountr1   r3   r4   )r   data_missingdfr7   r8   s        r   
test_countBaseMethodsTests.test_countV   sE    \\3-.y)99aV$
v0r   c                 b    [         R                  " U5      nUR                  5       nSnX4:X  d   eg )Nr>   )r0   r1   r]   )r   r^   rT   r7   r8   s        r   test_series_count"BaseMethodsTests.test_series_count\   s,    ii%!!!r   c                     [         R                  " U5      R                  [        5      n[	        U[         R                  5      (       d   eg N)r0   r1   applyidrI   r   r   r7   s      r   test_apply_simple_series)BaseMethodsTests.test_apply_simple_seriesc   s1    4&&r*&")),,,,r   	na_actionNignorec                 r    UR                  S US9nUR                  5       n[        R                  " X45        g )Nc                     U $ rf    xs    r   <lambda>+BaseMethodsTests.test_map.<locals>.<lambda>i   s    Ar   )rl   )mapto_numpyr3   assert_numpy_array_equal)r   r^   rl   r7   r8   s        r   test_mapBaseMethodsTests.test_mapg   s3    !!+!C((*
##F5r   c                     [         R                  " U5      R                  5       n[         R                  " [        R                  " / SQ[        R
                  S95      n[        R                  " X#5        g N   r   r>   r<   r0   r1   argsortr   rH   intpr3   r4   )r   data_for_sortingr7   r8   s       r   test_argsortBaseMethodsTests.test_argsortm   sE    +,44699RXXirww?@
v0r   c                     UR                  5       n[        R                  " / SQ[        R                  S9n[        R
                  " X#5        g r{   )r   r   rH   r   r3   rw   r   data_missing_for_sortingr7   r8   s       r   test_argsort_missing_array+BaseMethodsTests.test_argsort_missing_arrays   s2    )11388IRWW5
##F5r   c                     [         R                  " U5      R                  5       n[         R                  " [        R                  " / SQ[        R
                  S95      n[        R                  " X#5        g r{   r~   r   s       r   test_argsort_missing%BaseMethodsTests.test_argsort_missingy   sE    34<<>99RXXirww?@
v0r   c                 |   UR                   R                  nSnSnU(       a  SnSnUR                  5       U:X  d   eUR                  5       S:X  d   eUR	                  / SQ5      nUR                  5       U:X  d   eUR                  5       S:X  d   eUR                  5       S:X  d   eUR                  5       S:X  d   eg )Nr>      r   r}   )r}   r   r   r>   r>   r}   )r   _is_booleanargmaxargmintake)r   r   r   rM   is_bool
exp_argmaxexp_argmax_repeatedr   s           r   test_argmin_argmax#BaseMethodsTests.test_argmin_argmax~   s    "((44
J"#  &&(J666&&(A---  $$%78{{} 3333{{}!!! (..0A555'..0A555r   methodr   r   c                     Sn[         R                  " [        US9   [        US S U5      " 5         S S S 5        g ! , (       d  f       g = f)Nattempt to getmatchr   )r#   raises
ValueErrorrP   )r   r   r   err_msgs       r   test_argmin_argmax_empty_array/BaseMethodsTests.test_argmin_argmax_empty_array   s6     #]]:W5D!Hf%' 655s	   :
Ac                     Sn[        U5      R                  X3/UR                  S9n[        R                  " [
        US9   [        XQ5      " 5         S S S 5        g ! , (       d  f       g = f)Nr   r<   r   )r%   _from_sequencer   r#   r   r   rP   )r   r   r   rM   r   data_nas         r   test_argmin_argmax_all_na*BaseMethodsTests.test_argmin_argmax_all_na   sN     #t*++X,@

+S]]:W5G$& 655s    A
A(zop_name, skipna, expected))idxmaxTr   )idxminTr}   )r   Tr   )r   Tr}   )r   F)r   Fr   )r   Fr   )r   Fr   c                    [         R                  " U5      nUS:X  a2  [        R                  " [        SS9   [        XR5      " US9  S S S 5        g [        XR5      " US9n[        R                  " Xd5        g ! , (       d  f       g = f)Nr   Encountered an NA valuer   skipna)r0   r1   r#   r   r   rP   r3   assert_almost_equal)r   r   op_namer   r8   rT   r7   s          r   test_argreduce_series&BaseMethodsTests.test_argreduce_series   sh    " ii01r>z1JK%V4 LK S*&9F""64	 LKs   A55
Bc                    Un[         R                  " [        SS9   UR                  SS9  S S S 5        [         R                  " [        SS9   UR	                  SS9  S S S 5        g ! , (       d  f       N@= f! , (       d  f       g = f)Nr   r   Fr   )r#   r   r   r   r   )r   r   r   s      r   +test_argmax_argmin_no_skipna_notimplemented<BaseMethodsTests.test_argmax_argmin_no_skipna_notimplemented   sf    ']]:-FGKKuK% H ]]:-FGKKuK% HG HG HGs   A&A7&
A47
Bzna_position, expectedlastr|   r   r<   firstr>   r}   r   c                 B    [        XS9n[        R                  " XC5        g )N)na_position)r	   r3   rw   )r   r   r   r8   r7   s        r   test_nargsortBaseMethodsTests.test_nargsort   s     2L
##F5r   	ascendingc                    [         R                  " U5      nUR                  X#S9nUR                  / SQ   nU(       d7  UR	                  5       S:X  a  UR                  / SQ   nOUR                  / SQ   n[
        R                  " XV5        g )Nr   keyr|   r}   )r   r>   r}   )r>   r   r}   )r0   r1   sort_valuesilocnuniquer3   r4   )r   r   r   sort_by_keyrT   r7   r8   s          r   test_sort_values!BaseMethodsTests.test_sort_values   sj    ii()9F88I&{{}!88I.88I.
v0r   c                     [         R                  " U5      nUR                  X#S9nU(       a  UR                  / SQ   nOUR                  / SQ   n[        R
                  " XV5        g )Nr   r|   )r   r}   r>   )r0   r1   r   r   r3   r4   )r   r   r   r   rT   r7   r8   s          r   test_sort_values_missing)BaseMethodsTests.test_sort_values_missing   sP     ii019Fxx	*Hxx	*H
v0r   c                     [         R                  " / SQUS.5      nUR                  SS/5      n[         R                  " / SQUR                  / SQ5      S./ SQS9n[        R
                  " XE5        g )N)r>   r}   r>   rX   BrX   r   )r>   r>   r}   r|   r@   )r0   r\   r   r   r3   assert_frame_equal)r   r   r   r_   r7   r8   s         r   test_sort_values_frame'BaseMethodsTests.test_sort_values_frame   s\    \\	0@ABc
+<<"2"7"7	"BC9
 	f/r   c                    [         R                  " U5      nUR                  US9n[        UR                  5      (       d   eUR                  S5      n[        UR                  5      (       a$  [         R                  " U(       a  / SQO/ SQ5      nO#[         R                  " U(       a  / SQO/ SQ5      n[        R                  " XE5        g )N)r   rB   )      @r         ?)      ?r         @       @r   r   )r   r   r   )	r0   r1   rankr   r   rO   r   r3   r4   )r   r   r   rT   r7   r8   s         r   	test_rankBaseMethodsTests.test_rank   s    ii()I. fll++++y)##yyI?SHyyI?SH
v0r   averageminc                    [         R                  " UR                  / SQ5      5      nUR                  US9nUS:X  aF  [	        UR
                  5      (       d   eUR                  S5      n[         R                  " / SQ5      nOW[        UR
                  5      (       d  [	        UR
                  5      (       d   e[         R                  " / SQUR
                  S9n[        R                  " XE5        g )N)r   r}   r   )r   r   rB   )r   r   r   )r}   r>   r}   r<   )
r0   r1   r   r   r   r   rO   r   r3   r4   )r   r   r   rT   r7   r8   s         r   test_rank_method!BaseMethodsTests.test_rank_method  s    ii(--i89(Y!&,,////]]9-Fyy1H $FLL11^FLL5Q5QQQyy&,,?H
v0r   	na_option)keeptopbottomc                    [         R                  " U5      nUR                  US9n[        UR                  5      (       d   eUR                  S5      nUS:X  a(  [         R                  " S[        R                  S/5      nO7US:X  a  [         R                  " / SQ5      nO[         R                  " / SQ5      n[        R                  " XE5        g )	N)r   rB   r   r   r   r   )r   r   r   r   )
r0   r1   r   r   r   rO   r   rN   r3   r4   )r   r   r   rT   r7   r8   s         r   test_rank_missing"BaseMethodsTests.test_rank_missing  s    ii01I.fll++++y)yy#rvvs!34H%yy1H yy1H
v0r   r   )r   r   Fc                     UR                  / SQ5      nUR                  US9nUS:X  a  [        R                  " / SQ5      nO7US:X  a  [        R                  " / SQ5      nO[        R                  " / SQ5      n[        R
                  " XE5        g )Nr   r>   r   r>   )r   r   )FFTTr   TTFF)TTTT)r   
duplicatedr   rH   r3   rw   )r   r   r   arrr7   r8   s         r   test_duplicated BaseMethodsTests.test_duplicated$  si    ii%T*7?xx :;HV^xx :;Hxx 89H
##F5r   boxc                     U $ rf   rp   rq   s    r   rs   BaseMethodsTests.<lambda>0  s    !r   c                 "    U R                  5       $ rf   )rG   rq   s    r   rs   r   1  s
    !((*r   c                     U" UR                  US   US   /UR                  S95      nU" U5      n[        U5      S:X  d   e[        U[	        U5      5      (       d   eUS   US   :X  d   eg )Nr   r<   r>   )r   r   rK   rI   r%   )r   r   r   r   r   r7   s         r   test_uniqueBaseMethodsTests.test_unique0  sx     ,,d1gtAw-?tzz,RS

#6{a&$t*----ayJqM)))r   c                    [         R                  " USS9u  p#UR                  R                  nU(       a9  [        R
                  " / SQ[        R                  S9nUR                  SS/5      nO8[        R
                  " / SQ[        R                  S9nUR                  / SQ5      n[        R                  " X%5        [        R                  " X65        g )	NTuse_na_sentinel)r   r   r   r   r>   r>   r   r   r<   r      )r   r   r   r   r>   r>   r   r}   )r   r      )r0   	factorizer   r   r   rH   r   r   r3   rw   assert_extension_array_equal)r   data_for_groupingcodesuniquesr   expected_codesexpected_uniquess          r   test_factorizeBaseMethodsTests.test_factorize;  s    &7N#))55XX&@PN055q!f=XX&@PN055i@
##E:
''Br   c                 @   [         R                  " USS9u  p#UR                  SS9u  pE[        R                  " X$5        [        R                  " X55        [        U5      [        [         R                  " U5      5      :X  d   eUR                  UR                  :X  d   eg )NTr   )r0   r   r3   rw   r   rK   rG   r   )r   r   codes_1	uniques_1codes_2	uniques_2s         r   test_factorize_equivalence+BaseMethodsTests.test_factorize_equivalenceJ  s    \\*;TR.888N
##G5
''	=9~RYYy%9!::::"3"9"9999r   c                 $   [         R                  " US S 5      u  p#[        R                  " / [        R                  S9n[        U5      R                  / US S R                  S9n[        R                  " X$5        [        R                  " X55        g )Nr   r<   )r0   r   r   rH   r   r%   r   r   r3   rw   r   )r   r   r   r   r   r   s         r   test_factorize_empty%BaseMethodsTests.test_factorize_emptyS  sl    d2Ah/"BGG4:44RtBQx~~4N
##E:
''Br   c                     [         R                  " SUR                  / SQ5      05      n[         R                  " SUR                  / SQ5      05      nUR                  US   SS9n[        R
                  " XC5        g )NrX   r   r>   r>   r   r>   r>   valuelimit)r0   r\   r   fillnar3   r   )r   r^   r_   r8   r7   s        r   test_fillna_limit_frame(BaseMethodsTests.test_fillna_limit_frame[  sb    \\3 1 1, ?@A<<l&7&7&E FGa:
f/r   c                     [         R                  " UR                  / SQ5      5      n[         R                  " UR                  / SQ5      5      nUR                  US   SS9n[        R
                  " XC5        g )Nr   r  r>   r  )r0   r1   r   r  r3   r4   )r   r^   rT   r8   r7   s        r   test_fillna_limit_series)BaseMethodsTests.test_fillna_limit_seriesb  sX    ii)),7899\..|<=,q/;
v0r   c                    UR                  SS/5      n[        R                  " SU05      nUR                  5       nUR                  S   nUR                  U5      nXVR                  S'   [        R                  " X45        g )Nr>   rX   )r   r   )r   r0   r\   copyr   r  r3   r   )r   r^   r   r_   df_orig
filled_valr7   s          r   test_fillna_copy_frame'BaseMethodsTests.test_fillna_copy_framei  sh    A'\\3*%'')WWT]
:&&D
b*r   c                     UR                  SS/5      n[        R                  " USS9nUR                  5       nUS   nUR	                  U5      nXVR
                  S'   [        R                  " X45        g )Nr>   F)r  r   )r   r0   r1   r  r  r   r3   r4   )r   r^   r   rT   ser_origr  r7   s          r   test_fillna_copy_series(BaseMethodsTests.test_fillna_copy_seriesu  sa    A'ii%(88:V
J'#A
s-r   c                     Sn[         R                  " [        US9   UR                  UR	                  S/5      5        S S S 5        g ! , (       d  f       g = f)Nz!Length of 'value' does not match.r   r>   )r#   r   r   r  r   )r   r^   msgs      r   test_fillna_length_mismatch,BaseMethodsTests.test_fillna_length_mismatch  s<    1]]:S1 1 11# 67 211s   "A
Abool_combine_le_expected_dtypec           
         U" S5      u  p#[         R                  " U5      n[         R                  " U5      nUR                  US 5      n[         R                  " [         R                  " [	        [        U5      [        U5      SS9 VVs/ s H	  u  pxXx:*  PM     snnU R                  S95      n	[        R                  " Xi5        UR                  S   n
UR                  U
S 5      n[         R                  " [         R                  " [        U5       Vs/ s H  owU
:*  PM	     snU R                  S95      n	[        R                  " Xi5        g s  snnf s  snf )Nr}   c                 
    X:*  $ rf   rp   x1x2s     r   rs   2BaseMethodsTests.test_combine_le.<locals>.<lambda>  s    rxr   Tstrictr<   r   c                 
    X:*  $ rf   rp   r(  s     r   rs   r+    s    r   )
r0   r1   combinerH   ziplistr%  r3   r4   r   )r   data_repeated
orig_data1
orig_data2s1s2r7   abr8   vals              r   test_combine_le BaseMethodsTests.test_combine_le  s    "/q!1
YYz"YYz"B 7899HH #&d:&6Z8HQU"V"V F"V 55
 	v0ggajC!8999HH#'
#34#3ac#3455
 	v0# 5s   
EEc                 P   [        U[        U5      5      (       aL  UR                  [        [	        U5      [	        U5      SS9 VVs/ s H	  u  p4X4-   PM     snnUR
                  S9$ UR                  [	        U5       Vs/ s H  o3U-   PM	     snUR
                  S9$ s  snnf s  snf )NTr,  r<   )rI   r%   r   r0  r1  r   )r   leftrightr7  r8  s        r   _construct_for_combine_add+BaseMethodsTests._construct_for_combine_add  s    eT$Z((&&%(dT%[%NO%N6A%NOjj '  
 &&$(J/JqUJ/jj '  	 P
 0s   B
?B#c                    U" S5      u  p#[         R                  " U5      n[         R                  " U5      n [        R                  " SS9   U R	                  X#5      nS S S 5        [         R                  " W5      nUR                  US 5      n[        R                  " X5        UR                  S   n	UR                  U	S 5      nU R	                  X)5      n[         R                  " U5      n[        R                  " X5        g ! , (       d  f       N= f! [
         aK    [        R                  " [
        5         UR                  US 5        S S S 5        O! , (       d  f       O= f g f = f)Nr}   rm   )overc                 
    X-   $ rf   rp   r(  s     r   rs   3BaseMethodsTests.test_combine_add.<locals>.<lambda>  s    bgr   c                 
    X-   $ rf   rp   r(  s     r   rs   rD    s    rwr   r   c                 
    X-   $ rf   rp   r(  s     r   rs   rD    s    r   )r0   r1   r   errstater?  	TypeErrorr#   r   r/  r3   r4   r   )
r   r2  r3  r4  r5  r6  r   r8   r7   r9  s
             r   test_combine_add!BaseMethodsTests.test_combine_add  s   !.q!1
YYz"YYz"	(+55jM , 99S>B 67
v0ggajC!78--j>99S>
v0# ,+ 	 y)

256 *))	sG   D DD 
DD D $E+:E	E+
E%	!E+*E+c                     [         R                  " US S 5      n[         R                  " USS / SQS9nUR                  U5      n[         R                  " US S 5      n[        R                  " XE5        g )Nr   r}      )r}   r   r   r   )r0   r1   combine_firstr3   r4   )r   r   r7  r8  r7   r8   s         r   test_combine_first#BaseMethodsTests.test_combine_first  s\    IId2AhIId1Qiy1#99T"1X&
v0r   framezperiods, indices)r}   r   r   r   r   r   )r   r>   r}   r   r   r}   )r   r   r   r>   r}   c                    US S n[         R                  " USS9n[         R                  " UR                  USS9SS9nU(       ay  UR                  SS9R	                  SS9R                  U5      n[         R                  " U[         R                  " S/S-  SS9R                  U5      /SS	9n[        R                  nO!UR                  U5      n[        R                  nU" Xv5        g )
NrL  rX   rA   T
allow_fillr>   )r   r   rZ   )
r0   r1   r   to_frameassignshiftconcatr3   r   r4   )	r   r   rP  periodsindicessubsetr8   r7   compares	            r   test_container_shift%BaseMethodsTests.test_container_shift  s     bqyyc*99V[[T[BM]]],33a38>>wGFyy299aS1W37==gFGaH ++GZZ(F,,G!r   c                 n    UR                  S5      nUS   US   :w  d   eUS   US'   US   US   :w  d   eg )Nr   r>   rX  ri   s      r   test_shift_0_periods%BaseMethodsTests.test_shift_0_periods  sH    AAw$q'!!!q'QayF1I%%%r   rZ  r>   c                    US S n[        UR                  5      (       a  [        R                  nO[        R                  n U" X5        [        R                  " U5      nUR                  U5      n[        R                  " U" XR                  U5      5      5      n[        R                  " XV5        [        R                  " US/S-  S.5      nUR                  U5      nUS:X  a  [         R"                  SSSS/nO#SSS[         R"                  [         R"                  /n[        R                  " XhS.5      n[        R$                  " XV5        g ! [
         a&    [        R                  " [        U5       S35         GN3f = f)NrL  z does not support diffr   r   r>   r   )r   r   operatorxorsub	Exceptionr#   r$   r%   r0   r1   diffrX  r3   r4   r\   r   rN   r   )	r   r   rZ  opsr7   r8   r_   r8  s	            r   	test_diffBaseMethodsTests.test_diff  s"   BQx$$BB	?tN IIdO99Rjj&9:;
v0\\C51956!a<Aq!$AAq"&&"&&)A<<h 78
f/  	?KK4:,&<=>	?s   E ,E>=E>r   r   c                 |    US S nUR                  U5      nUR                  USS9n[        R                  " XV5        g )Nr}   TrT  rX  r   r3   r   )r   r   rZ  r[  r\  r7   r8   s          r   test_shift_non_empty_array+BaseMethodsTests.test_shift_non_empty_array  s=     bqg&;;w4;8
''9r   )rn  r   r   r>   r   c                 `    US S nUR                  U5      nUn[        R                  " XE5        g )Nr   )rX  r3   r   )r   r   rZ  emptyr7   r8   s         r   test_shift_empty_array'BaseMethodsTests.test_shift_empty_array  s0     RaW%
''9r   c                 f    UR                  S5      nX!Ld   eUS S R                  S5      nX!Ld   eg )Nr   r}   ra  ri   s      r   test_shift_zero_copies'BaseMethodsTests.test_shift_zero_copies  s;    A!!!bq"!!!r   c                     US S nUS   nUR                  SUS9nUR                  / SQ5      n[        R                  " XE5        UR                  SUS9nUR                  / SQ5      n[        R                  " XE5        g )Nr   r   r>   )
fill_value)r   r   r>   r}   rQ  )r}   r   r   r   rp  )r   r   r   r{  r7   r8   s         r   test_shift_fill_value&BaseMethodsTests.test_shift_fill_value'  sr    2Ah!W
1499\*
''92*599\*
''9r   c                     [         R                  " [        SS9   [        U5        S S S 5        g ! , (       d  f       g = f)Nzunhashable typer   )r#   r   rH  hashr   r   s     r   test_not_hashable"BaseMethodsTests.test_not_hashable2  s$    ]]9,=>J ?>>s   /
=c                    [         R                  " U5      nU(       a  UR                  5       n[         R                  R	                  U5      n[         R                  R	                  U5      n[
        R                  " X45        g rf   )r0   r1   rV  utilhash_pandas_objectr3   assert_equal)r   r   as_framer7  r8  s        r   test_hash_pandas_object_works.BaseMethodsTests.test_hash_pandas_object_works7  sQ    yy==?DGG&&t,GG&&t,
r   c                    UR                   R                  (       a  U R                  X5      $ Uu  p4nUR                  / SQ5      nU(       a  [        R
                  " U5      nUR                  U5      S:X  d   eUR                  USS9S:X  d   eUR                  U5      S:X  d   eUR                  USS9S:X  d   eUR                  U5      S:X  d   eUR                  USS9S:X  d   eUR                  UR                  SS/5      5      n[        R                  " SS/[        R                  S9n[        R                  " Xx5        [        R                  " / S	Q5      n	UR                  XYS
9S:X  d   eg )Nr|   r   r>  sider>   r}   r   r<   r   sorter)r   r   _test_searchsorted_bool_dtypesr   r0   r1   searchsortedr   rH   r   r3   rw   )
r   r   	as_seriesr8  cr7  r   r7   r8   r  s
             r   test_searchsorted"BaseMethodsTests.test_searchsorted@  s`   !!--667GSS"a##I.))C.C"a'''0A555"a'''0A555"a'''0A555!!#((Aq6"2388QF"''2
##F5 )$,,Q,>!CCCr   c                    UR                   n[        R                  " SS/US9nUu  pE[        U5      R	                  XT/US9nU(       a  [        R
                  " U5      nUR                  U5      S:X  d   eUR                  USS9S:X  d   eUR                  U5      S:X  d   eUR                  USS9S:X  d   eUR                  UR                  SS/5      5      n[        R                  " SS/[        R                  S9n[        R                  " Xx5        [        R                  " SS/5      n	UR                  XYS	9S:X  d   eg )
NTFr<   r   r>  r  r>   r}   r  )r   r0   rH   r%   r   r1   r  r   r   r   r3   rw   )
r   r   r  r   r8  r7  r   r7   r8   r  s
             r   r  /BaseMethodsTests._test_searchsorted_bool_dtypes[  s9    !&&88T5M?#$33QF%3H))C.C"a'''0A555"a'''0A555!!#((Aq6"2388QF"''2
##F5 1a&!,,Q,>!CCCr   c                    US   US   :w  d   e[        U5      nUS S u  pV[        R                  " UR                  XUXf/UR                  S95      nUR                  5       n[        R                  " / SQ5      n	U(       a!  UR                  SS9nU	R                  SS5      n	UR                  U	5      n
[        R                  " UR                  XUX"/UR                  S95      nU(       a  UR                  SS9n[        R                  " X5        UR                  U	) S	S
9  [        R                  " X5        UR                  5       nU(       a  UR                  SS9n[        R                  " / SQ5      n	UR                  XVXV/UR                  S9nU(       a0  [        R                  " SU05      n[        R                  " SU	05      n	UR                  X5      n
[        R                  " UR                  XVXf/UR                  S95      nU(       a  UR                  SS9n[        R                  " X5        UR                  U	) US	S
9  [        R                  " X5        g )Nr   r>   r}   r<   r   r7  rS  r   T)inplace)TFTT)r%   r0   r1   r   r   r  r   rH   rV  reshapewherer3   r  maskr\   )r   r   rM   r  clsr7  r8  origrT   condr7   r8   r6   s                r   test_where_series"BaseMethodsTests.test_where_seriesu  s   Aw$q'!!!4jBQxyy++Q1L

+KLiikxx23,,C,(C<<A&D499h9L
 ((c(2H
)$%
& iik,,C,(Cxx12""A!<tzz"BLL#u.E<<d,D4'99S//qDJJ/OP((c(2H
)$t,
&r   repeats)r>   r}   r   c                 j   [        U5      R                  US S UR                  S9nU(       a  [        R                  " U5      nU(       a  [
        R                  " XR5      OUR                  U5      n[        U[        5      (       a  U/S-  OUn[        XRSS9 VVV	s/ s H  u  px[        U5        H  oPM     M     n
nnn	[        U5      R                  XR                  S9n
U(       a-  [        R                  " XR                  R                  U5      S9n
[        R                  " Xj5        g s  sn	nnf )Nr   r<   Tr,  r   )r%   r   r   r0   r1   r   repeatrI   intr0  ranger@   r3   r  )r   r   r  r  	use_numpyr   r7   rr   n_r8   s              r   test_repeatBaseMethodsTests.test_repeat  s    4j''Ra

'C))C.C,53(3::g;N#-gs#;#;7)a-"%c4"@S"@$!%PQ(QA(A"@S:,,XZZ,Hyy1A1A'1JKH
) Ts    D.zrepeats, kwargs, error, msgr[   negativer   foobarz'foo'c                     [         R                  " XES9   U(       a  [        R                  " X40 UD6  OUR                  " U40 UD6  S S S 5        g ! , (       d  f       g = f)Nr   )r#   r   r   r  )r   r   r  kwargserrorr!  r  s          r   test_repeat_raises#BaseMethodsTests.test_repeat_raises  sA     ]]5,		$262G.v.	 -,,s   3A
Ac                     UR                  S5      nUSS  n[        R                  " X#5        UR                  SS/5      nUR                  US/   US/   USS  /5      n[        R                  " X#5        g )Nr   r>   r   r}   r   )deleter3   r   _concat_same_type)r   r   r7   r8   s       r   test_deleteBaseMethodsTests.test_delete  sq    Q8
''9aV$))49dA3iab*JK
''9r   c                    USS  R                  SUS   5      n[        R                  " X!5        USS  R                  [        USS  5      * US   5      n[        R                  " X!5        US S R                  SUS   5      n[        R
                  " [        U5      5      nUSS USS & [        U5      S-
  US'   UR                  U5      n[        R                  " X$5        g )Nr>   r   r   r   rL  )insertr3   r   rK   r   aranger   )r   r   r7   takerr8   s        r   test_insertBaseMethodsTests.test_insert  s    abDG,
''5ab#d12h-a9
''5 cr!!!T"X.		#d)$!BKab	t9q=a99U#
''9r   c                    Un[         R                  " [        [        45         UR	                  SU5        S S S 5        [         R                  " [        [        45         UR	                  SU5        S S S 5        [         R                  " [        [        45         UR	                  [        U5      S-
  U5        S S S 5        g ! , (       d  f       N= f! , (       d  f       Ng= f! , (       d  f       g = f)Nr   r   r>   )r#   r   rH  r   r  rK   )r   r   invalid_scalaritems       r   test_insert_invalid$BaseMethodsTests.test_insert_invalid  s    ]]Iz23KK4  4 ]]Iz23KK4  4 ]]Iz23KKD	At, 43 43 43 43s#   CCC#
C
C #
C1c                    [        U5      n[        R                  " [        5         UR	                  US-   US   5        S S S 5        [        R                  " [        5         UR	                  U* S-
  US   5        S S S 5        [        R                  " [
        5         UR	                  SUS   5        S S S 5        g ! , (       d  f       N= f! , (       d  f       NX= f! , (       d  f       g = f)Nr>   r   r   )rK   r#   r   
IndexErrorr  rH  )r   r   ubs      r   test_insert_invalid_loc(BaseMethodsTests.test_insert_invalid_loc  s    Y]]:&KKQQ( ' ]]:&KKaa) ' ]]9%KKT!W% &% '& '& &%s#   B<!CC<
C

C
C,c                    [        U5      R                  US   /[        U5      -  UR                  S9n[        U5      R                  U/[        U5      -  UR                  S9n[        R
                  " XSS9n[        R
                  " XTSS9n[        R
                  " XdSS9nUR                  U5      SL d   eUR                  UR                  5       5      SL d   eUR                  U5      SL d   eUR                  U5      SL d   eUS S R                  US S 5      SL d   eUS S R                  US S 5      SL d   eUR                  S 5      SL d   eUS/   R                  US   5      SL d   eg )Nr   r<   F)	transposeTr}   r   )r%   r   rK   r   r3   box_expectedequalsr  )r   r   rM   r  r   data2r   s          r   test_equalsBaseMethodsTests.test_equals  sr   T
))47)c$i*?tzz)Rt*++XJT,B$**+UtE:e<//'%@
 {{4 D((({{499;'4/// {{5!U***{{7#u,,, BQxtBQx(E111 BQxtBQx(D000 {{4 E)))QCyQ(E111r   c                     [         R                  " U5      R                  [         R                  " U5      5      (       d   eg rf   )r0   r1   r  r  s     r   &test_equals_same_data_different_object7BaseMethodsTests.test_equals_same_data_different_object  s)    yy%%biio6666r   rp   )O__name__
__module____qualname____firstlineno____doc__r   r,   r#   markparametrizer9   rU   r`   rc   rj   rx   r   r   r   r   r   r   r   r   r   rH   r   r   r   r   r   r   r   r   r   r0   r1   rG   r   r   r  r	  r  r  r  r  r"  r   r%  r   __annotations__r:  r?  rI  rN  r^  rb  rl  rq  ru  rx  r|  r  r  r  r  r  r  r   rH  r  r  r  r  r  r\   r  r  __static_attributes__rp   r   r   r   r      sI   /	'% [[Xe}5	1 6	11<1"- [[[4*:;6 <6
161
62 [[X(';<( =( [[X(';<' =' [[#		

5
5& [[RXXirxx/?@Abhhy0@AB	
66
 [[[4-81 91 [[[4-8	1 9	1 [[[4-80 90 [[[4-81 91 [[X	5'9:1 ;1 [[[*CD1 E1 [[V%=>	6 ?	6 [[URYY$<=[[X(<bii'HI* J >*C:C01
+	.8 )5V(<<18
181 [[WtUm4[[
	 1o"6<M8NO"	 5
"$& [[YB00 102 [[
r2h"q"gQFa"a\ABx=Q:	: [[Y(9:: ;:"	:
D6D4)'V [[YAq)(<=* >* [[%Z0Z,VRW-	73		
//::"
-& [[URXXryy",,$GH2 I287r   r   )r&   re  numpyr   r#   pandas._typingr   pandas.core.dtypes.commonr   r   r   pandas.core.dtypes.dtypesr   pandas.core.dtypes.missingr   pandasr0   pandas._testing_testingr3   pandas.core.sortingr	   r   rp   r   r   <module>r     s;          
 3 9   (|7 |7r   