
    Owg                      p    d dl Zd dlZd dlmZmZ d dlmZ  G d d      Z	 G d de	      Z
 G d de	      Zy)	    N)	DataFrameSeriesc                       e Zd Zej                  d        Zd Zd Zd Zej                  j                  dddg      d        Zd	 Zd
 Zy)SharedSetAxisTestsc                     t        d      )NzImplemented by subclasses)NotImplementedError)selfs    _/var/www/horilla/myenv/lib/python3.12/site-packages/pandas/tests/frame/methods/test_set_axis.pyobjzSharedSetAxisTests.obj   s    !"=>>    c                     t        d      d t        |       }|j                         }||_        |j	                  |d      }t        j                  ||       y )Nabcdr   axislistlencopyindexset_axistmassert_equalr	   r   	new_indexexpectedresults        r
   test_set_axisz SharedSetAxisTests.test_set_axis   sH    L3s8,	88:"ia0
&)r   c                 H   t        d      d t               }j                  d d  j                         }||_        j                  |dd      t        j                  |       usJ |sTj                  dk(  rt        j                        r/J t        fdt        j                  d         D              rJ j                  |dd      t        j                  |       usJ j                  dk(  rt        j                        s/J t        fdt        j                  d         D              sJ j                  |d	      t        j                  |       usJ |rTj                  dk(  rt        j                        sJ t        fd
t        j                  d         D              sVJ j                  dk(  rt        j                        r/J t        fdt        j                  d         D              rJ j                  |d      t        j                  |       j                  dk(  rt        j                        sJ y t        fdt        j                  d         D              sJ y )Nr   r   T)r   r      c              3      K   | ]<  }t        j                  j                  d d |f   j                  d d |f          > y wNr   shares_memoryiloc.0ir   r   s     r
   	<genexpr>z8SharedSetAxisTests.test_set_axis_copy.<locals>.<genexpr>(   s?       $$V[[A%6AG   AAFc              3      K   | ]<  }t        j                  j                  d d |f   j                  d d |f          > y wr!   r"   r%   s     r
   r(   z8SharedSetAxisTests.test_set_axis_copy.<locals>.<genexpr>4   s?         QT!2CHHQTNCr)   r   c              3      K   | ]<  }t        j                  j                  d d |f   j                  d d |f          > y wr!   r"   r%   s     r
   r(   z8SharedSetAxisTests.test_set_axis_copy.<locals>.<genexpr>B   s?       $$V[[A%6AGr)   c              3      K   | ]<  }t        j                  j                  d d |f   j                  d d |f          > y wr!   r"   r%   s     r
   r(   z8SharedSetAxisTests.test_set_axis_copy.<locals>.<genexpr>J   s?         QT!2CHHQTNCr)   )r   c              3      K   | ]<  }t        j                  j                  d d |f   j                  d d |f          > y wr!   r"   )r&   r'   origress     r
   r(   z8SharedSetAxisTests.test_set_axis_copy.<locals>.<genexpr>U   s>         !Q$1a4Ar)   )r   r   r$   r   r   r   r   r   ndimr#   anyrangeshapeall)r	   r   using_copy_on_writer   r   r.   r/   r   s    `   @@@r
   test_set_axis_copyz%SharedSetAxisTests.test_set_axis_copy   ss   L3s8,	xx{88:"iad;
&)S   "xx1}++FC888 "399Q<0   
 iae<
&)S   88q=##FC000 syy|,    ia0
&)S   xx1}''444 "399Q<0   
 XX]''444 syy|,   
 ll95l1
#&88q=##C... syy|,   r   c                     t        d      d t        |       }|j                         }||_        |j	                  |      }t        j                  ||       y )Nr   r   r   s        r
   !test_set_axis_unnamed_kwarg_warnsz4SharedSetAxisTests.test_set_axis_unnamed_kwarg_warnsZ   sD    L3s8,	88:"i(
)r   r      fooc                     t        j                  t        d      5  |j                  t	        d      |       d d d        y # 1 sw Y   y xY w)NzNo axis namedmatchabcr   )pytestraises
ValueErrorr   r   )r	   r   r   s      r
   test_set_axis_invalid_axis_namez2SharedSetAxisTests.test_set_axis_invalid_axis_named   s;     ]]:_= 	1LLe4L0	1 	1 	1s   AAc                 v    d}t        j                  t        |      5  d |_        d d d        y # 1 sw Y   y xY w)NzNIndex\(\.\.\.\) must be called with a collection of some kind, None was passedr<   )r?   r@   	TypeErrorr   r	   r   msgs      r
   *test_set_axis_setattr_index_not_collectionz=SharedSetAxisTests.test_set_axis_setattr_index_not_collectionj   s8    % 	 ]]9C0 	CI	 	 	s   /8c                    dt        |       dt        |      dz
   d}t        j                  t        |      5  t	        j
                  t        |      dz
        |_        d d d        |j                  dk(  r<t        j                  t        d      5  |j                  d d d   |_        d d d        y y # 1 sw Y   UxY w# 1 sw Y   y xY w)Nz#Length mismatch: Expected axis has z elements, new values have r   z	 elementsr<      zLength mismatch)	r   r?   r@   rA   nparanger   r0   columnsrE   s      r
   (test_set_axis_setattr_index_wrong_lengthz;SharedSetAxisTests.test_set_axis_setattr_index_wrong_lengths   s     2#c( <"3xzl)5 	 ]]:S1 	0		#c(Q,/CI	0 88q=z1BC /!kk#A#./ / 	0 	0/ /s   'B5C5B>C
N)__name__
__module____qualname__r?   fixturer   r   r6   r8   markparametrizerB   rG   rM    r   r
   r   r      s\    ^^? ?*@D* [[VaZ01 11
/r   r   c                   0    e Zd Zej                  d        Zy)TestDataFrameSetAxisc                 8    t        g dg dg ddg d      }|S )N)g?g@gffffff
@)g      @gffffff@g@)g@g      @gffffff@)ABC)i  i  i  )r   )r   )r	   dfs     r
   r   zTestDataFrameSetAxis.obj   s     !oN$
 	r   NrN   rO   rP   r?   rQ   r   rT   r   r
   rV   rV      s    ^^ r   rV   c                   0    e Zd Zej                  d        Zy)TestSeriesSetAxisc                 L    t        t        j                  d      g dd      }|S )N   )r   r9         int64)r   dtype)r   rJ   rK   )r	   sers     r
   r   zTestSeriesSetAxis.obj   s    RYYq\WE
r   Nr\   rT   r   r
   r^   r^      s    ^^ r   r^   )numpyrJ   r?   pandasr   r   pandas._testing_testingr   r   rV   r^   rT   r   r
   <module>rj      s<      s/ s/l- * r   