
    I j                     p    S SK rS SKrS SKJrJr  S SKJr   " S S5      r	 " S S\	5      r
 " S S\	5      rg)	    N)	DataFrameSeriesc                       \ rS rSr\R
                  S 5       rS rS rS r	\R                  R                  SSS/5      S	 5       rS
 rS rSrg)SharedSetAxisTests   c                     [        S5      e)NzImplemented by subclasses)NotImplementedError)selfs    y/root/GenerationalWealth/GenerationalWealth/venv/lib/python3.13/site-packages/pandas/tests/frame/methods/test_set_axis.pyobjSharedSetAxisTests.obj   s    !"=>>    c                     [        S5      S [        U5       nUR                  5       nX#l        UR	                  USS9n[
        R                  " X45        g )Nabcdr   axislistlencopyindexset_axistmassert_equalr
   r   	new_indexexpectedresults        r   test_set_axis SharedSetAxisTests.test_set_axis   sD    L3s8,	88:"ia0
)r   c                   ^^^^ [        S5      S [        T5       nTR                  S S  mTR                  5       nX#l        TR                  USS9m[        R                  " UT5        TTLd   eTR                  S:X  a  [        R                  " TT5      (       d   eO3[        UU4S j[        TR                  S   5       5       5      (       d   eTR                  USS9m[        R                  " UT5        TTLd   eTR                  S:X  a  [        R                  " TT5      (       d   eO3[        UU4S j[        TR                  S   5       5       5      (       d   eTR                  U5      m[        R                  " UT5        TR                  S:X  a  [        R                  " TT5      (       d   eg [        UU4S j[        TR                  S   5       5       5      (       d   eg )Nr   r   r      c              3      >#    U  H>  n[         R                  " TR                  S S 2U4   TR                  S S 2U4   5      v   M@     g 7fNr   shares_memoryiloc.0ir   r   s     r   	<genexpr>8SharedSetAxisTests.test_set_axis_copy.<locals>.<genexpr>'   A      ,A   QT!2CHHQTNCC,   AA	c              3      >#    U  H>  n[         R                  " TR                  S S 2U4   TR                  S S 2U4   5      v   M@     g 7fr$   r%   r(   s     r   r+   r,   3   r-   r.   c              3      >#    U  H>  n[         R                  " TR                  S S 2U4   TR                  S S 2U4   5      v   M@     g 7fr$   r%   )r)   r*   origress     r   r+   r,   >   s@      ,A   !Q$1a4AA,r.   )r   r   r'   r   r   r   r   r   ndimr&   allrangeshapeany)r
   r   r   r   r1   r2   r   s    `  @@@r   test_set_axis_copy%SharedSetAxisTests.test_set_axis_copy   s   L3s8,	xx{88:"ia0
&)S   88q=##FC0000 syy|,    
 ia0
&)S   88q=##FC0000 syy|,    
 ll9%
#&88q=##C.... syy|,    r   c                     [        S5      S [        U5       nUR                  5       nX#l        UR	                  U5      n[
        R                  " XC5        g )Nr   r   r   s        r   !test_set_axis_unnamed_kwarg_warns4SharedSetAxisTests.test_set_axis_unnamed_kwarg_warnsC   s@    L3s8,	88:"i(
)r   r      fooc                     [         R                  " [        SS9   UR                  [	        S5      US9  S S S 5        g ! , (       d  f       g = f)NzNo axis namedmatchabcr   )pytestraises
ValueErrorr   r   )r
   r   r   s      r   test_set_axis_invalid_axis_name2SharedSetAxisTests.test_set_axis_invalid_axis_nameM   s2     ]]:_=LLe4L0 >==s	   =
Ac                 |    Sn[         R                  " [        US9   S Ul        S S S 5        g ! , (       d  f       g = f)NzNIndex\(\.\.\.\) must be called with a collection of some kind, None was passedr@   )rC   rD   	TypeErrorr   r
   r   msgs      r   *test_set_axis_setattr_index_not_collection=SharedSetAxisTests.test_set_axis_setattr_index_not_collectionS   s/    % 	 ]]9C0CI 100s   -
;c                    S[        U5       S[        U5      S-
   S3n[        R                  " [        US9   [        R
                  " [        U5      S-
  5      Ul        S S S 5        UR                  S:X  a:  [        R                  " [        SS9   UR                  S S S2   Ul        S S S 5        g g ! , (       d  f       NY= f! , (       d  f       g = f)Nz#Length mismatch: Expected axis has z elements, new values have r"   z	 elementsr@      zLength mismatch)	r   rC   rD   rE   nparanger   r3   columnsrJ   s      r   (test_set_axis_setattr_index_wrong_length;SharedSetAxisTests.test_set_axis_setattr_index_wrong_length\   s     2#c( <"3x!|nI7 	 ]]:S1		#c(Q,/CI 2 88q=z1BC!kk#A#. DC  21 DCs   (B3C3
C
C N)__name__
__module____qualname____firstlineno__rC   fixturer   r   r8   r;   markparametrizerF   rL   rS   __static_attributes__rU   r   r   r   r      s[    ^^? ?*)V* [[VaZ01 11
/r   r   c                   >    \ rS rSr\R
                  S 5       rS rSrg)TestDataFrameSetAxisj   c                 2    [        / SQ/ SQ/ SQS./ SQS9nU$ )N)g?g@gffffff
@)g      @gffffff@g@)g@g      @gffffff@)ABC)i  i  i  )r   )r   )r
   dfs     r   r   TestDataFrameSetAxis.objk   s     !oN$
 	r   c                     [        SS/SS//SS/S9R                  SS	9nUR                  S
S/SS9n[        SS/SS//S
S/SS/S9n[        R                  " X#SS9  g )Nr"   rO   r=      ab)rR   F)allows_duplicate_labelsxyr   )labelsr   )r   rR   )check_flags)r   	set_flagsr   r   assert_frame_equal)r
   re   r   r   s       r   0test_set_axis_with_allows_duplicate_labels_falseETestDataFrameSetAxis.test_set_axis_with_allows_duplicate_labels_falses   s~    AA'#s<FF$) G 
 S#JQ7q!fq!f-c3Z#sT
fEBr   rU   N)	rV   rW   rX   rY   rC   rZ   r   rr   r]   rU   r   r   r_   r_   j   s    ^^ Cr   r_   c                   8    \ rS rSr\R
                  S 5       rSrg)TestSeriesSetAxis~   c                 H    [        [        R                  " S5      / SQSS9nU$ )Nrh   )r"   r=         int64)r   dtype)r   rP   rQ   )r
   sers     r   r   TestSeriesSetAxis.obj   s    RYYq\WE
r   rU   N)rV   rW   rX   rY   rC   rZ   r   r]   rU   r   r   ru   ru   ~   s    ^^ r   ru   )numpyrP   rC   pandasr   r   pandas._testing_testingr   r   r_   ru   rU   r   r   <module>r      s>      \/ \/~C- C(* r   