
    A>i[                         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   " 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                       \ 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     i/var/www/html/land-tabula/venv/lib/python3.13/site-packages/pandas/tests/indexes/numeric/test_indexing.pytest_get_locTestGetLoc.test_get_loc   s#    i }}Q1$$$    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              ?       @dtypez^'foo'$r   fooz^1\.5$      ?z^True$Tz^False$F)r   np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   r+   
slice_locs)r   r-   msgs      r   test_get_loc_naTestGetLoc.test_get_loc_na*   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   r+   r1   r   r,   s     r   test_get_loc_missing_nan#TestGetLoc.test_get_loc_missing_nan=   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)Nr1   r   )r   r   r   r+   r   r)   r1   )r   r:   r-   s      r   (test_get_loc_float_index_nan_with_method3TestGetLoc.test_get_loc_float_index_nan_with_methodI   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   r+   r   )r   r&   arrr-   s       r   %test_get_loc_numericindex_none_raises0TestGetLoc.test_get_loc_numericindex_none_raisesP   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   r+   strr   _engine)r   r-   vals      r   test_get_loc_overflows!TestGetLoc.test_get_loc_overflowsX   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   r.   r4   r8   r   markparametrizer   r=   rF   rO   __static_attributes__rQ   r   r   r   r      s    %"

#&
" [[VqcC59\3J2KfX%VW  X  [[W&89 :	%r   r   c                   6   \ rS rSrS r\R                  R                  SSS/5      \R                  R                  S/ SQS4/ SQS	4/ S
QS4/ S
Q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      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.S//5      S0 5       5       r\R                  R                  S-/ S1Q5      S2 5       rS3 rS4 rS5 rS6 rS7 r\R                  R                  S8S9S:S;\4\\4/5      S< 5       rS= rS> r\R                  R                  S?S;S@/5      SA 5       r\R                  R                  SBSCSD/5      SE 5       rSF r SGr!g)HTestGetIndexerd   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   r7         r   r^      )r   r7   r%   )r   get_indexerr)   arrayintptmassert_almost_equal)r   index1index2r1e1s        r   test_get_indexerTestGetIndexer.test_get_indexere   sH    'y!'XXj0
r&r   reverseTFzexpected,method)rb   r   r   r   r   padffill)r   r   r   r   r   backfillbfillc                     [        / SQ5      n[        / SQ5      n[        R                  " U[        R                  S9nU(       a  US S S2   nUS S S2   nUR	                  XCS9n[
        R                  " Xb5        g )Nr]   r`   r%   rb   method)r   r)   rd   re   rc   rf   rg   )r   rn   expectedru   rh   ri   results          r   test_get_indexer_methods'TestGetIndexer.test_get_indexer_methodsm   sj     'y!88HBGG4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)   rD   r   r   
ValueErrorrc   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   r_   	   nearestr   皙?g?      !@)r   r      )r   r   r   )r   r   r   r   r   )r   rb   rb   )rb   r   rb   r   r   rb   c                     [        [        R                  " S5      5      nUR                  X1US9n[        R
                  " U[        R                  " U[        R                  S95        g )Nr{   ru   r|   r%   r   r)   rD   rc   rf   assert_numpy_array_equalrd   re   )r   ru   r|   indexerrv   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   )rb   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|   rv   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   )ru   r~   ztolerance size must matchr   r   r7   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{   rb   r   rt   )r   r^   r   r%   r   r   )r   ru   rv   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)rJ   r*   uint64rangeru   rc   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   r^   r%   )TFT)rb   rb   rb   rc   r7   r   r   )	r   r   r)   rD   getattrrd   re   rf   r   )r   ru   r   numeric_indexotherrw   rv   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   )ro   rq   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   rt   )r   r   r   	TypeErrorrc   )r   ru   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   TFrb   r%   r   )	r   rc   r)   oneslenre   rf   r   r   )r   r   r   resrv   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?ro   )rb   r   r   rq   r   )r   r   r   )r   r)   r*   rf   r   rc   rd   re   )r   r-   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)   r1   r*   rc   rd   re   rf   r   )r   rw   rv   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{   )
