
    bMh                         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dS )    N)		DataFrameDatetimeIndexIndexIntervalIndexSeries	Timestampbdate_range
date_rangetimedelta_rangec                   n   e Zd Zd Zd Zd Zd Zd Ze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	        
                    dddg          ej	        
                    dddg          d                         ZdS )TestTransposec                 P   t          dd          }t          j        |                              dt          j                  t                    }|j        }t          fdt          t                              D                       }t          j        ||           d S )Nz0 Daysz3 Daysc                 ,    i | ]}|||d z            S )    ).0iiis     i/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/pandas/tests/frame/methods/test_transpose.py
<dictcomp>z?TestTranspose.test_transpose_td64_intervals.<locals>.<dictcomp>   s'    GGG1aAAIGGG    )r   r   from_breaksinsertnpnanr   Trangelentmassert_frame_equal)selftdidfresultexpectedr   s        @r   test_transpose_td64_intervalsz+TestTranspose.test_transpose_td64_intervals   s    h11&s++YYr26""r]]GGGGc"ggGGGHH
fh/////r   c                 2   t          g d          }t          |          }t          g dd           }|j                                        j        }|                    d          j        }t          j        ||           t          j        ||           d S )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!   s    Bh///S!!! +;$GGG$((**"&&a&..&
gx000
gx00000r   c                     t          ddd          }t          |          }|j        |j        k                                    sJ |j        }|j        |j        k                                    sJ d S )N2016-04-05 04:30   UTCperiodstzr
   r   dtypesr*   allr   )r"   r1   r$   ress       r   %test_transpose_tzaware_1col_single_tzz3TestTranspose.test_transpose_tzaware_1col_single_tz.   sq    +Q5AAAs^^	SY&++-----d
ci',,.......r   c                     t          ddd          }t          ||d          }|j        |j        k                                    sJ |j        }|j        |j        k                                    sJ d S )Nr6   r7   r8   r9   ABr<   )r"   r1   df3res3s       r   %test_transpose_tzaware_2col_single_tzz3TestTranspose.test_transpose_tzaware_2col_single_tz7   s{    +Q5AAAc,,--
ci',,.....usy(--///////r   c                    t          ddd          }|                    d          }t          ||d          }|j        |j        |j        gk                                    sJ |j        j        t          k                                    sJ t          j	        |j        j        |
                    t                               d S )Nr6   r7   r8   r9   
US/PacificrB   )r
   
tz_convertr   r=   r*   r>   r   objectr    r!   astype)r"   r1   dti2df4s       r   $test_transpose_tzaware_2col_mixed_tzz2TestTranspose.test_transpose_tzaware_2col_mixed_tz@   s    +Q5AAA~~l++c--..
sy$*55::<<<<<&++-----
cegszz&'9'9:::::r   r;   NzAmerica/New_Yorkc                 D   t          ddd|          }t          t          t          |                    t          t          |                    d|          }|j        |j        k    }t          dt          d          |	          }t          j        ||           d S )
N20161101201611304h)r-   r;   abr+   Tab)r,   columns)r
   r   r   r   r   listr    r!   )r"   r;   idxr$   r%   r&   s         r   2test_transpose_preserves_dtindex_equality_with_dstz@TestTranspose.test_transpose_preserves_dtindex_equality_with_dstJ   s     ZdrBBBU3s88__5S??CC3OOOTdSAAA
fh/////r   c                    t          ddd          }|                    d          }t          ||g          }|j        t          k                                    sJ |j        }|j        t          k                                    sJ d S )Nr6   r7   r8   r9   rI   )r
   rJ   r   r=   rK   r>   r   )r"   r1   rM   df2res2s        r   )test_transpose_object_to_tzaware_mixed_tzz7TestTranspose.test_transpose_object_to_tzaware_mixed_tzS   s    +Q5AAA~~l++ d$$
f$))+++++uv%**,,,,,,,r   c                     t          t          j        d          g ddt          j                  }|j        }t          |j        j                  }ddg|_        t          j        ||           d S )Nr7   )l            l           l   
        rB   r)   rC   rD   )	r   r   arangeuint64r   valuesr,   r    r!   r"   r$   r%   r&   s       r   test_transpose_uint64z#TestTranspose.test_transpose_uint64^   ss    )A,,%C%C%CDD)
 
 
 RY[))s
