
    g                     @    d dl Zd dlZd dlmZ d dlmZ  G d d      Zy)    N)	DataFramec                   $    e Zd Zd Zd Zd Zd Zy)TestSwapAxesc                    t        t        j                  j                  d      j	                  d            }d}t        j                  t        |      5  t        j                  |j                  |j                  dd             t        j                  |j                  |j                  dd             d d d        y # 1 sw Y   y xY w)N   
      "'DataFrame.swapaxes' is deprecatedmatchr      )r   nprandomdefault_rngstandard_normaltmassert_produces_warningFutureWarningassert_frame_equalTswapaxesselfdfmsgs      ]/var/www/openai/venv/lib/python3.12/site-packages/pandas/tests/frame/methods/test_swapaxes.pytest_swapaxeszTestSwapAxes.test_swapaxes	   s    ryy,,Q/??HI2''SA!!"$$Aq(9:!!"$$Aq(9: BAAs   A!B??Cc                 "   t        t        j                  j                  d      j	                  d            }d}t        j                  t        |      5  t        j                  ||j                  dd             d d d        y # 1 sw Y   y xY w)Nr   r   r   r   r   )
r   r   r   r   r   r   r   r   r   r   r   s      r   test_swapaxes_noopzTestSwapAxes.test_swapaxes_noop   sc    ryy,,Q/??HI2''SA!!"bkk!Q&78 BAAs   'BBc                 ^   t        t        j                  j                  d      j	                  d            }d}t        j                  t        |      5  d}t        j                  t        |      5  |j                  dd       d d d        d d d        y # 1 sw Y   xY w# 1 sw Y   y xY w)Nr   r   r   r   z)No axis named 2 for object type DataFramer
   )r   r   r   r   r   r   r   r   pytestraises
ValueErrorr   r   s      r   test_swapaxes_invalid_axisz'TestSwapAxes.test_swapaxes_invalid_axis   s{    ryy,,Q/??HI2''SA=Cz5Aq! 6 BA55 BAs$   B#3BB#B 	B##B,c                     t        dddgi      }d}t        j                  t        |      5  |j	                  dd      }d d d        t        j
                  |       ||usJ y # 1 sw Y   &xY w)Nar   r   r   r   index)r   r   r   r   r   r   )r   r   r   results       r   test_round_empty_not_inputz'TestSwapAxes.test_round_empty_not_input   sg    aV}%2''SA[['2F B
b&) BAs   A%%A.N)__name__
__module____qualname__r   r    r%   r*        r   r   r      s    ;9" r/   r   )	numpyr   r"   pandasr   pandas._testing_testingr   r   r.   r/   r   <module>r4      s          r/   