
    I j1                     |    S SK Jr  S SKrS SKrS SKrS SKJrJrJ	r	J
r
JrJr  S SKJr   " S S5      r " S S5      rg)    )ascii_lettersN)	DataFrameIndexSeries	Timestamp
date_rangeoption_contextc                   b    \ rS rSr\R
                  R                  SSS/5      S 5       rS rSr	g)	TestCaching   do_refTFc                     / SQn[        X"SS  US S -   [        R                  " S5      S.5      nU(       a  UR                  S     SUR                  S'   UR                  S   S:X  d   eUR                  S   S	:X  d   eg )
N)onetwothreefourfivesixseven      abc)r   r      )r   r           g      ?)r   nparangeloc)selfr   contdfs       ڀ/root/GenerationalWealth/GenerationalWealth/venv/lib/python3.13/site-packages/pandas/tests/indexing/test_chaining_and_caching.pytest_setitem_cache_updating'TestCaching.test_setitem_cache_updating   s}     GTQR48(;"))A,OP FF6N vvvf~$$$vvf~$$$    c                 *   [        S/ SQ0[        SS5      S9n[        S/ SQ0[        SS5      S9n[        / SQ/ SQS	.5      n[        S5      n[        S5      nUR                  5        H3  u  pgUR                  XE2US
   4   US   -   UR                  XE2US
   4'   M5     [
        R                  " X!5        [
        R                  " US   US   5        [        S/ SQ0[        SS5      S9nUR                  5       nUR                  5        H;  u  pgX'S
      XE US   -   n	[
        R                  " 5          XUS
      XE& S S S 5        M=     [
        R                  " X(5        [
        R                  " US   US   5        [        S/ SQ0[        SS5      S9nUR                  5        H%  u  pgUR                  XE2US
   4==   US   -  ss'   M'     [
        R                  " X!5        [
        R                  " US   US   5        g ! , (       d  f       GM  = f)NA)X  r*   r*   z5/7/2014z5/9/2014)index)r   r   r   )r)   r)   r)   )d      i,  )CDr.   r/   )
r   r   r   iterrowsr    tmassert_frame_equalassert_series_equalcopyraises_chained_assignment_error)
r!   expectedoutr#   r   eixixrowout_originalvs
             r$   "test_setitem_cache_updating_slices.TestCaching.test_setitem_cache_updating_slices$   s    /"*Z*L
 i(
:z0RS_?CD 
#
#{{}GB),#c(1B)Cc#h)NCGGCGSX%& % 	c,
s3x#7 i(
:z0RSxxz{{}GBHc&S1A335)*CHc& 65 %
 	c0
s3xc):;i(
:z0RS{{}GBGGCGSX%&#c(2& % 	c,
s3x#7 65s   4
H
H	 N)
__name__
__module____qualname____firstlineno__pytestmarkparametrizer%   r=   __static_attributes__r?   r'   r$   r   r      s.    [[Xe}5% 6% #8r'   r   c            	          \ rS rSrS r\R                  R                  S 5       r\R                  R                  S 5       r	\R                  R                  S 5       r
\R                  R                  S 5       r\R                  R                  S 5       r\R                  R                  S 5       r\R                  R                  S	 5       r\R                  R                  S
 5       r\R                  R                  S 5       r\R                  R                  S 5       r\R                  R                  S 5       rS rS rS r\R                  R-                  SS\" S/ SQ05      /5      S 5       rS rS rS rS rS rSrg)TestChainingJ   c                 4   / SQn[        S[        R                  " U5      05      nUR                  S:H  n[        R
                  " 5          SUR                  U'   S S S 5        [        R                  " U[        SU05      5        [        R                  R                  U/S/S9n[        U5      nUR                  S:H  n[        R
                  " 5          SUR                  U'   S S S 5        [        R                  " U[        SU05      5        [        XS.5      nUR                  5       nUR                  S:H  n[        R
                  " 5          SUR                  U'   S S S 5        [        R                  " X%5        [        S[        R                  SS	S
S/05      n[        S[        R                  " / SQ5      05      n[        R
                  " 5          [        R                  US   R                  S'   S S S 5        [        S/ SQ05      nUR                  5       n[        R                  " Xv5        [        S[        R                  " / SQ5      05      n[        R
                  " 5          [        R                  UR                  R                  S'   S S S 5        UR                  5       n[        R                  " Xv5        g ! , (       d  f       GNn= f! , (       d  f       GN= f! , (       d  f       GN= f! , (       d  f       GN	= f! , (       d  f       N}= f)N)rightleftrM   rM   rL   rM   timeoutresponserN   none)names)rO   	response1r)   barbahfoo)rU   rS   rT   rU   rS   r   )r   r   arrayrO   r1   r5   r2   rec
fromarraysr4   nanilocheadr)   )r!   datar#   maskrecarraydf_originalr6   results           r$   test_setitem_chained_setfault*TestChaining.test_setitem_chained_setfaultK   s3   L
BHHTN34{{i'//1 &BKK 2
b)Z,>"?@66$$dVJ<$@x {{i'//1 &BKK 2
b)Z,>"?@D<=ggi{{i'//1 &BKK 2
b. cBFFE5%#GHIRXX&IJKL//1 ffBsGLLO 2c#FGH
f/RXX&IJKL//166BDDIIaL 2
f/? 21 21 21 21 21s<   
KKK%!K7*(L	
K
K"%
K47
L	
Lc                    [        SS5         [        [        R                  " S5      R	                  SS5      [        S5      SS9nUR                  5       n[        R                  " 5          SUS	   S
'   S S S 5        [        R                  " 5          SUS	   S'   S S S 5        [        R                  " X5        S S S 5        g ! , (       d  f       NS= f! , (       d  f       N>= f! , (       d  f       g = f)Nchained_assignmentraise      ABint64)columnsdtyper)   r   r   )
r	   r   r   r   reshapelistr4   r1   r5   r2   r!   r#   r_   s      r$   test_detect_chained_assignment+TestChaining.test_detect_chained_assignmentr   s    0':		!$$Q*DJgB '')K3353
 63353
 6!!"2 ;: 6555 ;:s<   AC#+	C4C#	CC#
C	C#
C 	C##
C1c           	         [        [        [        S5      SS9[        R                  " [        R
                  " SS5      [        R                  S9S.5      nUR                  5       n[        R                  " 5          SUS   S'   S S S 5        [        R                  " 5          S	US   S
'   S S S 5        [        R                  " X5        g ! , (       d  f       NK= f! , (       d  f       N6= f)Nrg   ri   rk   rf   r)   Brl   r)   r   rm   r   )r   r   ranger   rV   r   float64r4   r1   r5   r2   rp   s      r$   %test_detect_chained_assignment_raises2TestChaining.test_detect_chained_assignment_raises   s     E!HG4XXbii1oRZZ@
 ggi//1BsGAJ 2//1BsGAJ 2
b.	 2111s   ;	C	!	C	
C
C(c           	      0   [        [        [        S5      SS9[        R                  " [        R
                  " SS5      [        R                  S9S.5      n[        R                  " 5          SUR                  S   S'   S S S 5        g ! , (       d  f       g = f)	Nrg   ri   rt   rf   ru   rl   r   r)   )
r   r   rw   r   rV   r   rx   r1   r5   r    r!   r#   s     r$   $test_detect_chained_assignment_fails1TestChaining.test_detect_chained_assignment_fails   sh     E!HG4XXbii1oRZZ@
 //1BFF1IcN 211s   +B
Bc                     [        / SQ[        [        S5      SS9S.5      nUR                  R                  R                  S5      n[        R                  " 5          SX   S'   S S S 5        g ! , (       d  f       g = f)	N)r   r   r   r   r   r   r   r   ri   rt   )r   r   o*   r   )r   r   rw   r   str
startswithr1   r5   )r!   r#   indexers      r$   *test_detect_chained_assignment_doc_example7TestChaining.test_detect_chained_assignment_doc_example   sb     HE!HG4
 $$((%%c*//1!BK 211s   A--
A;c                     [        [        / SQ[        S9/ SQS.5      nUR                  5       n[        R
                  " 5          SUS   S'   S S S 5        [        R                  " X5        g ! , (       d  f       N%= f)Naaabbbcccrt   r   rg   r   ru   o   r)   r   )r   r   objectr4   r1   r5   r2   rp   s      r$   +test_detect_chained_assignment_object_dtype8TestChaining.test_detect_chained_assignment_object_dtype   s^    .f=IN
 ggi//1BsGAJ 2
b. 21s   	A,,
A:c                     [        SSS/05      n[        U5      nUR                  U5        [        R                  " U5      nUS   US'   US   US'   g )Nr)   r   rg   rv   )r   r   	to_picklepdread_pickle)r!   	temp_filer#   pathdf2s        r$   -test_detect_chained_assignment_is_copy_pickle:TestChaining.test_detect_chained_assignment_is_copy_pickle   sS     aV}%9~
TnnT"s8Cs8Cr'   c                    [         R                  R                  S5      R                  [	        [
        5      SS9nUR                  SS9  U Vs/ s H  n[
        US   US    PM     nn[        US/S9nUR                  R                  S	 5      nUR                  US4   R                  [        R                  5      UR                  US4'   g s  snf )
Nrg   )r,   rg   )sizer   )axisr   lettersrj   c                     [        U 5      S:  $ )N
   )len)xs    r$   <lambda>ATestChaining.test_detect_chained_assignment_str.<locals>.<lambda>   s    SVb[r'   )r   randomdefault_rngintegersr   r   sortr   r   applyr    r   lower)r!   idxsr   stringsr#   r   s         r$   "test_detect_chained_assignment_str/TestChaining.test_detect_chained_assignment_str   s    yy$$Q'00]1C(0S		q	7;<t!=1!-t<w4**""#89%'VVGY,>%?%E%Ecii%Pw	!"	 =s   Cc                 l   [        [        R                  R                  S5      R	                  S5      5      nUR
                  S S 2S4   R                  5       n[        R                  " X!R
                  S S 2S4   R                  5       5        [        R                  " X!S   R                  5       5        g )Nrg   r   rf   r   )	r   r   r   r   standard_normalrZ   sort_valuesr1   r3   )r!   r#   sers      r$   &test_detect_chained_assignment_sorting3TestChaining.test_detect_chained_assignment_sorting   s~    ryy,,Q/??HIggadm'')
sGGAqDM$=$=$?@
sqE$5$5$78r'   c                     [        / SQ/ SQS.5      n[        U5        US   S-   US'   [        U5        XS   S:g     n[        U5        US   S-   US'   [        U5        g )	N)r   r   r   )rf      	   )column1column2r   r   r   r   r   )r   r   r|   s     r$   .test_detect_chained_assignment_false_positives;TestChaining.test_detect_chained_assignment_false_positives   sd     ?yIJB9+9B9"#B9+9Br'   c                    [        [        R                  " SS5      S/S9nSUS'   UR                  5       n[        R
                  " 5          SUR                  SS S'   S S S 5        [        R                  " X5        g ! , (       d  f       N%= f)	Nr   r   countr   r   groupr      )r   r   r   r4   r1   r5   rZ   r2   rp   s      r$   /test_detect_chained_assignment_undefined_column<TestChaining.test_detect_chained_assignment_undefined_column   sl     ryyA	:7ggi//1$'BGGAaL! 2
b. 21s   A==
Bc           	         [        [        SSS9[        R                  R	                  S5      R                  S5      [        R                  " SSS9/ SQS.5      nUR                  5       n[        R                  " 5          S	UR                  S   S
'   S S S 5        [        R                  " 5          S	UR                  S   S'   S S S 5        [        R                  " X5        [        R                  " [        SS9   [        R                  " 5          S	US   S'   S S S 5        S S S 5        g ! , (       d  f       N= f! , (       d  f       N~= f! , (       d  f       N9= f! , (       d  f       g = f)N20130101r   )periodsrg   ri   rt   r   r   r   de)r)   rv   r.   r/   rU   r/   r.   zInvalid valuematch)r   r   r   r   r   r   r   r4   r1   r5   r    r2   rD   raises	TypeErrorrp   s      r$   -test_detect_chained_assignment_changing_dtype:TestChaining.test_detect_chained_assignment_changing_dtype   s     
A6YY**1-==a@YYq0.	
 ggi//1"BFF1IcN 2//1"BFF1IcN 2
b.]]9O<335"3
 6 =< 2111
 65 =<s<   ?D)/D:9E	EE)
D7:
E
E	E
E*c                    [        [        [        S5      5      [        S5      SS[        R                  S/S.5      nUR                  5       n[        R                  " UR                  5      n[        R                  " 5          US/   U   US/   U'   S S S 5        [        R                  " X5        g ! , (       d  f       N%= f)Nrf   zab..r   r   r   r   r   )r   ro   rw   r   rY   r4   r   isnar   r1   r5   r2   )r!   r#   r_   r]   s       r$   test_setting_with_copy_bug'TestChaining.test_setting_with_copy_bug  s    uQx.tF|3RVVS:QR
 ggiwwrtt}//1 #ioBuIdO 2
b. 21s   B22
C c                 f    [        [        / SQ5      [        / SQ5      S.5      nUS/   n/ SQUS'   g )Nr   )r   r   f)r   yr   )ghir   )r   r   )r!   df1r   s      r$   %test_setting_with_copy_bug_no_warning2TestChaining.test_setting_with_copy_bug_no_warning  s3     f_5F?<STU3%j #Cr'   c                     [        / SQ/ SQS.5      n[        R                  " 5          SUR                  S   S'   S S S 5        g ! , (       d  f       g = f)Nr   r   ru   r   r   r)   )r   r1   r5   r    r|   s     r$   .test_detect_chained_assignment_warnings_errors;TestChaining.test_detect_chained_assignment_warnings_errors  s;    2CD//1 BFF1IcN 211s   A
Arhsr   r   )r   rg   r   rf   c                     [        [        R                  " S5      R                  SS5      5      nUR	                  5       nUR
                  S S nXS'   [        R                  " X#5        g )N   r   r   rg   )r   r   r   rn   r4   r    r1   r2   )r!   r   r#   r_   chaineds        r$   1test_detect_chained_assignment_warning_stacklevel>TestChaining.test_detect_chained_assignment_warning_stacklevel  sR     ryy},,Q23ggi&&!*

b.r'   c                    [        S[        R                  " S5      /-  S[        R                  " S5      /-  S.5      nUS   R                  S   nUR
                  S   n[        R                  " X25        UR                  S   S   n[        R                  " XB5        US   R
                  S   n[        R                  " XR5        US   R                  S   n[        R                  " Xb5        g )Nr   r   ru   r)   rg   )rg   r)   )r   r   zerosonesrZ   r    r1   assert_numpy_array_equal)r!   r#   r6   r`   result2result3result4s          r$   test_chained_getitem_with_lists,TestChaining.test_chained_getitem_with_lists'  s    
 Q"((1+.Q"''!*5EFGc7<<?
##F5''!*S/
##G6S'++a.
##G6S',,q/
##G6r'   c                     [        [        R                  " S5      [        [	        S5      [
        S9S9nUS     UR                  S   UR                  S'   SUS   R                  ;   d   eSUS   R                  ;   d   eg )	Nr   ABCDrt   r   r)   r   zHello Friendrv   )	r   r   r   r   ro   r   rZ   r    r+   r|   s     r$   test_cache_updating TestChaining.test_cache_updating7  ss     HHW$v,f5
 	3!#~C...C...r'   c                 D   [        [        R                  " SSS9/ SQ[        S5      S9nSUS'   UR	                  5       n[
        R                  " [        S	S
9   SUR                  R                  S'   S S S 5        [        R                  " X5        g ! , (       d  f       N%= f)N)r   r   ri   rt   r   r   )rj   r+   r   r   z	read-onlyr   r   r   )r   r   r   rw   r4   rD   r   
ValueErrorr   valuesr1   r2   )r!   r#   df_origs      r$   test_cache_updating2!TestChaining.test_cache_updating2C  su    HHV7+-(

 3'')]]:[9BDDKKN :
b* :9s   B
Bc                    [        SS 5         [        [        S5      S/S-  S.5      nSUS'   S/[        U5      -  n[        R
                  " 5          SUS	   R                  S
'   S S S 5        UR                  U     [        R
                  " 5          SUS	   R                  S
'   S S S 5        US	   R                  S
   S:X  d   e S S S 5        g ! , (       d  f       No= f! , (       d  f       NA= f! , (       d  f       g = f)Nrd   r   g@)aabbr   ccTgp=
ף?r   r   g333333?)r	   r   rw   r   r1   r5   rZ   )r!   r#   cks      r$   $test_iloc_setitem_chained_assignment1TestChaining.test_iloc_setitem_chained_assignmentP  s    0$7%(3%!)<=BBtH#b'!B335#'4a  6 GGBK335#'4a  6 d8==#s*** 87 65 65 87s<   AC0C%,C0C$ C0
C	C0
C-	)C00
C>c                 4   [        S/ SQ05      n[        R                  " 5          SUS   R                  S'   S S S 5        [        R                  " U[        S/ SQ05      5        [        R
                  " US   [        / SQSS95        g ! , (       d  f       NW= f)Nr   )r         (   rf   )name)r   r1   r5   r    r2   r3   r   r|   s     r$   'test_getitem_loc_assignment_slice_state4TestChaining.test_getitem_loc_assignment_slice_statec  sq    \*+//1BsGKKN 2
b)S,,?"@A
r#w|#(FG 21s   B		
Br?   N)r@   rA   rB   rC   ra   rD   rE   arm_slowrq   ry   r}   r   r   r   r   r   r   r   r   r   r   r   rF   r   r   r   r   r   r   r  rG   r?   r'   r$   rI   rI   J   s   %0N [[3 3 [[/ / [[
  
  [[" " [[/ / [[  [[Q Q [[9 9 [[  [[/ / [[# #,	/#!
 [[UQ	1l2C(D$EF/ G/7 
/++&Hr'   rI   )stringr   numpyr   rD   pandasr   r   r   r   r   r   r	   pandas._testing_testingr1   r   rI   r?   r'   r$   <module>r     s9          58 58p_H _Hr'   