
    Mhch                         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  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)NumpyEADtype)na_value_for_dtype)nargsortc                   D   \ 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SSSSS\R.                  4S S\R.                  4S!S"/5      S# 5       rS$ r\R                  R                  S%S&\R4                  " / S'Q\R6                  " S(5      S)94S*\R4                  " / S+Q\R6                  " 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      S0 5       r\R                  R                  S1/ S2Q5      S3 5       r \R                  R                  S4\!RD                  S5 /5      \R                  R                  SS6 \!RF                  /5      S7 5       5       r$S8 r%S9 r&S: r'S; r(S< r)S= r*\+" S>5      r,\-\.S?'   S@ r/SA r0SB r1\R                  R                  SCSS/5      \R                  R                  SDSE/ SFQ4SG/ SHQ4SI/ SJQ4/5      SK 5       5       r2SL r3\R                  R                  SMSNSE/5      SO 5       r4\R                  R                  SDSPSQSQ//SQSNSQ//SGSGSN//SNSQSG//SRSQSQ///5      SS 5       r5\R                  R                  SM/ STQ5      SU 5       r6SV r7SW r8SX r9SY r:SZ r;S[ r<S\ r=\R                  R                  S]SGSNSI/ S^Q/5      S_ 5       r>\R                  R                  S`SISaSN0\?Sa4SQ0 \?Sb4SNSI/0 \?Sc4SISdSe0\@Sf4/5      Sg 5       rASh rBSi rCSj rDSk rE\R                  R                  S4\!R4                  \!RD                  \!R                  /5      Sl 5       rGSm rHSnrIg)o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       U/var/www/html/env/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                 8   US S n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(   s~    CR=mmo-.EE8$111@KKM99U#000?JJL
