
    Owg                     z    d dl Zd dlZd dlmc mZ d dlZd dlm	Z	m
Z
mZmZmZmZmZmZmZ d dlmZ  G d d      Zy)    N)		DataFrameDatetimeIndexIndexIntervalIndexSeries	Timestampbdate_range
date_rangetimedelta_rangec                   h   e Zd Zd Zd Zd Zd Zd Zej                  j                  dddg      d	        Zd
 Zd Zd Zd Zej"                  d        Zej"                  d        Zd Zd Zej                  j                  dddg      ej                  j                  dddg      d               Zy)TestTransposec           
      N   t        dd      }t        j                  |      }|j                  dt        j
                        }t        |      }|j                  }t        t        t        |            D ci c]  }||||dz     c}      }t        j                  ||       y c c}w )Nz0 Daysz3 Days   )r   r   from_breaksinsertnpnanr   Trangelentmassert_frame_equal)selftdiiidfresultiexpecteds          `/var/www/horilla/myenv/lib/python3.12/site-packages/pandas/tests/frame/methods/test_transpose.pytest_transpose_td64_intervalsz+TestTranspose.test_transpose_td64_intervals   s    h1&&s+YYr266"r]c"gG1aAA.GH
fh/ Hs   4B"c                 *   t        g d      }t        |      }t        g dd       }|j                  j                         j                  }|j                  d      j                  }t        j                  ||       t        j                  ||       y )NzM8[ns]dtypeindexzdatetime64[ns])r%   freqr   )axis)r   r   r   sumr'   r   assert_index_equal)r   dtir   r    result1result2s         r!   ,test_transpose_empty_preserves_datetimeindexz:TestTranspose.test_transpose_empty_preserves_datetimeindex!   sq    Bh/S! +;$G$$((*""&&a&.&&
gx0
gx0    c                     t        ddd      }t        |      }|j                  |j                  k(  j	                         sJ |j
                  }|j                  |j                  k(  j	                         sJ y )N2016-04-05 04:30   UTCperiodstzr
   r   dtypesr%   allr   )r   r,   r   ress       r!   %test_transpose_tzaware_1col_single_tzz3TestTranspose.test_transpose_tzaware_1col_single_tz.   sa    +Q5As^		SYY&++---dd

cii',,...r0   c                     t        ddd      }t        ||d      }|j                  |j                  k(  j	                         sJ |j
                  }|j                  |j                  k(  j	                         sJ y )Nr2   r3   r4   r5   ABr8   )r   r,   df3res3s       r!   %test_transpose_tzaware_2col_single_tzz3TestTranspose.test_transpose_tzaware_2col_single_tz7   sg    +Q5Ac,-

cii',,...uusyy(--///r0   c                    t        ddd      }|j                  d      }t        ||d      }|j                  |j                  |j                  gk(  j                         sJ |j                  j                  t        k(  j                         sJ t        j                  |j                  j                  |j                  t                     y )Nr2   r3   r4   r5   
US/Pacificr>   )r
   
tz_convertr   r9   r%   r:   r   objectr   r   astype)r   r,   dti2df4s       r!   $test_transpose_tzaware_2col_mixed_tzz2TestTranspose.test_transpose_tzaware_2col_mixed_tz@   s    +Q5A~~l+c-.

syy$**55::<<<&++---
ceeggszz&'9:r0   r7   NzAmerica/New_Yorkc                    t        ddd|      }t        t        t        |            t        t        |            d|      }|j                  |j                  k(  }t        dt        d      |	      }t        j                  ||       y )
N20161101201611304h)r(   r7   abr&   Tab)r'   columns)r
   r   r   r   r   listr   r   )r   r7   idxr   r   r    s         r!   2test_transpose_preserves_dtindex_equality_with_dstz@TestTranspose.test_transpose_preserves_dtindex_equality_with_dstJ   sh     ZdrBU3s8_5S?C3OTdSA
fh/r0   c                     t        ddd      }|j                  d      }t        ||g      }|j                  t        k(  j                         sJ |j                  }|j                  t        k(  j                         sJ y )Nr2   r3   r4   r5   rE   )r
   rF   r   r9   rG   r:   r   )r   r,   rI   df2res2s        r!   )test_transpose_object_to_tzaware_mixed_tzz7TestTranspose.test_transpose_object_to_tzaware_mixed_tzS   sn    +Q5A~~l+ d$

f$))+++uuv%**,,,r0   c                     t        t        j                  d      g ddt        j                        }|j                  }t        |j
                  j                        }ddg|_        t        j                  ||       y )Nr3   )l            l           l   
        r>   r$   r?   r@   )	r   r   arangeuint64r   valuesr'   r   r   r   r   r   r    s       r!   test_transpose_uint64z#TestTranspose.test_transpose_uint64^   s\    ))A,%CD))
 RYY[[)s
fh/r0   c                     |}|j                   }|j                         D ]]  \  }}|j                         D ]E  \  }}t        j                  |      rt        j                  ||   |         r7J |||   |   k(  rEJ  _ y )N)r   itemsr   isnan)r   float_frameframedftrV   seriescolvalues           r!   test_transpose_floatz"TestTranspose.test_transpose_floath   s}    gg99; 	4KC$lln 4
U88E?88E#JsO444 E#JsO333	4	4r0   c           	          t        g dg dg dt        dd      dt        g dt        	      
      }|j                  }|j                         D ]$  \  }}|j                  t        j                  k(  r$J  y )N)              ?g       @g      @g      @)rm   rn   rm   rn   rm   )foo1foo2foo3foo4foo5z1/1/2009   )r6   )r?   r@   CD)rQ   rR   cder$   r&   )	r   r	   r   rG   r   rc   r%   r   object_)r   mixedmixed_Tri   ss        r!   test_transpose_mixedz"TestTranspose.test_transpose_mixedr   sk    ..= Q7	 1@
 ''mmo 	)FC77bjj(((	)r0   c                     |j                   }d|j                  d d ddf<   |r#|j                  dd dk7  j                         sJ y |j                  dd dk(  j                         sJ y )Nrt   
   )r   ilocr_   r:   )r   re   using_copy_on_writerg   s       r!   test_transpose_get_viewz%TestTranspose.test_transpose_get_view   sl    mmAbD&&q,166888&&q,166888r0   c                 0   t        ddd      }|j                  j                  dd      }t        |      }|j                  j
                  dk(  sJ |j                  }|j                  j
                  dk(  sJ |j                  j                  d   j                  }|rNt        j                  |j                  j                  d   j                  j                  |j                        sJ y t        j                  |j                  |j                        sJ y )	Nz
2016-01-01   rE   r5   r3      r   r   )r
   _datareshaper   _mgrnblocksr   blocksr_   r   shares_memory_ndarray)r   r   r,   arrr   r   rtrips          r!   &test_transpose_get_view_dt64tzget_viewz4TestTranspose.test_transpose_get_view_dt64tzget_view   s    q\Bii1%s^ww!###{{""a'''""1%,,##BGGNN1$5$<$<$E$Eu~~VVV##CLL%..AAAr0   c                     t        dt        d      t        d      git              }|j                  }t        t        d      t        d      ggddgdgt              }t	        j
                  ||       y )NrQ   
2019-12-31r$   r   r   rT   r'   r%   )r   r   rG   r   r   r   r`   s       r!   test_transpose_not_inferring_dtz-TestTranspose.test_transpose_not_inferring_dt   su    i-y/FG 	
 %y'>?@F%	
 	fh/r0   c                 T   t        t        t        d      t        d      gt              t        d      t        d      gd      }|j                  }t        t        d      t        d      gt        d      t        d      ggddgddgt              }t        j                  ||       y )	Nr   r$   rP   r   r   rQ   rR   r   )r   r   r   rG   r   r   r   r`   s       r!   ,test_transpose_not_inferring_dt_mixed_blocksz:TestTranspose.test_transpose_not_inferring_dt_mixed_blocks   s    |,i.EFf  -y/FG	
 <()L*AB<()L*AB F*
 	fh/r0   dtype1Int64Float64dtype2c                 :   t        t        j                  g d|      t        j                  g d|      d      }|j                  }|j                  j
                  D ]:  }||k(  r|j                  j                  n|j                  }|j                  d   r:J  y )N)r   r   r   r$   )r3      rt   rP   F_CONTIGUOUS)	r   pdarrayr   r   r   r_   r   flags)r   r   r   r   r   blkdatas          r!   test_transposezTestTranspose.test_transpose   s     XXiv6XXiv6
 ;;%% 	.C'-'73::##SZZD::n---	.r0   )__name__
__module____qualname__r"   r/   r<   rC   rK   pytestmarkparametrizerW   r[   ra   rk   r~   tdskip_array_manager_invalid_testr   r   r   r   r    r0   r!   r   r      s    	01/0; [[TD*<#=>0 ?0	-04)  ''9 (9 ''B (B0"0, [[X';<[[X';<. = =.r0   r   )numpyr   r   pandas.util._test_decoratorsutil_test_decoratorsr   pandasr   r   r   r   r   r   r   r	   r
   r   pandas._testing_testingr   r   r   r0   r!   <module>r      s4      ) ) 
 
 
 |. |.r0   