
    MhX                        S SK rS SKrS SKJr  S SKJrJrJrJ	r	J
r
  S SKJr  S SKJrJr  \R"                  S 5       r " S S5      r " S S	5      r " S
 S5      r " S S5      r " S S5      r " S S5      r " S S5      rg)    N)InvalidIndexError)NAIndex
RangeIndexSeries	Timestamp)ArrowExtensionArrayFloatingArrayc                  :    / SQn [        U [        R                  S9$ )N)            l   
        l           l           l           dtype)r   npuint64)larges    \/var/www/html/env/lib/python3.13/site-packages/pandas/tests/indexes/numeric/test_indexing.pyindex_larger      s     DEbii((    c                       \ rS rSrS rS rS rS rS r\	R                  R                  SS/S	/\" S
5      /S//5      S 5       r\	R                  R                  S/ SQ5      S 5       rS rSrg)
TestGetLoc   c                 L    [        / SQ5      nUR                  S5      S:X  d   eg )Nr         r   )r   get_locselfindexs     r   test_get_locTestGetLoc.test_get_loc   s#    i }}Q1$$$r   c                     [        / SQ5      n[        R                  " [        SS9   UR	                  SS/5        S S S 5        g ! , (       d  f       g = f)Nr   z\[1, 2\]matchr   r   )r   pytestraisesr   r   r   s     r   test_get_loc_raises_bad_label(TestGetLoc.test_get_loc_raises_bad_label    s8    i ]],K@MM1a&! A@@s   A
Ac                 X   [        / SQ[        R                  S9n[        R                  " [
        SS9   UR                  S5        S S S 5        [        R                  " [
        SS9   UR                  S5        S S S 5        [        R                  " [
        SS9   UR                  S	5        S S S 5        [        R                  " [
        S
S9   UR                  S5        S S S 5        g ! , (       d  f       N= f! , (       d  f       N= f! , (       d  f       Nd= f! , (       d  f       g = f)Ng              ?       @r   z^'foo'$r#   fooz^1\.5$      ?z^True$Tz^False$F)r   r   float64r%   r&   KeyErrorr   r   idxs     r   test_get_loc_float64TestGetLoc.test_get_loc_float64%   s    O2::6]]895KK 6]]895KK 6]]884KK 5]]895KK 65 65554455s/   C('C9D
D(
C69
D

D
D)c                 t   [        [        R                  SS/[        R                  S9nUR	                  S5      S:X  d   eUR	                  [        R                  5      S:X  d   e[        [        R                  S[        R                  /[        R                  S9nUR	                  S5      S:X  d   eSn[
        R                  " [        US9   UR                  [        R                  5        S S S 5        [        [        R                  S[        R                  [        R                  /[        R                  S9nUR	                  S5      S:X  d   eSn[
        R                  " [        US9   UR                  [        R                  5        S S S 5        g ! , (       d  f       N= f! , (       d  f       g = f)Nr   r   r   r   z7'Cannot get left slice bound for non-unique label: nan'r#   z6'Cannot get left slice bound for non-unique label: nan)	r   r   nanr/   r   r%   r&   r0   
slice_locs)r   r2   msgs      r   test_get_loc_naTestGetLoc.test_get_loc_na1   s'   RVVQN"**5{{1~"""{{266"a'''RVVQ'rzz:{{1~""" H]]83/NN266" 0 RVVQ/rzzB{{1~"""F]]83/NN266" 0/ 0/ 0/s    F/ F)
F&)
F7c                 8   [        SS/[        R                  S9nUR                  S5      S:X  d   e[        R
                  " [        SS9   UR                  S5        S S S 5        [        R
                  " [        SS9   UR                  [        R                  5        S S S 5        [        R
                  " [        S	S9   UR                  [        R                  /5        S S S 5        g ! , (       d  f       N= f! , (       d  f       Nb= f! , (       d  f       g = f)
Nr   r   r   r   z^3$r#      z^nan$z\[nan\])	r   r   r/   r   r%   r&   r0   r6   r   r1   s     r   test_get_loc_missing_nan#TestGetLoc.test_get_loc_missing_nanD   s    QF"**-{{1~"""]]862KKN 3]]873KK 4]],J?KK! @?	 3233??s$   C)> C:?!D)
C7:
D
Dvalsr   r+   z
2019-12-31testc                     [        U5      n[        R                  " [        SS9   UR	                  [
        R                  5        S S S 5        g ! , (       d  f       g = f)Nr6   r#   )r   r%   r&   r0   r   r   r6   )r   r?   r2   s      r   (test_get_loc_float_index_nan_with_method3TestGetLoc.test_get_loc_float_index_nan_with_methodP   s7     Dk]]851KK 211s    A
Ar   )f8i8u8c                     [         R                  " SUS9n[        U5      n[        R                  " [
        SS9   UR                  S 5        S S S 5        g ! , (       d  f       g = f)Ni r   Noner#   )r   aranger   r%   r&   r0   r   )r   r   arrr2   s       r   %test_get_loc_numericindex_none_raises0TestGetLoc.test_get_loc_numericindex_none_raisesW   sC     iiU+Cj]]862KK 322s   A
A#c                    [        / SQ5      n[        R                  " [        R                  5      R                  S-   n[
        R                  " [        [        U5      S9   UR                  U5        S S S 5        [
        R                  " [        [        U5      S9   UR                  R                  U5        S S S 5        g ! , (       d  f       NU= f! , (       d  f       g = f)N)r   r   r   r   r#   )r   r   iinfoint64maxr%   r&   r0   strr   _engine)r   r2   vals      r   test_get_loc_overflows!TestGetLoc.test_get_loc_overflows_   s    Ihhrxx $$q(]]83s84KK 5]]83s84KK$ 54 5444s   !CC
C
C! N)__name__
__module____qualname____firstlineno__r    r'   r3   r9   r=   r%   markparametrizer   rB   rK   rT   __static_attributes__rV   r   r   r   r      s    %"

#&
" [[VqcC59\3J2KfX%VW  X  [[W&89 :	%r   r   c                      \ rS rSrS r\R                  R                  SSS/5      \R                  R                  S\R                  " / SQ\R                  S9S	4\R                  " / SQ\R                  S9S
4\R                  " / SQ\R                  S9S4\R                  " / SQ\R                  S9S4/5      S 5       5       rS r\R                  R                  SS	S/ SQ/ SQ4SS/ SQ/ SQ4SS/ SQ/ SQ4S	S/ SQ/ SQ4SS/ SQ/ SQ4SS/ SQ/ SQ4S	S/ SQ/ SQ4SS/ SQ/ SQ4SS/ SQ/ SQ4S	S/ SQ/ SQ4SS/ SQ/ SQ4SS/ SQ/ SQ4S	S/ SQ/ SQ4SS/ SQ/ SQ4SS/ SQ/ SQ4/5      S 5       r\R                  R                  S\\\\R                  /5      \R                  R                  S \" \" / S!Q/ S"Q/ S#Q// SQ/ SQ/ S$Q/5      5      5      S% 5       5       rS& r\R                  R                  S'S	/ S(Q4S/ S)Q4S/ S*Q4/5      S+ 5       r\R                  R                  S,/ S-Q5      \R                  R                  S.S/S0/5      S1 5       5       r\R                  R                  S./ S2Q5      S3 5       rS4 rS5 rS6 rS7 rS8 r\R                  R                  S9S:S;S<\4\\4/5      S= 5       rS> rS? r\R                  R                  S@S<SA/5      SB 5       r \R                  R                  SCSDSE/5      SF 5       r!SG r"SHr#g)ITestGetIndexerk   c                     [        / SQ5      n[        / SQ5      nUR                  U5      n[        R                  " / SQ[        R                  S9n[
        R                  " X45        g )Nr   r   r<         r   rc      )r   r<   r   )r   get_indexerr   arrayintptmassert_almost_equal)r   index1index2r1e1s        r   test_get_indexerTestGetIndexer.test_get_indexerl   sH    'y!'XXj0
r&r   reverseTFzexpected,method)rg   r   r   r   r   r   padffill)r   r   r   r   r   backfillbfillc                     [        / SQ5      n[        / SQ5      nU(       a  US S S2   nUS S S2   nUR                  XCS9n[        R                  " Xb5        g )Nrb   re   rg   method)r   rh   rk   rl   )r   rs   expectedrz   rm   rn   results          r   test_get_indexer_methods'TestGetIndexer.test_get_indexer_methodst   sV     'y!DbD\F"~H##F#:
v0r   c                 V   [        [        R                  " S5      5      n[        R                  " [
        SS9   UR                  SS/SS9  S S S 5        [        R                  " [
        SS9   UR                  SS/SS9  S S S 5        g ! , (       d  f       NC= f! , (       d  f       g = f)	N
   ztolerance argumentr#   r   r   )	tolerancelimit argument)limitr   r   rI   r%   r&   
ValueErrorrh   r   s     r   test_get_indexer_invalid'TestGetIndexer.test_get_indexer_invalid   s    biim$]]:-ABq!f2 C ]]:-=>q!fA. ?> CB ?>s   B	-B	
B
B(z$method, tolerance, indexer, expectedNr   rd   	   nearestr   皙?g?      !@)r   r      )r   r   r   )r   r   r   r   r   )r   rg   rg   )rg   r   rg   r   r   rg   c                     [        [        R                  " S5      5      nUR                  X1US9n[        R
                  " U[        R                  " U[        R                  S95        g )Nr   rz   r   r   r   r   rI   rh   rk   assert_numpy_array_equalri   rj   )r   rz   r   indexerr{   r   actuals          r   test_get_indexer_nearest'TestGetIndexer.test_get_indexer_nearest   sJ    * biim$""7Y"O
##FBHHXRWW,MNr   listtypeztolerance, expected)333333?r   皙?)r   r   r   )r         ?r   )rg   r   r   c                     [        [        R                  " S5      5      nUR                  / SQSU" U5      S9n[        R
                  " U[        R                  " U[        R                  S95        g )Nr   r   r   r   r   r   )r   r   r{   r   r   r   s         r   +test_get_indexer_nearest_listlike_tolerance:TestGetIndexer.test_get_indexer_nearest_listlike_tolerance   sX     biim$""I)9L # 
 	##FBHHXRWW,MNr   c                 ^   [        [        R                  " S5      5      n[        R                  " [
        SS9   UR                  SS/SSS9  S S S 5        [        R                  " [
        SS9   UR                  SS/S/ S	QS
9  S S S 5        g ! , (       d  f       NF= f! , (       d  f       g = f)Nr   r   r#   r   r   r   )rz   r   ztolerance size must matchr   r   r<   r   r   r   s     r   test_get_indexer_nearest_error-TestGetIndexer.test_get_indexer_nearest_error   s    biim$]]:-=>q!fYa@ ? ]]:-HIq!fY)L JI ?> JIs   B.B
B
B,zmethod,expected)r      r   )r   r   r   )r   r   r   c                 z   [        [        R                  " S5      5      S S S2   nUR                  / SQUS9n[        R
                  " U[        R                  " / SQ[        R                  S95        UR                  / SQUS9n[        R
                  " U[        R                  " U[        R                  S95        g )Nr   rg   r   ry   )r   rc   r   r   r   r   )r   rz   r{   r   r   s        r   #test_get_indexer_nearest_decreasing2TestGetIndexer.test_get_indexer_nearest_decreasing   s    
 biim$TrT*""9V"<
##FBHHYbgg,NO""?6"B
##FBHHXRWW,MNr   	idx_dtype)rO   r/   r   rangerz   rh   get_indexer_non_uniquec                    US:X  a  [        S5      nO[        [        R                  " SUS95      n[        / SQ5      n[	        X15      " U5      n[        R
                  " / SQ[        R                  S9nUS:X  a  [        R                  " XV5        g [        R                  " S[        R                  S9n[        R                  " US   U5        [        R                  " US	   U5        g )
Nr   rc   r   )TFT)rg   rg   rg   rh   r<   r   r   )	r   r   r   rI   getattrri   rj   rk   r   )r   rz   r   numeric_indexotherr|   r{   missings           r   -test_get_indexer_numeric_index_boolean_target<TestGetIndexer.test_get_indexer_numeric_index_boolean_target   s    
 &qMM!"))AY"?@M)*/688L8]"''9ii1G''q	8<''q	7;r   )rt   rv   r   c                 @   [        / SQ5      n[        SS/5      n[        R                  " [        SS9   UR	                  X1S9  S S S 5        [        R                  " [        SS9   UR	                  X!S9  S S S 5        g ! , (       d  f       N@= f! , (       d  f       g = f)Nr   TFzCannot comparer#   ry   )r   r%   r&   	TypeErrorrh   )r   rz   leftrights       r   ,test_get_indexer_with_method_numeric_vs_bool;TestGetIndexer.test_get_indexer_with_method_numeric_vs_bool   s{    YtUm$]]9,<=U2 > ]]9,<=d2 >= >= >=s   A>%B>
B
Bc                    [        / SQ5      n[        SS/5      nUR                  U5      nS[        R                  " [	        U5      [        R
                  S9-  n[        R                  " X45        UR                  U5      nS[        R                  " [	        U5      [        R
                  S9-  n[        R                  " X45        UR                  U5      S   nS[        R                  " [	        U5      [        R
                  S9-  n[        R                  " X45        UR                  U5      S   nS[        R                  " [	        U5      [        R
                  S9-  n[        R                  " X45        g )Nr   TFrg   r   r   )	r   rh   r   oneslenrj   rk   r   r   )r   r   r   resr{   s        r    test_get_indexer_numeric_vs_bool/TestGetIndexer.test_get_indexer_numeric_vs_bool   s   YtUm$u%E
"''::
##C2%D	99
##C2))%03E
"''::
##C2**403D	99
##C2r   c                    [        / SQ[        R                  S9n[        R                  " UR                  U5      [        R                  " / SQ[        R                  S95        / SQn[        R                  " UR                  US5      [        R                  " / SQ[        R                  S95        [        R                  " UR                  US5      [        R                  " / SQ[        R                  S95        [        R                  " UR                  US5      [        R                  " / S	Q[        R                  S95        g )
Nr*   r   r   )gr   g?rt   )rg   r   r   rv   r   )r   r   r   )r   r   r/   rk   r   rh   ri   rj   )r   r2   targets      r   test_get_indexer_float64'TestGetIndexer.test_get_indexer_float64
  s    O2::6
##OOC "((9BGG"D	
 "
##OOFE*BHHZrww,O	
 	##OOFJ/)2771S	
 	##OOFI."''0R	
r   c                    [        SS[        R                  /[        R                  S9R	                  [        R                  /5      n[        R
                  " S/[        R                  S9n[        R                  " X5        g )Nr   r   r   )	r   r   r6   r/   rh   ri   rj   rk   r   )r   r|   r{   s      r   test_get_indexer_nan#TestGetIndexer.test_get_indexer_nan  sQ    1bff~RZZ8DDbffXN88QCrww/
##F5r   c                    [        [        SSS5      [        R                  S9n[        [        R                  " S5      [        R                  S9nUR                  U5      n[        R                  " / SQ[        R                  S9n[        R                  " X45        [        [        R                  " S5      [        R                  S9nUR                  USS9n[        R                  " / S	Q[        R                  S9n[        R                  " X45        [        [        R                  " S5      [        R                  S9nUR                  US
S9n[        R                  " / SQ[        R                  S9n[        R                  " X45        g )Nr      r   r   r   )
r   rg   r   rg   r   rg   r<   rg   rc   rg   rt   ry   )
r   r   r   r   r   r   r<   r<   rc   rc   rv   )
r   r   r   r   r   r<   r<   rc   rc   rd   )
r   r   r   rO   rI   rh   ri   rj   rk   r   )r   r   r   r   r{   s        r   test_get_indexer_int64%TestGetIndexer.test_get_indexer_int64!  s   eAr1oRXX6ryy}BHH5##F+88?rwwO
##G6ryy}BHH5##F5#988:"''J
##G6ryy}BHH5##F:#>88:"''J
##G6r   c                     [        [        R                  " S5      R                  S5      S-  S-   5      nUR	                  U5      n[        R
                  " / SQ[        R                  S9n[        R                  " X45        [        [        R                  " S5      R                  S5      S-  S-   5      nUR	                  USS9n[        R
                  " / S	Q[        R                  S9n[        R                  " X45        [        [        R                  " S5      R                  S5      S-  S-   5      nUR	                  US
S9n[        R
                  " / SQ[        R                  S9n[        R                  " X45        g )Nr   r   rd   r   )
r   rg   r   r   r<   rc   rg   rg   rg   rg   r   rt   ry   )
r   r   r   r   r<   rc   rc   rc   rc   rc   rv   )
r   r   r   r   r<   rc   rg   rg   rg   rg   )	r   r   rI   astyperh   ri   rj   rk   r   )r   r   r   r   r{   s        r   test_get_indexer_uint64&TestGetIndexer.test_get_indexer_uint642  s   ryy}++H59EAB))&188?rwwO
##G6ryy}++H59EAB))&)?88:"''J
##G6ryy}++H59EAB))&)D88>bggN
##G6r   z	val, val2)rc   rd   )rc   rc   rc   c                     [        SSSX/US9nUR                  S5      nUS:X  d   e[        R                  " [        SS9   UR                  S5        S S S 5        g ! , (       d  f       g = f)Nr   r   r<   r   9r#   r   )r   r   r%   r&   r0   )r   rS   val2any_numeric_ea_and_arrow_dtyper2   r|   s         r   test_get_loc_masked"TestGetIndexer.test_get_loc_maskedB  sX     Q1c(0NOQ{{]]83/KKN 0//s   A
A*c                    [        SS[        /US9nUR                  [        5      nUS:X  d   e[        SS[        [        /US9nUR                  [        5      n[        R                  " U[
        R                  " / SQ5      5        [        / SQUS9n[        R                  " [        SS9   UR                  [        5        S S S 5        g ! , (       d  f       g = f)Nr   r   r   )FFTTr   r   r#   )
r   r   r   rk   r   r   ri   r%   r&   r0   )r   r   r2   r|   s       r   test_get_loc_masked_na%TestGetIndexer.test_get_loc_masked_naL  s    Q2J&DER{{Q2rN*HIR
##FBHH5O,PQI%CD]]840KKO 100s   ,C
Cc           
          [        [        [        R                  " SSS[        R                  /5      [        R                  " / SQ5      S95      nUR                  [        5      nUS:X  d   eUR                  [        R                  5      nUS:X  d   e[        [        [        R                  " / SQ5      [        R                  " / SQ5      S95      nUR                  [        5      nUS:X  d   e[        R                  " [        SS	9   UR                  [        R                  5        S S S 5        [        [        [        R                  " SS[        R                  /5      [        R                  " / S
Q5      S95      nUR                  [        R                  5      nUS:X  d   e[        R                  " [        SS	9   UR                  [        5        S S S 5        g ! , (       d  f       N= f! , (       d  f       g = f)Nr   r   )FFTF)maskr<   )r   r   r+   )FFTr6   r#   )FFFr   )
r   r
   r   ri   r6   r   r   r%   r&   r0   )r   r2   r|   s      r   test_get_loc_masked_na_and_nan-TestGetIndexer.test_get_loc_masked_na_and_nanZ  s`   !Q266*+"((;V2W

 R{{RVV${{"((;/bhh?S6TU
 R{{]]851KK 2 !Q(rxx8M/N

 RVV${{]]840KKO 10 21 10s   	 G?G/
G,/
G=rS   r   c                     [        SS[        SU/US9nUR                  S[        S/5      n[        R                  " / SQ5      n[
        R                  " XESS9  g )	Nr   r   r<   r   rd   r   F)check_dtype)r   r   get_indexer_forr   ri   rk   r   )r   r   rS   r2   r|   r{   s         r   test_get_indexer_masked_na)TestGetIndexer.test_get_indexer_masked_nax  sP     Q2q#&.LM$$aQZ088J'
##F%Hr   r   booleanbool[pyarrow]c                     US:X  a  [         R                  " S5        [        SS[        /US9nUR	                  S5      nUS:X  d   eUR	                  [        5      nUS:X  d   eg )Nr   pyarrowTFr   r   r   )r%   importorskipr   r   r   )r   r   r2   r|   s       r   "test_get_indexer_masked_na_boolean1TestGetIndexer.test_get_indexer_masked_na_boolean  s`     O#	*T5"%U3U#{{R{{r   c                    [         R                  " S5      n[        [        UR	                  SS/UR                  UR                  5       UR                  5       5      S95      5      n[        S/5      nUR                  U5      n[        R                  " SS/[        R                  S9n[        R                  " XE5        UR                  U5      u  pg[        R                  " SS/[        R                  S9[        R                  " S/[        R                  S9p[        R                  " Xh5        [        R                  " Xy5        g )Nr   r   r   )typer   rg   r   )r%   r   r   r	   ri   
dictionaryint8rh   r   rO   rk   r   r   )
r   par   r2   r|   r{   result_1result_2
expected_1
expected_2s
             r   (test_get_indexer_arrow_dictionary_target7TestGetIndexer.test_get_indexer_arrow_dictionary_target  s      +!QbmmBGGIrwwy&IJ

 QCj(88QG2884
##F5 77?!#1b'!BBHHCrxxE
J 	##H9
##H9r   rV   )$rW   rX   rY   rZ   rq   r%   r[   r\   r   ri   rj   r}   r   r   listtupler   zipr   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r]   rV   r   r   r_   r_   k   sp   ' [[Yu6[[XX&bgg6>XX&bgg6@XXoRWW5zBXXoRWW5w?		
	1 7	1/ [[.D)Y/y)4i3Ay),Iy19i0D/95	:oy9A	2OY7?I6C+6o{;_j9	
(O)(O [[Z$vrxx)HI[[ /?C[*5	
O JOM [[
	j)4y)6LMO	O [[[*QR[[X7O'PQ< R S<& [[X'EF3 G33(
"67"7  [[[66Ar7RH*MN O< [[UQF+I ,I [[Wy/&BC D:r   r_   c                       \ rS rSr\R
                  R                  S\" \R                  " SSS95      \" \
" SSS5      \R                  S9\" \R                  " SS	S95      /5      S
 5       rS rS rSrg)	TestWherei  r   rd   r/   r   r   r   r   r   c                 H   S/[        U5      -  nUnUR                  U" U5      5      nS/S/[        U5      S-
  -  -   n[        UR                  /USS  R	                  5       -   [
        R                  S9nUR                  U" U5      5      n[        R                  " XT5        g )NTFr   r   )	r   wherer   	_na_valuetolistr   r/   rk   assert_index_equal)r   listlike_boxr   condr{   r|   s         r   
test_whereTestWhere.test_where  s     vE
"\$/0w$3u:>22%//*U12Y-=-=-??rzzR\$/0
f/r   c                    [        / SQ[        R                  S9n[        R                  " / SQ5      n[        R                  " S/[        R                  S9n[        / SQ[        R                  S9nUR                  X#5      n[        R                  " XT5        UR                  U) U5      n[        R                  " XT5        g )N)r   rf   r   r   )FTFr   )r   rf   r   )	r   r   r   ri   rO   r   rk   r   putmask)r   r2   r   r   r{   r|   s         r   test_where_uint64TestWhere.test_where_uint64  s    IRYY/xx,-!BHH-"))44'
f/dUE*
f/r   c                     [        S[        R                  /5      nUR                  5       n[        SS/SS9n[        SS/5      nUR	                  X#5      n[
        R                  " XT5        g )Nr   abstringr   r+   )r   r   r6   notnar   rk   r   )r   r   r   r   r{   r|   s         r   Ctest_where_infers_type_instead_of_trying_to_convert_string_to_floatMTestWhere.test_where_infers_type_instead_of_trying_to_convert_string_to_float  sY    q"&&k"{{}sCj1#s$T)
f/r   rV   N)rW   rX   rY   rZ   r%   r[   r\   r   r   rI   r   rO   r   r   r  r]   rV   r   r   r   r     sq    [["))AY/0%2q/2"))AX./	
000	0r   r   c                      \ rS rSr\R
                  R                  S\R                  \R                  \R                  /5      S 5       rS r\R
                  R                  S\R                  \R                  /5      S 5       rSrg)	TestTakei  r   c                 |    [        / SQUSS9nUR                  / SQ5      nUR                  UR                  :X  d   eg )N)r   r   r<   rc   r-   r   name)r<   r   r   )r   taker  )r   r   r   takens       r   test_take_preserve_name TestTake.test_take_preserve_name  s4    l)%@

9%zzUZZ'''r   c                    [        / SQS[        R                  S9nUR                  [        R                  " / SQ5      5      n[        / SQ[        R                  SS9n[
        R                  " X#5        UR                  [        R                  " / SQ5      SS9n[        S	S
[        R                  /[        R                  SS9n[
        R                  " X#5        UR                  [        R                  " / SQ5      SSS9n[        / SQ[        R                  SS9n[
        R                  " X#5        Sn[        R                  " [        US9   UR                  [        R                  " / SQ5      SS9  S S S 5        [        R                  " [        US9   UR                  [        R                  " / SQ5      SS9  S S S 5        Sn[        R                  " [        US9   UR                  [        R                  " SS/5      5        S S S 5        g ! , (       d  f       N= f! , (       d  f       Nk= f! , (       d  f       g = f)N)r+   r,         @xxx)r  r   r   r   rg   )r,   r+   r  r  T
fill_valuer,   r+   F
allow_fillr  zJWhen allow_fill=True and fill_value is not None, all indices must be >= -1r#   r   r   r   r   3index -5 is out of bounds for (axis 0 with )?size 3r   r  )r   r   r/   r  ri   rk   r   r6   r%   r&   r   
IndexError)r   r2   r|   r{   r8   s        r   test_take_fill_value_float64%TestTake.test_take_fill_value_float64  s   O%rzzB"((:./

G
f/ "((:.4@#sBFF+2::EJ
f/ "((:.5TR

G
f/( 	 ]]:S1HHRXXj)dH; 2]]:S1HHRXXj)dH; 2 D]]:S1HHRXXq"g&' 21 2111 21s$   'H'H!(H2
H!
H/2
I r   c                 L   [        / SQUSS9nUR                  [        R                  " / SQ5      5      n[        / SQUSS9n[        R
                  " X45        [        U5      R                  nSU S3n[        R                  " [        US9   UR                  [        R                  " / SQ5      S	S
9  S S S 5        UR                  [        R                  " / SQ5      SS	S9n[        / SQUSS9n[        R
                  " X45        [        R                  " [        US9   UR                  [        R                  " / SQ5      S	S
9  S S S 5        [        R                  " [        US9   UR                  [        R                  " / SQ5      S	S
9  S S S 5        Sn[        R                  " [        US9   UR                  [        R                  " SS/5      5        S S S 5        g ! , (       d  f       GN5= f! , (       d  f       N= f! , (       d  f       N}= f! , (       d  f       g = f)Nr   r  r  r  )r   r   r<   zUnable to fill values because z cannot contain NAr#   Tr  Fr  r  r  r  r   r  )r   r  r   ri   rk   r   r   rW   r%   r&   r   r  )r   r   r2   r|   r{   r  r8   s          r   test_take_fill_value_ints"TestTake.test_take_fill_value_ints  s}    IU7"((:./%e<
f/Cy!!.tf4FG ]]:S1HHRXXj)dH; 2 "((:.5TR%e<
f/]]:S1HHRXXj)dH; 2]]:S1HHRXXj)dH; 2 D]]:S1HHRXXq"g&' 21 21 2111 21s0   'G!'G3&'H0(H!
G03
H
H
H#rV   N)rW   rX   rY   rZ   r%   r[   r\   r   r/   rO   r   r  r   r#  r]   rV   r   r   r
  r
    sp    [[[2::rxx*KL( M(
(< [[Wrxx&;<( =(r   r
  c                       \ rS rSr\R
                  R                  S\R                  \R                  \R                  /5      S 5       rS rS rSrg)TestContainsi  r   c                 ,    [        / SQUS9nS U;  d   eg )N)r   r   r   r<   rc   r   )r   )r   r   r   s      r   test_contains_noneTestContains.test_contains_none  s     oU35   r   c                     [        SS[        R                  /[        R                  S9n[        R                  U;   d   eg Nr+   r,   r   r   r   r6   r/   r   s     r   test_contains_float64_nans'TestContains.test_contains_float64_nans  s-    sC(

;vvr   c                 f    [        SS[        R                  /[        R                  S9nSU;   d   eg r+  r,  r   s     r   test_contains_float64_not_nans+TestContains.test_contains_float64_not_nans  s)    sC(

;e||r   rV   N)rW   rX   rY   rZ   r%   r[   r\   r   r/   rO   r   r(  r-  r0  r]   rV   r   r   r&  r&    sC    [[Wrzz288RYY&GH! I!
r   r&  c                       \ rS rSr\R
                  R                  S\\/5      S 5       r	\R
                  R                  S\\/5      S 5       r
\R
                  R                  S\\/5      S 5       rS rS rSrg	)
TestSliceLocsi  r   c                    [        [        R                  " / SQUS95      n[        U5      nUR	                  SS9SU4:X  d   eUR	                  SS9SU4:X  d   eUR	                  SS5      S:X  d   eUR	                  SS	5      SU4:X  d   eUR	                  SS
9S:X  d   eUR	                  SS
9S:X  d   eUS S S2   nUR	                  SS5      S:X  d   eUR	                  SS5      S:X  d   eg )Nr   r   r   rd   rf   r   r   r   r   r   startr<   r   )r<   rf   rd   r   end)r   rf   r   )r   r   rg   r   rf   r   )r   rd   r   r   ri   r   r7   r   r   r   nrn   s        r   test_slice_locsTestSliceLocs.test_slice_locs   s   bhh8FGJa(QF222a(QF2221%///2&1a&000A&&000A&&000 tt  A&&000  A&&000r   c                 2   [        [        R                  " / SQUS95      n[        U5      nUR	                  SS5      SU4:X  d   eUR	                  SS5      S:X  d   eUS S S	2   nUR	                  S
S5      S:X  d   eUR	                  SS	5      SU4:X  d   eg )Nr5  r   g      @g      $@r<   g      @g      %@)r<   r   rg   r   r.   r:  r   r;  r<  s        r   test_slice_locs_float_locs(TestSliceLocs.test_slice_locs_float_locs1  s    bhh8FGJT*q!f444T*f444tt  c*f444  r*q!f444r   c                    [        [        R                  " / SQUS95      nUR                  SS5      S:X  d   eUR                  SS5      S:X  d   eUS S S2   nUR                  SS5      S:X  d   eUR                  SS5      S:X  d   eg )N)r      rD     r   rD  r   r<         rg   )r   r   ri   r7   )r   r   r   rn   s       r   test_slice_locs_dup_numeric)TestSliceLocs.test_slice_locs_dup_numeric<  s    bhh/u=>B'6111B'6111tt  R(F222  R(F222r   c                 V   [        [        R                  SS/5      nUR                  S5      S:X  d   eUR                  [        R                  5      S:X  d   e[        S[        R                  [        R                  SS/5      nUR                  [        R                  5      S:X  d   eg )Nr   r   rF  )r   r<   r   )r   rd   )r   r   r6   r7   r   s     r   test_slice_locs_na TestSliceLocs.test_slice_locs_naF  s    rvvq!n%"f,,,'6111q"&&"&&!Q/0'6111r   c                 D   [        [        R                  SS/5      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    r#   r.   r6  r8  )r   r   r6   r%   r&   r0   r7   r   s     r   test_slice_locs_na_raises'TestSliceLocs.test_slice_locs_na_raisesN  su    rvvq!n%]]82.3' / ]]82.% /. /. /.s   B 'B 
B
BrV   N)rW   rX   rY   rZ   r%   r[   r\   intfloatr>  rA  rI  rL  rP  r]   rV   r   r   r3  r3    s    [[WsEl31 41  [[WsEl35 45 [[WsEl33 432&r   r3  c                       \ rS rSr\R
                  R                  SSS/5      S 5       r\R
                  R                  SSS/5      \R
                  R                  S	S
S/5      S 5       5       rSr	g)TestGetSliceBoundsiW  zside, expected)r   rc   )r   rd   c                 Z    [        [        S5      5      nUR                  SUS9nXB:X  d   eg )Nrf   rc   sider   r   get_slice_bound)r   rX  r{   r   r|   s        r   test_get_slice_bounds_within/TestGetSliceBounds.test_get_slice_bounds_withinX  s0    eAh&&qt&4!!!r   rX  r   r   zbound, expected)rg   r   )r   rf   c                 X    [        [        S5      5      nUR                  X1S9nXR:X  d   eg )Nrf   rW  rY  )r   rX  r{   boundr   r|   s         r   test_get_slice_bounds_outside0TestGetSliceBounds.test_get_slice_bounds_outside^  s0     eAh&&u&8!!!r   rV   N)
rW   rX   rY   rZ   r%   r[   r\   r[  r_  r]   rV   r   r   rU  rU  W  sv    [[-\/JK" L"
 [[Vfg%67[[.'0BC" D 8"r   rU  )numpyr   r%   pandas.errorsr   pandasr   r   r   r   r   pandas._testing_testingrk   pandas.core.arraysr	   r
   fixturer   r   r_   r   r
  r&  r3  rU  rV   r   r   <module>rh     s      +   ) )M% M%`r: r:j	)0 )0X@( @(F  5& 5&p" "r   