
    m*iQ8              
       N   S SK rS SKrS SKJs  Jr  S SKJrJ	r	J
r
JrJrJrJrJrJrJrJrJrJrJr  S SKJr  S rS rS rS rS rS rS	 rS
 r S r!S r"S r#S r$S r%S r&S r'S r(\RR                  \RT                  RW                  SSS/5      \RT                  RW                  SSS \" S 5      /5      S 5       5       5       r,S r-S r.\RT                  RW                  S\" SS5      \" SS5      /\" SS5      \" SS5      S/S S /S S \R^                  /4\" SS5      \" SS5      /\" SS5      \" SS5      /S S /S S /4/5      S! 5       r0S" r1S# r2S$ r3\RT                  RW                  S%S&/S'/// / //5      S( 5       r4S) r5S* r6\RT                  RW                  SS+S,/5      S- 5       r7g).    N)NACategoricalFloat64DtypeIndex
MultiIndexNaTPeriodPeriodIndex
RangeIndexSeries	Timedelta	Timestamp
date_rangeisnac                    UR                  UR                  5      n[        R                  " UR                  UR                  5      (       d   eUR                  R	                  UR                  5      (       d   eUR                  R                  UR                  5      (       d   eUR                  SS nUR                  U5      nUR                  5        H  u  pVXaU   :X  a  M   e   U R                  SS nU R                  U5      nUR                  5        H  u  pVX`U   :X  a  M   e   U R                  U5      n	[        R                  " U	5      R                  5       (       d   eU R                  S S S2   n
U R                  U
5      nUR                  5        H  u  pVX`U   :X  a  M   e   U R                  5       nXLd   eg )N
         )
reindexindextmshares_memoryis_	identicalitemsnpisnanall)datetime_seriesstring_seriesidentitysubIndex	subSeriesidxval	subIndex2subTSstuffSeriesnonContigIndexsubNonContigresults                h/var/www/html/land-doc-ocr/venv/lib/python3.13/site-packages/pandas/tests/series/methods/test_reindex.pytest_reindexr-      s   $$]%8%89HM//@@@@>>m112222>>##M$7$78888""2b)H%%h/IOO%C(((( &  %%b,I##I.EKKMc**** "!))(3K88K $$&&&& %**3Q3/N"**>:L &&(c**** ) $$&F(((    c                     [        / SQSS[        R                  S/S9n [        R                  S[        R                  SS[        R                  // SQp![        R                  " U R                  U5      U R                  U   5        U R                  R                  S5      U l        [        R                  " U R                  U5      U R                  U   SS	9  g )
N)r                     r   )r   r   r   r0   r3   r   objectF)check_index_type)	r   r   nanr   assert_series_equalr   ilocr   astype)tsijs      r,   test_reindex_nanr@   ;   s    	Q2661$5	6BFFArvvq!RVV,.@q2::a="''!*5xxx(BH 2::a="''!*uMr.   c                  d   [        SSSS9n [        U 5      nUR                  [        S5      5      n[        R
                  " UR                  [        R                  " S5      5      (       d   eUR                  5       nUSS  R                  5       (       d   eUS S R                  5       (       a   eg )Nz1/1/2000 00:00:00r   10speriodsfreq   zM8[ns])
r   r   r   ranger   
issubdtypedtyper   r   any)rngseriesr+   masks       r,   test_reindex_series_add_natrO   G   s    
("5
ACC[F^^E"I&F==rxx'9::::;;=D9==???CRy}}r.   c                     [        SSS9n [        [        R                  R	                  S5      R                  S5      U S9nUR                  [        UR                  SS 5      5      nUSS nUR                  R                  S 5      Ul        [        R                  " X#5        U[        UR                  SS 5         n[        R                  " X#5        g )Nz1/1/2000r   rD   r   r6   r1   r   )r   r   r   randomdefault_rngstandard_normalr   listr   
_with_freqr   r:   )rL   r=   r+   expecteds       r,   test_reindex_with_datetimesrX   S   s    
Z
,C			%%a(88<C	HBZZRXXa^,-F!BxH^^..t4HN6,RXXa^$%F6,r.   c                    [        / S9nUR                  U R                  SS9  UR                  U R                  SS9nU R                  [        U R                  5      5      nU R                  R	                  S 5      U l        [
        R                  " X5        U S S S2   nSn[        R                  " [        US9   UR                  U R                  SS9  S S S 5        g ! , (       d  f       g = f)Nr6   padmethodr   zVInvalid fill method\. Expecting pad \(ffill\), backfill \(bfill\) or nearest\. Got foomatchfoo)
r   r   r   rU   rV   r   r:   pytestraises
ValueError)r   empty	reindexedr=   msgs        r,   test_reindex_cornerrf   `   s    E	MM/''M6 o33EBI  ''_-B-B(CDI+11<<TBO?6 
1	B	*  
z	-


?((
7 
.	-	-s   ;C
C-c                  *   [        [        R                  " S5      SS9n U S S S2   nUR                  U R                  SS9nUR                  U R                  SS9n[
        R                  " X#5        [        / SQ5      n[
        R                  " X$5        g )	Nr   int64rJ   r   rZ   r[   ffill)
r   r   r   r   r4   r4      rk   r5   r5   )r   r   aranger   r   r   r:   )ss2rd   
reindexed2rW   s        r,   test_reindex_padrp   w   sv    ryy}G,A	
3Q3B

1775
1IAGGG4J9145H9/r.   c                     [        / SQ/ SQS9n / SQn[        / SQUS9nU R                  U5      R                  5       n[        R                  " X2R                  S5      5        Sn[        R                  " [        US9   U R                  U5      R                  S	S
9nS S S 5        [        R                  " X25        [        / SQUS9nU R                  USS9n[        R                  " X25        g ! , (       d  f       NW= f)Nr3   r   r0   r4   r1   )abcder6   )rs   gru   f)r3   r3   r0   r0   float64z-The 'downcast' keyword in ffill is deprecatedr]   infer)downcast)r3   r1   r0   r1   rj   r[   )r   r   rj   r   r:   r<   assert_produces_warningFutureWarning)rm   	new_indexrW   r+   re   s        r,   test_reindex_pad2r      s    &?@A$Il)4H YYy!'')F6??9#=>
9C		#	#M	=9%++W+= 
>6,l)4HYYyY1F6, 
>	=s   >C..
C<c                  D   [        / SQ[        S5      S9n SnSn[        R                  " [        US9   U R                  [        U5      5      R                  5       nS S S 5        [        / SQ[        U5      S9n[        R                  " WU5        g ! , (       d  f       N;= f)N)TFFTabcdr6   agc"Downcasting object dtype arrays onr]   )TTF)r   rU   r   r}   r~   r   rj   r:   )rm   r   re   r+   rW   s        r,   test_reindex_inferencer      sw    )f>AI
.C		#	#M	=4	?+113 
>)iAH68, 
>	=s   )B
Bc                  &   [        S[        S5      S9n Sn[        R                  " [        US9   U R                  S5      R                  5       nS S S 5        [        S[        S5      S9n[        R                  " WU5        g ! , (       d  f       N9= f)NFr1   r6   r   r]   r3   )r   rH   r   r}   r~   shiftbfillr:   )rm   re   r+   rW   s       r,   test_reindex_downcastingr      sk    uE!H%A
.C		#	#M	=!!# 
>e58,H68, 
>	=s    B
Bc                  &   [        [        R                  " SSS95      n / SQnU R                  USS9n[        [        R                  " U5      R                  S5      U5      n[        R                  " X25        U R                  USSS9n[        S	S
[        R                  S/U5      n[        R                  " X25        U R                  US/ SQS9n[        S	[        R                  [        R                  S/U5      n[        R                  " X25        g )Nr   rh   ri   )g?g?g      ?       @nearestr[   g?)r\   	tolerancer   r3   r   )g333333?g{Gz?g?r0   )	r   r   rl   r   aroundr<   r   r:   r9   )rm   targetr+   rW   s       r,   test_reindex_nearestr      s    ryy7+,A!FYYviY0Fbii'..w7@H8,YYvi3Y?Fq!RVVQ'0H8,YYvi;NYOFq"&&"&&!,f5H8,r.   c                    U S S S2   n[        [        R                  " [        U5      [        S9UR
                  S9nUR                  U R
                  5      nUR                  [        R                  :X  d   eUR                  UR
                  S S S2   5      nUR                  [        R                  " [        5      :X  d   eg Nr   ri   r6   )	r   r   zeroslenintr   r   rJ   rz   )r   r=   int_tsreindexed_ints       r,   test_reindex_intr      s    	1	BBHHSWC0AF NN?#8#89M "**,,, NN6<<!#45M"((3-///r.   c                 t   U S S S2   n[        [        R                  " [        U5      [        S9UR
                  S9nUR                  U R
                  5      nUR                  [        R                  :X  d   eUR                  UR
                  S S S2   5      nUR                  [        R                  :X  d   eg r   )
r   r   r   r   boolr   r   rJ   object_bool_)r   r=   bool_tsreindexed_bools       r,   test_reindex_boolr      s    	1	BRXXc"gT2"((CG ___%:%:;N 2::--- __W]]3Q3%78N288+++r.   c                     U SS  n[        [        R                  " [        U5      [        S9UR
                  S9nUR                  U R
                  SS9n[        US S 5      R                  5       (       d   eg )Nr1   ri   r6   rZ   r[   )	r   r   r   r   r   r   r   r   r   )r   r=   r   filled_bools       r,   test_reindex_bool_padr      se    		BRXXc"gT2"((CG///"7"7/FKBQ $$&&&&r.   c                  N   [        SSS9n [        / SQSS9nUR                  U 5      n[        [        [        R
                  [        R
                  [        R
                  // SQS95      nXl        [        R                  " X#5        [        [        SS	// SQS95      nS
S/Ul        UR                  S
S/5      n[        R                  " X#5        [        [        S	[        R
                  // SQS95      nSS/Ul        UR                  SS/5      n[        R                  " X#5        g )N20000101r0   rQ   )rs   rt   ru   categoryri   )values
categoriesrt   ru   r3   r   )	r   r   r   r   r   r9   r   r   r:   )r   rm   r+   rW   s       r,   test_reindex_categoricalr      s    z1-E 	j1AYYuFBFFBFFBFF3PH N6, k#sPQHVHNYY1vF6,k#rvv?STHVHNYY1vF6,r.   c                      [        / SQ/ SQS9n / SQnSn[        nU R                  U5      R                  U5      R                  U5      nU R                  U5      R                  U5      R                  U5      n[        R
                  " XE5        g )Nr3   r   r0   )r   r   r3   r6   )r   r3   r   r   )r   strr   r<   r   r:   )serr   
temp_dtype	new_dtyper+   rW   s         r,   %test_reindex_astype_order_consistencyr      ss    
)
,CIJI[[#**:6==iHFzz*%--i8??	JH6,r.   c                     [        / SQ5      n U R                  / SQ5      n[        SS[        R                  // SQS9n[        R
                  " X5        U R                  / SQSS9n[        / SQ/ SQS9n[        R
                  " X5        [        / SQ5      nUR                  / SQ5      n[        SS[        R                  // SQS9n[        R
                  " X5        UR                  / SQSS9n[        / S	Q/ SQS9n[        UR                  R                  [        R                  5      (       d   e[        R
                  " X5        [        / SQ[        S
9nUR                  / SQ5      n[        SS[        R                  // SQ[        S9n[        R
                  " X5        UR                  / SQSS9n[        / SQ/ SQ[        S9n[        R
                  " X5        [        / SQ5      nUR                  / SQ5      n[        SS[        R                  // SQ[        S9n[        R
                  " X5        UR                  / SQSS9n[        / SQ/ SQS9n[        R
                  " X5        g )N)      ?r         @r   r   r   r6   r   
fill_value)r   r   r   )r   r0   r   ri   r   r0   r   rJ   r_   )r   r0   r_   )TFTFT)FTF)r   r   r   r9   r   r:   
issubclassrJ   typeintegerr7   )floatsr+   rW   intsobjectsboolss         r,   test_reindex_fill_valuer     s    O$F^^I&FsC(	:H6,^^I!^4Fm95H6, )D\\)$FsC(	:H6, \\)\2Fiy1Hfll''44446, Yf-G__Y'Fq!RVVnIVDH6,__Y5_9Fm9FCH6, &'E]]9%FudBFF+9FKH6,]]9]7F*)<H6,r.   rJ   zdatetime64[ns]timedelta64[ns]r   stringc                     U S:X  a  U[        S5      :X  a  [        S5      n[        [        /U S9nUR	                  SS/US9n[        [        U/SS/[
        S9n[        R                  " XE5        g )Nr   r   ri   r3   r   r   )r   r   r   r   r   r7   r   r:   )rJ   r   using_array_managerr   r+   rW   s         r,   +test_reindex_fill_value_datetimelike_upcastr   9  sj    
 !!jIaL&@q\

#e
$C[[!QJ[7FsJ'1vVDH6,r.   c                      [        SSSS9n [        SSSS9n[        [        S5      U S	9nS
n[        R                  " [
        US9   UR                  USS9  S S S 5        g ! , (       d  f       g = f)N20131101zAmerica/Chicagor2   )tzrD   20131103r   hrC   r6   zLCannot compare dtypes datetime64\[ns, America/Chicago\] and datetime64\[ns\]r]   rj   r[   )r   r   rH   r`   ra   	TypeErrorr   )r$   newidxrm   re   s       r,   /test_reindex_datetimeindexes_tz_naive_and_awarer   I  se    
Z$5q
AC
BS9FuQxs#A	   
y	,			&	) 
-	,	,s   A
A-c                      [        SS9R                  SS/5      n [        [        /S-  SS9n[        R                  " X5        g )Nzdatetime64[ns, UTC]ri   r   r3   r   )r   r   r   r   assert_equal)r+   rW   s     r,   "test_reindex_empty_series_tz_dtyper   V  s;    /088!Q@Fseai'<=HOOF%r.   z+p_values, o_values, values, expected_values2019Q1zQ-DEC2019Q2Allr   c                     [        U 5      n[        U5      n[        X$S9nUR                  U5      n[        X5S9n[        R
                  " Xx5        g )Nr6   )r
   r   r   r   r   r:   )	p_valueso_valuesr   expected_valuesperiod_indexobject_indexr   r+   rW   s	            r,   $test_reindex_periodindex_with_objectr   ]  sG    & x(L?L

,C[[&Fo:H6,r.   c                      [        SS/5      n Sn[        R                  " [        US9   U R	                  SS/S5        S S S 5        g ! , (       d  f       g = f)Nr3   r   zCreindex\(\) takes from 1 to 2 positional arguments but 3 were givenr]   r0   Fr   r`   ra   r   r   r   re   s     r,   test_reindex_too_many_argsr   y  s@    
!Q.C
PC	y	,QFE" 
-	,	,   A
Ac                      [        SS/5      n Sn[        R                  " [        US9   U R	                  SS/SS/S9  S S S 5        g ! , (       d  f       g = f)Nr3   r   z4reindex\(\) got multiple values for argument 'index'r]   r0   r4   r6   r   r   s     r,   test_reindex_double_indexr     sF    
!Q.C
AC	y	,QF1a&) 
-	,	,r   c                      [        SS/5      n U R                  SS/S9n[        SS/SS/S9n[        R                  " X5        g )Nr3   r   r   r6   )r   r   r   r:   )r   r+   rW   s      r,   test_reindex_no_posargsr     sE    
!Q.C[[1v[&Fq!fQF+H6,r.   r   rs   xc                    [        [        [        U S   5      5      [        R                  " U 5      SS9nUR                  [        R                  " S/5      SS9n[        [        S/U S   // / /S9SS9n[        R                  " X#5        g )Nr   r7   r   rt   )levelr3   )levelscodes)
r   rH   r   r   from_arraysr   r   arrayr   r:   )r   r   r+   rW   s       r,   test_reindex_empty_with_levelr     s     c&)nZ%;%;F%C8C [[3%[2F#q	 22r(C8H 6,r.   c                      [        / SQSS9n Sn[        R                  " [        US9   U R	                  / SQSS9  S S S 5        g ! , (       d  f       g = f)	N)r3   r   r0   r3   r   ri   z:Cannot setitem on a Categorical with a new category \(-1\)r]   rr   r   r   r   s     r,   test_reindex_missing_categoryr     s=    
Z
0C
GC	y	,O3 
-	,	,s   A
Ac                  D   [        S[        /[        5       S9n U R                  [	        S5      5      nUR
                  R                  n[        R                  " S[        R                  [        R                  /5      n[        R                  " X#5        [        R                  " S 5         [        R                  " U5      n[        S[        R                  [        R                  /[        5       S9n[        R                  " XE5        S S S 5        g ! , (       d  f       g = f)Nr   ri   r0   r3   r   )r   r   r   r   rH   r   _datar   r   r9   r   assert_numpy_array_equalr}   logr:   )rm   	s_reindexr+   rW   
result_logexpected_logs         r,   #test_reindexing_with_float64_NA_logr     s    Ry/A		%(#I##FxxBFFBFF+,H1		#	#D	)VVI&
q"&&"&&1H
z8 
*	)	)s   *AD
Dtimedelta64
datetime64c           	         [        [        R                  " S/U  S3S95      nUR                  [	        S5      5      n[        [        R                  " S[        [        U 5      " SS5      /U  S3S95      n[        R                  " X#5        g )Nr3   z[s]ri   r   natrm   )r   r   r   r   r   getattrr   r:   )rJ   r   r+   rW   s       r,   test_reindex_expand_nonnano_natr     st     1#wc]3
4C[[A'F
!WR's34ugSMJH 6,r.   )8numpyr   r`   pandas.util._test_decoratorsutil_test_decoratorstdpandasr   r   r   r   r   r   r	   r
   r   r   r   r   r   r   pandas._testing_testingr   r-   r@   rO   rX   rf   rp   r   r   r   r   r   r   r   r   r   r   &skip_array_manager_not_yet_implementedmarkparametrizer   r   r   r9   r   r   r   r   r   r   r   r    r.   r,   <module>r     s     ) )     )D	N	
-8.	0-(--- 0, '-0-0-f **#35F"GH!Yq\'BC
- D I +
-
*& 1 Hg&x(ABHg&x(A5I#J#rvv		
 Hg&x(ABHg&x(AB#J#J		
"-#"-#*- cUSENRH#=>	- ?	-4
9 =,"?@- A-r.   