fh/////r   c                    |}|j         }|                                D ]j\  }}|                                D ]P\  }}t          j        |          r#t          j        ||         |                   sJ <|||         |         k    sJ Qkd S )N)r   itemsr   isnan)r"   float_frameframedftrZ   seriescolvalues           r   test_transpose_floatz"TestTranspose.test_transpose_floath   s    g99;; 	4 	4KC$llnn 4 4
U8E?? 48E#JsO444444 E#JsO33333	4	4 	4r   c           	          t          g dg dg dt          dd          dt          g dt          	          
          }|j        }|                                D ]\  }}|j        t          j        k    sJ d S )N)              ?g       @g      @g      @)rq   rr   rq   rr   rq   )foo1foo2foo3foo4foo5z1/1/2009   )r:   )rC   rD   CD)rU   rV   cder)   r+   )	r   r	   r   rK   r   rg   r*   r   object_)r"   mixedmixed_Trm   ss        r   test_transpose_mixedz"TestTranspose.test_transpose_mixedr   s    ......=== Q777	  111@@@
 
 
 'mmoo 	) 	)FC7bj(((((	) 	)r   c                     |j         }d|j        d d ddf<   |r)|j        dd         dk                                    sJ d S |j        dd         dk                                    sJ d S )Nrx   
   )r   ilocrc   r>   )r"   ri   using_copy_on_writerk   s       r   test_transpose_get_viewz%TestTranspose.test_transpose_get_view   s    mAbD 	9&qt,1668888888&qt,1668888888r   c                    t          ddd          }|j                            dd          }t          |          }|j        j        dk    sJ |j        }|j        j        dk    sJ |j        j        d         j        }|r8t          j
        |j        j        d         j        j        |j                  sJ d S t          j
        |j        |j                  sJ d S )	Nz
2016-01-01   rI   r9   r7      r   r   )r
   _datareshaper   _mgrnblocksr   blocksrc   r   shares_memory_ndarray)r"   r   r1   arrr$   r%   rtrips          r   &test_transpose_get_view_dt64tzget_viewz4TestTranspose.test_transpose_get_view_dt64tzget_view   s    q\BBBi1%%s^^w!####{"a''''"1%, 	B#BGN1$5$<$Eu~VVVVVVV#CL%.AAAAAAAr   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        ||           d S )NrU   
2019-12-31r)   r   r   rX   r,   r*   )r   r   rK   r   r    r!   rd   s       r   test_transpose_not_inferring_dtz-TestTranspose.test_transpose_not_inferring_dt   s    i--y/F/FG 	
 
 
 %%y'>'>?@F%	
 
 
 	fh/////r   c                    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        ||           d S )	Nr   r)   rT   r   r   rU   rV   r   )r   r   r   rK   r   r    r!   rd   s       r   ,test_transpose_not_inferring_dt_mixed_blocksz:TestTranspose.test_transpose_not_inferring_dt_mixed_blocks   s    |,,i.E.EFf    --y/F/FG	 
 
 <(()L*A*AB<(()L*A*AB F*
 
 
 	fh/////r   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         sJ +d S )N)r   r   r   r)   )r7      rx   rT   F_CONTIGUOUS)	r   pdarrayr   r   r   rc   r   flags)r"   r   r   r$   r%   blkdatas          r   test_transposezTestTranspose.test_transpose   s     Xiiiv666Xiiiv666 
 
 ;% 	. 	.C'-'7'73:##SZD:n-----	. 	.r   )__name__
__module____qualname__r'   r4   r@   rG   rO   pytestmarkparametrizer[   r_   re   ro   r   tdskip_array_manager_invalid_testr   r   r   r   r   r   r   r   r   r      s       	0 	0 	01 1 1/ / /0 0 0; ; ; [TD*<#=>>0 0 ?>0	- 	- 	-0 0 04 4 4) ) )  '9 9 ('9 'B B ('B0 0 0"0 0 0, [X';<<[X';<<. . =< =<. . .r   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   r   r   <module>r      s         ) ) ) ) ) ) ) ) )    
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
      |. |. |. |. |. |. |. |. |. |.r   