U
|
ý°d ã @ s\ d dl Zd dlZd dlm mZ d dlmZm Z m
|
Z
|
m Z m Z d dl mZ G dd dZdS )é N)Ú DataFrameÚ DatetimeIndexÚ IntervalIndexÚ
|
date_rangeÚtimedelta_rangec @ s e Zd Zdd Zdd Zdd Zdd Zd d
|
Zej
|
d d d g¡dd Z dd Z dd Z dd Zejdd Zejdd Zd S )Ú TestTransposec s^ t dd}t |¡ dtj¡ t }|j}t fddtt D }t
|
||¡ d S )Nz0 Daysz3 Dayséÿÿÿÿc s i | ]}| ||d
qS )é © )Ú.0Úi©Úiir
|
ú`d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/methods/test_transpose.pyÚ
|
<dictcomp> s z?TestTranspose.test_transpose_td64_intervals.<locals>.<dictcomp>) r r Z from_breaksÚinsertÚnpÚnanr ÚTÚrangeÚlenÚtmÚassert_frame_equal)ÚselfZtdiÚdfÚresultÚexpectedr
|
r r Ú test_transpose_td64_intervals s
|
|
z+TestTranspose.test_transpose_td64_intervalsc C sR t tg d}tg dd d}|j ¡ j}|jddj}t ||¡ t ||¡ d S )N©Úindexzdatetime64[ns])ÚdtypeÚfreqr )Zaxis)r r r Úsumr r Zassert_index_equal)r r r Zresult1Zresult2r
|
r
|
r Ú,test_transpose_empty_preserves_datetimeindex s z:TestTranspose.test_transpose_empty_preserves_datetimeindexc C sH t dddd}t|}|j|jk ¡ s*t|j}|j|jk ¡ sDtd S )Nú2016-04-05 04:30é ÚUTC©ZperiodsÚtz©r r Údtypesr ÚallÚAssertionErrorr )r Údtir Úresr
|
r
|
r Ú%test_transpose_tzaware_1col_single_tz( s
|
z3TestTranspose.test_transpose_tzaware_1col_single_tzc C sN t dddd}t||d}|j|jk ¡ s0t|j}|j|jk ¡ sJtd S )Nr$ r% r& r' ©ÚAÚBr) )r r- Zdf3Zres3r
|
r
|
r Ú%test_transpose_tzaware_2col_single_tz1 s
|
z3TestTranspose.test_transpose_tzaware_2col_single_tzc C sh t dddd}| d¡}t||d}|j|j|jgk ¡ s@t|jjtk ¡ sTtt
|
|jj|¡ d S )Nr$ r% r& r' ú
|
US/Pacificr0 ) r Ú
|
tz_convertr r* r r+ r, r Úobjectr r )r r- Údti2Zdf4r
|
r
|
r Ú$test_transpose_tzaware_2col_mixed_tz: s
|
z2TestTranspose.test_transpose_tzaware_2col_mixed_tzr( NzAmerica/New_Yorkc C s` t ddd|d}ttt|tt|d|d}|j|jk}tdtd|d }t ||¡ d S )
|
NZ20161101Z20161130Z4H)r! r( )ÚaÚbr TÚab)r Úcolumns)r r r r r Úlistr r )r r( Úidxr r r r
|
r
|
r Ú2test_transpose_preserves_dtindex_equality_with_dstD s
|
" z@TestTranspose.test_transpose_preserves_dtindex_equality_with_dstc C sZ t dddd}| d¡}t||g}|jtk ¡ s6t|j}|j|j|jgk ¡ sVtd S )Nr$ r% r& r' r4 ) r r5 r r* r6 r+ r, r r )r r- r7 Zdf2Zres2r
|
r
|
r Ú)test_transpose_object_to_tzaware_mixed_tzM s
|
z7TestTranspose.test_transpose_object_to_tzaware_mixed_tzc C s, |j }t|jj }ddg|_t ||¡ d S )Nr1 r2 )r r Úvaluesr r r )r Z uint64_framer r r
|
r
|
r Útest_transpose_uint64X s
|
z#TestTranspose.test_transpose_uint64c C s¨ |}|j }| ¡ D ]P\}}| ¡ D ]>\}}t |¡rLt || | ¡s`tq"||| | ks"tq"qt ¡ \}} t| |d}
|
|
|
j } | ¡ D ]\}} | jtj kstqd S )Nr )
|
r Úitemsr Úisnanr, r ZgetMixedTypeDictr r Zobject_) r Ú float_frameÚframeÚdftr> ZseriesÚcolÚvaluer ÚdataÚmixedZmixed_TÚsr
|
r
|
r Útest_transpose_float^ s
|
z"TestTranspose.test_transpose_floatc C sZ |j }d|jd d
dd
f< |r<|jdd
dk ¡ sVtn|jdd
dk ¡ sVtd S )Né é
|
)r ZilocrA r+ r, )r rE Úusing_copy_on_writerG r
|
r
|
r Útest_transpose_get_viewp s
|
z%TestTranspose.test_transpose_get_viewc C s t dddd}|j dd¡}t|}|jjdks4t|j}|jjdksJt|jjd j }|r|t
|
|jjd j j |j ¡stnt
|
|j |j ¡std S ) Nz
|
2016-01-01é r4 r' r% é r r ) r Ú_dataZreshaper Z_mgrZnblocksr, r ÚblocksrA r Z shares_memoryZ_ndarray)r rP r- Zarrr r Zrtripr
|
r
|
r Ú&test_transpose_get_view_dt64tzget_viewz s z4TestTranspose.test_transpose_get_view_dt64tzget_view)Ú__name__Ú
|
__module__Ú __qualname__r r# r/ r3 r8 ÚpytestÚmarkZ parametrizer? r@ rB rM ÚtdZ skip_array_manager_invalid_testrQ rV r
|
r
|
r
|
r r s
|
|
|
r )Únumpyr rZ Z pandas.util._test_decoratorsÚutilZ_test_decoratorsr\ Zpandasr r r r r Zpandas._testingZ_testingr r r
|
r
|
r
|
r Ú<module> s
|
|