v0r   c                    US S 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                  S
S5      S:X  d)  [        UR                  [        R                  5      (       a  UR                  S5      nOd[        UR                  S
S5      S:X  a  UR                  S5      nO7[!        UR                  5      [        R"                  L a  UR                  S5      n[$        R&                  " XE5        g )Nr-   r   T)	normalize   
proportion)indexnameg        r   storage pyarrowzdouble[pyarrow]pyarrow_numpyfloat64Float64)uniquer   arrayr.   r/   r0   r   r   r1   
isinstanceCategoricallenr?   getattr
ArrowDtypeastyper   NAr2   r3   )r   r   valuesserr6   r7   s         r    test_value_counts_with_normalize1BaseMethodsTests.test_value_counts_with_normalize5   sZ   CRy!$		|,-iiJJ/!!D!1<<>$//yyS[!CK/v||,H yyFLL|LH#$s6{?HaZ 4::y"-:jJJ?
 ?
  '89HTZZB/?By1H

+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=   )r/   	DataFramecountr0   r2   r3   )r   data_missingdfr6   r7   s        r   
test_countBaseMethodsTests.test_countS   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=   )r/   r0   rZ   )r   r[   rQ   r6   r7   s        r   test_series_count"BaseMethodsTests.test_series_countY   s,    ii%!!!r   c                     [         R                  " U5      R                  [        5      n[	        U[         R                  5      (       d   eg N)r/   r0   applyidrI   r   r   r6   s      r   test_apply_simple_series)BaseMethodsTests.test_apply_simple_series`   s1    4&&r*&")),,,,r   	na_actionNignorec                 r    UR                  S US9nUR                  5       n[        R                  " X45        g )Nc                     U $ rc    xs    r   <lambda>+BaseMethodsTests.test_map.<locals>.<lambda>f   s    Ar   )ri   )mapto_numpyr2   assert_numpy_array_equal)r   r[   ri   r6   r7   s        r   test_mapBaseMethodsTests.test_mapd   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;   )r/   r0   argsortr   rH   intpr2   r3   )r   data_for_sortingr6   r7   s       r   test_argsortBaseMethodsTests.test_argsortj   sE    +,44699RXXirww?@
v0r   c                     UR                  5       n[        R                  " / SQ[        R                  S9n[        R
                  " X#5        g rx   )r{   r   rH   r|   r2   rt   )r   data_missing_for_sortingr6   r7   s       r   test_argsort_missing_array+BaseMethodsTests.test_argsort_missing_arrayp   s2    )11388IRWW5
##F5r   c                 V   Sn[         R                  " [        US9   [        R                  " U5      R                  5       nS S S 5        [        R                  " [        R                  " / SQ[        R                  S95      n[         R                  " WU5        g ! , (       d  f       N_= f)Nz;The behavior of Series.argsort in the presence of NA valuesmatch)r=   r   r;   )
r2   assert_produces_warningFutureWarningr/   r0   r{   r   rH   r|   r3   )r   r   msgr6   r7   s        r   test_argsort_missing%BaseMethodsTests.test_argsort_missingv   sl    K''SAYY78@@BF B99RXXj@A
vx0 BAs   %B
B(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   rz   )rz   r   r   r=   r=   rz   )r   _is_booleanargmaxargmintake)r   r}   r   na_value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 getr   r   )r!   raises
ValueErrorrL   )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   rL   )r   r   r   r   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   )idxminTrz   )r   Tr   )r   Trz   r   r   )r   Fr   )r   Fr   c                    S nSnUR                  S5      (       a  US:X  a  [        nUR                  S5      (       a&  [        R                  " U5      (       a  [        nSU 3n[        R
                  " U5      n[        R                  " XVS9   [        Xr5      " US9nS S S 5        [        R                  " WU5        g ! , (       d  f       N&= f)Nz$The behavior of Series.argmax/argminargr   idxzThe behavior of Series.r   skipna)

startswithr   r   isnanr/   r0   r2   r   rL   assert_almost_equal)	r   r   op_namer   r7   warnr   rQ   r6   s	            r   test_argreduce_series&BaseMethodsTests.test_argreduce_series   s    " 4e$$R De$$();); D+G95Cii01''8S*&9F 9
vx0 98s   B==
C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)NrB   r   Fr   )r!   r   NotImplementedErrorr   r   )r   r   r   s      r   +test_argmax_argmin_no_skipna_notimplemented<BaseMethodsTests.test_argmax_argmin_no_skipna_notimplemented   sb    ']].b9KKuK% : ]].b9KKuK% :9 :9 :9s   A&A7&
A47
Bzna_position, expectedlastry   r|   r;   firstr=   rz   r   c                 B    [        XS9n[        R                  " XC5        g )N)na_position)r   r2   rt   )r   r   r   r7   r6   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   keyry   rz   )r   r=   rz   )r=   r   rz   )r/   r0   sort_valuesilocnuniquer2   r3   )r   r}   r   sort_by_keyrQ   r6   r7   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   ry   )r   rz   r=   )r/   r0   r   r   r2   r3   )r   r   r   r   rQ   r6   r7   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=   rz   r=   rU   BrU   r   )r=   r=   rz   ry   r?   )r/   rY   r   r   r2   assert_frame_equal)r   r}   r   r\   r6   r7   s         r   test_sort_values_frame'BaseMethodsTests.test_sort_values_frame   s\    \\	0@ABc
+<<"2"7"7	"BC9
 	f/r   keep)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   r2   rt   )r   r   r   arrr6   r7   s         r   test_duplicated BaseMethodsTests.test_duplicated   si    ii%T*7?xx :;HV^xx :;Hxx 89H
##F5r   boxc                     U $ rc   rm   rn   s    r   rp   BaseMethodsTests.<lambda>  s    !r   c                 "    U R                  5       $ rc   )rG   rn   s    r   rp   r     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   r6   s         r   test_uniqueBaseMethodsTests.test_unique  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   rz   )r   r      )r/   	factorizer   r   r   rH   r|   r   r2   rt   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   )r/   r   r2   rt   r   rK   rG   r   )r   r   codes_1	uniques_1codes_2	uniques_2s         r   test_factorize_equivalence+BaseMethodsTests.test_factorize_equivalence!  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;   )r/   r   r   rH   r|   r#   r   r   r2   rt   r   )r   r   r   r   r   r   s         r   test_factorize_empty%BaseMethodsTests.test_factorize_empty*  sl    d2Ah/"BGG4:44RtBQx~~4N
##E:
''Br   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=   rU   )r   r   )r   r/   rY   copyr   fillnar2   r   )r   r[   r   r\   df_orig
filled_valr6   s          r   test_fillna_copy_frame'BaseMethodsTests.test_fillna_copy_frame2  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   r/   r0   r   r   r   r2   r3   )r   r[   r   rQ   ser_origr   r6   s          r   test_fillna_copy_series(BaseMethodsTests.test_fillna_copy_series>  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[   r   s      r   test_fillna_length_mismatch,BaseMethodsTests.test_fillna_length_mismatchI  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      5       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 )Nrz   c                 
    X:*  $ rc   rm   x1x2s     r   rp   2BaseMethodsTests.test_combine_le.<locals>.<lambda>W  s    rxr   r;   r   c                 
    X:*  $ rc   rm   r  s     r   rp   r  a  s    r   )
r/   r0   combinerH   ziplistr  r2   r3   r   )r   data_repeated
orig_data1
orig_data2s1s2r6   abr7   vals              r   test_combine_le BaseMethodsTests.test_combine_leQ  s    "/q!1
YYz"YYz"B 7899HH&)$z*:D<L&MN&MFQ&MN55
 	v0ggajC!8999HH#'
#34#3ac#3455
 	v0 O 5s   EEc                    U" S5      u  p#[         R                  " U5      n[         R                  " U5      n [        R                  " SS9   [         R                  " UR	                  [        [        U5      [        U5      5       VVs/ s H	  u  pgXg-   PM     snn5      5      nS S S 5        UR                  US 5      n	[        R                  " U	W5        UR                  S   n
UR                  U
S 5      n	[         R                  " UR	                  [        U5       Vs/ s H  ofU
-   PM	     sn5      5      n[        R                  " X5        g s  snnf ! , (       d  f       N= f! [         aK    [        R                  " [        5         UR                  US 5        S S S 5        O! , (       d  f       O= f g f = fs  snf )Nrz   rj   )overc                 
    X-   $ rc   rm   r  s     r   rp   3BaseMethodsTests.test_combine_add.<locals>.<lambda>}  s    bgr   c                 
    X-   $ rc   rm   r  s     r   rp   r     s    rwr   r   c                 
    X-   $ rc   rm   r  s     r   rp   r     s    r   )r/   r0   r   errstater   r  r  	TypeErrorr!   r   r  r2   r3   r   )r   r  r  r  r  r  r  r  r7   r6   r  s              r   test_combine_add!BaseMethodsTests.test_combine_addj  s[   !.q!1
YYz"YYz"	(+99---0j1A4
CS-TU-T6A-TU , B 67
vx0ggajC!7899%%Z8H&I8H13w8H&IJ
 	v0% V ,+  	 y)

256 *))	 'Js_   E( =E	EE%E( !G E
E%!E( %E( ($F=F) 	F=)
F7	3F=<F=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   rz      )rz   r   r   r   )r/   r0   combine_firstr2   r3   )r   r   r  r  r6   r7   s         r   test_combine_first#BaseMethodsTests.test_combine_first  s\    IId2AhIId1Qiy1#99T"1X&
v0r   framezperiods, indices)rz   r   r   r   r   r   )r   r=   rz   r   r   rz   )r   r   r   r=   rz   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 )
Nr(  rU   r@   T
allow_fillr=   )r   r   rW   )
r/   r0   r   to_frameassignshiftconcatr2   r   r3   )	r   r   r,  periodsindicessubsetr7   r6   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=   r4  rf   s      r   test_shift_0_periods%BaseMethodsTests.test_shift_0_periods  sH    AAw$q'!!!q'QayF1I%%%r   r6  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)Nr(  z does not support diffg      ?r   r=   r   )r   r   operatorxorsub	Exceptionr!   r"   r#   r/   r0   diffr4  r2   r3   rY   r   nanr   )	r   r   r6  opsr6   r7   r\   r  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 )Nrz   Tr0  r4  r   r2   r   )r   r   r6  r7  r8  r6   r7   s          r   test_shift_non_empty_array+BaseMethodsTests.test_shift_non_empty_array  s=     bqg&;;w4;8
''9r   )rK  r   r   r=   r   c                 `    US S nUR                  U5      nUn[        R                  " XE5        g )Nr   )r4  r2   r   )r   r   r6  emptyr6   r7   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   rz   r=  rf   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=   rz   r-  )rz   r   r   r   rM  )r   r   r   rX  r6   r7   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   r$  hashr   r   s     r   test_not_hashable"BaseMethodsTests.test_not_hashable  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 rc   )r/   r0   r2  utilhash_pandas_objectr2   assert_equal)r   r   as_framer  r  s        r   test_hash_pandas_object_works.BaseMethodsTests.test_hash_pandas_object_works  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 )Nry   r   rightsider=   rz   r   r;   r   sorter)r   r   _test_searchsorted_bool_dtypesr   r/   r0   searchsortedr   rH   r|   r2   rt   )
r   r}   	as_seriesr  cr  r   r6   r7   rl  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/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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   rh  ri  r=   rz   rk  )r   r/   rH   r#   r   r0   rn  r   r   r|   r2   rt   )
r   r}   ro  r   r  r  r   r6   r7   rl  s
             r   rm  /BaseMethodsTests._test_searchsorted_bool_dtypes  s4    !&&88T5M?#$33QF;))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=   rz   r;   r   r  r/  r   T)inplace)TFTT)r#   r/   r0   r   r   r   r   rH   r2  reshapewherer2   rc  maskrY   )r   r   r   rd  clsr  r  origrQ   condr6   r7   r5   s                r   test_where_series"BaseMethodsTests.test_where_series3  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=   rz   r   c                 l   [        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[        XR5       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;   r   )r#   r   r   r/   r0   r   repeatrI   intr  ranger?   r2   rc  )r   r   r  ro  	use_numpyr   r6   ro   n_r7   s              r   test_repeatBaseMethodsTests.test_repeat^  s    4j''Ra