r   rb   r   rb   r   rb   r7   rb   r^   rb   ro   rt   )
r   r   r   r   r   r   r7   r7   r^   r^   rq   )
r   r   r   r   r   r7   r7   r^   r^   r_   )
r   r   r)   rJ   rD   rc   rd   re   rf   r   )r   r   r   r   rv   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                 4   [        / SQ[        R                  S9n[        [        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 )N)            l   
        l           l           l           r%   r{   r   r_   r   )
r   rb   r   r   r7   r^   rb   rb   rb   rb   ro   rt   )
r   r   r   r   r7   r^   r^   r^   r^   r^   rq   )
r   r   r   r   r7   r^   rb   rb   rb   rb   )
r   r)   r   rD   astyperc   rd   re   rf   r   )r   index_larger   r   rv   s        r   test_get_indexer_uint64&TestGetIndexer.test_get_indexer_uint64*  s0   C))
 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)r^   r_   )r^   r^   r^   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   r7   r%   9r   r   )r   r   r   r   r+   )r   rN   val2any_numeric_ea_and_arrow_dtyper-   rw   s         r   test_get_loc_masked"TestGetIndexer.test_get_loc_masked>  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   rf   r   r)   rd   r   r   r+   )r   r   r-   rw   s       r   test_get_loc_masked_na%TestGetIndexer.test_get_loc_masked_naH  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Q5      nU(       a  SUS'   [        [        [         R                  " SSS[         R                  /5      US95      nU(       ay  UR                  [        5      n[         R                  " / SQ5      n[        R                  " XE5        UR                  [         R                  5      n[        R                  " XE5        ODU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U(       a(  UR                  [         R                  5      nUS:X  d   eOA[        R                  " [        SS9   UR                  [         R                  5        S S S 5        [         R                  " / SQ5      nU(       a  SUS'   [        [        [         R                  " SS[         R                  /5      US95      nUR                  [         R                  5      nUS:X  d   eU(       a  UR                  [        5      nUS:X  d   eg [        R                  " [        SS9   UR                  [        5        S S S 5        g ! , (       d  f       N= f! , (       d  f       g = f)N)FFTFTrb   r   r   )maskr   r7   )r   r   r#   )FFTr1   r   )FFFr   )r)   rd   r   r
   r1   r   r   rf   r   r   r   r+   )r   using_nan_is_nar   r-   rw   rv   s         r   test_get_loc_masked_na_and_nan-TestGetIndexer.test_get_loc_masked_na_and_nanV  s   xx34DHM"((Aq!RVV+<"=DIJ[[_Fxx :;H''9[[(F''9[[_FQ;;[[(FQ;;"((;/bhh?S6TU
 R{{[[(FQ;;xu5BFF# 6 xx-.DHM"((Aq"&&>":FGRVV${{[[_FQ;;xt4B 54 65 54s    K/K
K
K-rN   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   r7   r%   r_   r   F)check_dtype)r   r   get_indexer_forr)   rd   rf   r   )r   r   rN   r-   rw   rv   s         r   test_get_indexer_masked_na)TestGetIndexer.test_get_indexer_masked_na  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&   r-   rw   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   rb   r%   )r   r   r   r	   rd   
dictionaryint8rc   r)   rJ   rf   r   r   )
r   par   r-   rw   rv   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?HHaWBHH-HHaS)  	##H9
##H9r   rQ   )"rR   rS   rT   rU   rl   r   rV   rW   rx   r   r   listtupler   r)   rd   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   rX   rQ   r   r   rZ   rZ   d   s,   ' [[Yu6[[u%w'j)g&		
	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[[j)k*j)	
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+ Z [[UQF+I ,I [[Wy/&BC D:r   rZ   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   r_   r*   r%   r   r   r   r   c                 F   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       Q[
        R                  S9nUR                  U" U5      5      n[        R                  " XT5        g )NTFr   r%   )	r   wherer   	_na_valuetolistr)   r*   rf   assert_index_equal)r   listlike_boxr   condrv   rw   s         r   
test_whereTestWhere.test_where  s     vE
"\$/0w$3u:>22%//?E!"I,<,<,>?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   ra   r   r%   )FTFr   )r   ra   r   )	r   r)   r   rd   rJ   r   rf   r   putmask)r   r-   r   r   rv   rw   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)   r1   notnar   rf   r   )r   r   r   r   rv   rw   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   rQ   N)rR   rS   rT   rU   r   rV   rW   r   r)   rD   r   rJ   r   r   r  rX   rQ   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   r7   r^   r'   r&   name)r7   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   rb   )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  rd   rf   r   r1   r   r   r   
IndexError)r   r-   rw   rv   r3   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/ Y 	 ]]: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   r7   zUnable to fill values because z cannot contain NAr   Tr  Fr  r  r  r  r   r  )r   r  r)   rd   rf   r   r   rR   r   r   r   r  )r   r&   r-   rw   rv   r  r3   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#rQ   N)rR   rS   rT   rU   r   rV   rW   r)   r*   rJ   r   r  r  r!  rX   rQ   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   r7   r^   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)   r1   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   rQ   N)rR   rS   rT   rU   r   rV   rW   r)   r*   rJ   r   r&  r+  r.  rX   rQ   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   r_   ra   r   r   r{   r%   r   startr7   r   )r7   ra   r_   r{   end)r   ra   r   )r   r   rb   r   ra   r   )r   r_   r   r)   rd   r   r2   r   r&   r   nri   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 )Nr3  r%   g      @g      $@r7   g      @g      %@)r7   r   rb   r   r(   r8  r   r9  r:  s        r   test_slice_locs_float_locs(TestSliceLocs.test_slice_locs_float_locs<  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{      rB     r%   rB  r   r7         rb   )r   r)   rd   r2   )r   r&   r   ri   s       r   test_slice_locs_dup_numeric)TestSliceLocs.test_slice_locs_dup_numericG  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   rD  )r   r7   r   )r   r_   )r   r)   r1   r2   r   s     r   test_slice_locs_na TestSliceLocs.test_slice_locs_naQ  s    rvvq!n%"f,,,'6111q"&&"&&!Q/0'6111r   c                 H   [        [        R                  SS/5      nSn[        R                  " [
        US9   UR                  SS9  S S S 5        [        R                  " [
        US9   UR                  SS9  S S S 5        g ! , (       d  f       N@= f! , (       d  f       g = f)Nr   r   z,non-monotonic index with a missing label 1.5r   r(   r4  r6  )r   r)   r1   r   r   r+   r2   )r   r   r3   s      r   test_slice_locs_na_raises'TestSliceLocs.test_slice_locs_na_raisesY  sz    rvvq!n%<]]83/3' 0 ]]83/% 0/ 0/ 0/s   B)B
B
B!rQ   N)rR   rS   rT   rU   r   rV   rW   intfloatr<  r?  rG  rJ  rM  rX   rQ   r   r   r1  r1  *  s    [[WsEl31 41  [[WsEl35 45 [[WsEl33 432&r   r1  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)TestGetSliceBoundsic  zside, expected)r   r^   )r   r_   c                 Z    [        [        S5      5      nUR                  SUS9nXB:X  d   eg )Nra   r^   sider   r   get_slice_bound)r   rU  rv   r   rw   s        r   test_get_slice_bounds_within/TestGetSliceBounds.test_get_slice_bounds_withind  s0    eAh&&qt&4!!!r   rU  r   r   zbound, expected)rb   r   )r{   ra   c                 X    [        [        S5      5      nUR                  X1S9nXR:X  d   eg )Nra   rT  rV  )r   rU  rv   boundr   rw   s         r   test_get_slice_bounds_outside0TestGetSliceBounds.test_get_slice_bounds_outsidej  s0     eAh&&u&8!!!r   rQ   N)
rR   rS   rT   rU   r   rV   rW   rX  r\  rX   rQ   r   r   rR  rR  c  sv    [[-\/JK" L"
 [[Vfg%67[[.'0BC" D 8"r   rR  )numpyr)   r   pandas.errorsr   pandasr   r   r   r   r   pandas._testing_testingrf   pandas.core.arraysr	   r
   r   rZ   r   r  r$  r1  rR  rQ   r   r   <module>rd     sq      +  M% M%`E: E:P
)0 )0X?( ?(D  6& 6&r" "r   