'C))C.C,53(3::g;N#-gs#;#;7)a-"%c"3F"3$!U1XAXA"3F:,,XZZ,Hyy1A1A'1JKH
) Gs    D/zrepeats, kwargs, error, msgrX   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_raisesn  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   rz   r   )deleter2   r   _concat_same_type)r   r   r6   r7   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   r(  )insertr2   r   rK   r   aranger   )r   r   r6   takerr7   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   r$  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   g      ?)rK   r!   r   
IndexErrorr  r$  )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)	transposeTrz   r   )r#   r   rK   r   r2   box_expectedequalsr   )r   r   r   ro  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 rc   )r/   r0   r  r]  s     r   &test_equals_same_data_different_object7BaseMethodsTests.test_equals_same_data_different_object  s)    yy%%biio6666r   rm   )J__name__
__module____qualname____firstlineno____doc__r   r*   r!   markparametrizer8   rR   r]   r`   rg   ru   r~   r   r   r   r   r   r   rF  r   r   rH   r   r   r   r   r   r   r/   r0   rG   r   r   r   r   r   r  r  r   r  r   __annotations__r  r%  r*  r:  r>  rI  rN  rR  rU  rY  r^  re  rq  rm  r}  r  r   r$  r  r  r  r  r  rY   r  r  __static_attributes__rm   r   r   r	   r	      s   /	'% [[Xe}5
1 6
11<1"- [[[4*:;6 <6
16162 [[X(';<( =( [[X(';<' =' [[#ubff%ubff%!!		
11 & [[RXXirxx/?@Abhhy0@AB	
66
 [[[4-81 91 [[[4-8	1 9	1 [[[4-80 90 [[V%=>	6 ?	6 [[URYY$<=[[X(<bii'HI* J >*C:C
+	.8 )5V(<<121@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$   rA  numpyr   r!   pandas._typingr   pandas.core.dtypes.commonr   pandas.core.dtypes.dtypesr   pandas.core.dtypes.missingr   pandasr/   pandas._testing_testingr2   pandas.core.sortingr   r	   rm   r   r   <module>r     s2          3 2 9   (~
7 ~
7r   