U
|
ý°d> ã @ s² d dl m Z mZ d dlZd dlZd dlZd dlmZ d dlm m
|
Z d dl Z d dl mZmZmZmZmZmZ d dlmZ d dlmZmZ e jZG dd dZ dd Z d
|
d Z dS ) é )ÚdatetimeÚ timedeltaN)ÚPerformanceWarning)Ú CategoricalÚ DataFrameÚSeriesÚ TimestampÚ
|
date_rangeÚoption_context)Ú NumericBlockÚ ObjectBlockc @ s e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dd Z dd Zdd Zdd Zd d Zd d Zd!d" Zd#S )$ÚTestDataFrameBlockInternalsc C sx t dddd}|d }td|i}|d jjd ks6ttj|jd< |d jjd ksVt|jdksdt|d |ksttd S ) NÚ20130101é ú
|
US/Eastern)ÚperiodsÚtzé ÚB)r r ÚD)r r Z_valuesÚfreqÚAssertionErrorÚpdZNaTZiloc)ÚselfZdtiÚtsÚdf© r ú^d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/test_block_internals.pyÚ,test_setitem_invalidates_datetime_index_freq& s zHTestDataFrameBlockInternals.test_setitem_invalidates_datetime_index_freqc C sX t |jtd}t |jtd}t ||¡ t |jtjd}t |jtjd}t ||¡ d S )N©Údtype)r Ú_mgrÚintZ_seriesÚtmÚassert_frame_equalÚnpÚint32)r Ú float_frameZcastedÚexpectedr r r Útest_cast_internals7 s z/TestDataFrameBlockInternals.test_cast_internalsc C s d|d< | ¡ }t|jjdks$t| ¡ }||k s8tt ||¡ d|d< t|jjdks`t| ¡ }|d ksttt|jjdkstd S )Nç @ÚEr g @ÚFr )Z _consolidateÚlenr! Úblocksr r# r$ Z_consolidate_inplace)r r' Z consolidatedZreconsZ return_valuer r r Útest_consolidate@ s z,TestDataFrameBlockInternals.test_consolidatec C s4 | ¡ }ttdtdD ]}t||t|< qd S )NÚAÚZ)ÚcopyÚrangeÚordÚchr)r r' ÚframeÚletterr r r Útest_consolidate_inplaceQ s z4TestDataFrameBlockInternals.test_consolidate_inplacec C s¦ |rBt jtdd d|jd< W 5 Q R X |jd dk ¡ s>td S d|jd< |jd dk ¡ sbtd|d< |d }d|jd< |jd dk ¡ rt|dk ¡ s¢td S )Nz read-only©Úmatché r* r+ é é )ÚpytestÚraisesÚ
|
ValueErrorÚvaluesÚallr )r r' Úusing_copy_on_writeÚcolr r r Útest_modify_valuesX s
|
|
z.TestDataFrameBlockInternals.test_modify_valuesc C s< d|d< |j ¡ }d||dk< d||dk< t ||j ¡ d S )Nr* r+ é r )rA r2 r# Zassert_almost_equal)r r' r( r r r Útest_boolean_set_unconsl s
|
|
z3TestDataFrameBlockInternals.test_boolean_set_unconsc C s t ddgi}|d }tt dgtj¡dd}t ||¡ t ddgi}|d }tt dgtj¡dd}t ||¡ t dtddddgi}|d }tt tddddgtj ¡dd}t ||¡ t dd dgi}|d }tt tj
|
dgtj ¡dd}t ||¡ t ddd gi}|d }tt dd gtj ¡dd}t ||¡ t dd
|
d gi}|d }tt d
|
d gtj ¡dd}t ||¡ t dd
|
d gi}|d }tt d
|
d gtj ¡dd}t ||¡ t dd
|
dgi}|d }tt d
|
dgtj ¡dd}t ||¡ t ddd gi}|d }tt dtj
|
gtj ¡dd}t ||¡ t dd
|
d gi}|d }tt d
|
tj
|
gtj ¡dd}t ||¡ t dd ddd gi}|d }tt d ddd gtj ¡dd}t ||¡ t dd dtdddd gi}|d }tt d dtdddd gtj ¡dd}t ||¡ d S )Nr0 l ÿÿÿÿ )Únamel iÕ r Tg ð?rF y ð? @r g @g @iÖ ) r r r% ZasarrayÚint64r# Úassert_series_equalZuint64r Zobject_ÚnanZfloat_Zcomplex_)r r Úresultr( r r r Ú test_constructor_with_convertu sl ÿ ÿz9TestDataFrameBlockInternals.test_constructor_with_convertc C sð t dddtjt dddgt dddt dddt dddgg}t|}|j}tddi}t ¡ |d< tddd |d
|
< |d jd kst |d
|
jd kst |j}tt d ¡gd t d¡t d¡t d¡g t
|
dddd
|
g d}t ||¡ d S )NéÑ r r; rF iÐ r zdatetime64[ns]r )ÚdaysÚsecondsr úM8[ns]zm8[ns]Úfloat64é Úobjectztimedelta64[ns]ÚABCDZfoo©Úindex) r r% rK r Údtypesr Únowr r r Úlistr# rJ )r Úfloat_string_frameÚdatar rL r( r r r Ú test_construction_with_mixed¸ s* þ ýÿù z8TestDataFrameBlockInternals.test_construction_with_mixedc C sî t jdddgdd}ttdd}||d< tdtjddd d
|
itdd}t |d ¡ |¡ tt d t
|
d dd d ¡dtdd}|j d dkst |j d d ksªt ttdd}t d¡|d< t jdddgdd|d< t ||¡ d S )Nr rF r ztimedelta64[s]r rV r0 z00:00:01Ús)r r r )r zM8[s])Údt1Údt2r_ rQ r` z
|
2013-01-01z
|
2013-01-02z
|
2013-01-03z datetime64[D])r% Úarrayr r3 r Ztimedelta_ranger# Zassert_numpy_array_equalZto_numpyr r ÚastyperX r Z
|
datetime64r$ )r Úarrr r( r r r Ú"test_construction_with_conversions× s. ÿþú ÿ
|
z>TestDataFrameBlockInternals.test_construction_with_conversionsc C s¢ dd }d}t jt|d |dddg W 5 Q R X t jtdd |d W 5 Q R X t jtdd |d
|
W 5 Q R X d }t jt|d |d W 5 Q R X d S ) Nc S s2 t t tdddddfd¡}t|dddg| d S )
|
NrN r Zaaé é r0 r ÚC)r\ Úcolumnsr )rZ Ú itertoolsÚrepeatr r )r r\ r r r Úfü s zGTestDataFrameBlockInternals.test_constructor_compound_dtypes.<locals>.fz@compound dtypes are not implemented in the DataFrame constructorr9 )r0 z datetime64[h])r Ústr)rg r& zargument must berI rR zD^Unknown datetime string format, unable to parse: aa, at position 0$rQ )r> r? ÚNotImplementedErrorÚ TypeErrorr@ )r rk Úmsgr r r Ú test_constructor_compound_dtypesø s z<TestDataFrameBlockInternals.test_constructor_compound_dtypesc C sP t }t |¡}t ||¡ |jj t |¡}t| t |¡}t ||¡ d S ©N)r r# Zround_trip_pickler$ r! ÚndimÚrepr)r r[ Ztimezone_frameZ empty_frameZ unpickledr r r Ú test_pickle s
|
|
|
z'TestDataFrameBlockInternals.test_picklec C sÎ t t dddddg¡t dddddg¡dddd dgd
|
}|j}|j|_| d ¡}| d ¡}d |j_|j }|j|_| d ¡}| d ¡}d|j_|j|_|j|_ t
|
t |j¡|j¡ t
|
t |j ¡|j¡ d S )Nz2012-06-21 00:00z2012-06-23 07:00z2012-06-23 16:30z2012-06-25 08:00z2012-06-26 12:00z2012-06-27 08:00éM éA r )ÚstartingÚendingZmeasurer ÚUTCrw rx ) r r Ú to_datetimerw rA rW Z tz_localizeZ
|
tz_convertrH rx r# Zassert_index_equalZ DatetimeIndex)r r Z ser_startingZ
|
ser_endingr r r Útest_consolidate_datetime64 sD ûÿ ûÿ íÿ
|
|
|
|
z7TestDataFrameBlockInternals.test_consolidate_datetime64c C s |j r
|
t|j std S rq )Z_is_mixed_typer )r r' r[ r r r Útest_is_mixed_typeM s
|
z.TestDataFrameBlockInternals.test_is_mixed_typec
|
C sÎ t dd º ttj d¡ddd}t| |d d¡|d< |rdt ¡ tj|d d < W 5 Q R X ntj|d d < t| | ¡ }|d ¡ }|rªt
|
|d d ¡rÀt nt
|
|d d ¡sÀt W 5 Q R X d S )
|
NZchained_assignment)rS rS )ÚaÚbÚcÚd)Úerk ÚgÚh)rW rh r rT r r ) r
|
r r% Úrandomrs rb r# Z raises_chained_assignment_errorÚNaNÚsumr Úisnar )r rC ÚYrL Úexpr r r Ú test_stale_cached_series_bug_473Q s$
|
ý
|
z<TestDataFrameBlockInternals.test_stale_cached_series_bug_473c
|
C sî t ddgd}tj|d< i }t t¡d t|jD ]R\}}tddD ]>}||krbd||< tj||< |rv||j ||f< qD||| |< qDq2W 5 Q R X d}t
|
|j t || ¡|gf }t
|
|j t || ¡|gf } || krädksên t d S )Nr r rV éd éÈ )r r% rK r# Zassert_produces_warningr Ú enumeraterW r3 Úlocr- r r r )
|
r rC r ZwasColÚiÚdtrD ZmyidÚfirstÚsecondr r r Ú$test_strange_column_corruption_issueh s
|
|
z@TestDataFrameBlockInternals.test_strange_column_corruption_issuec C sT t dddgj}td|i}tddddgi}t ||¡ t|jjd tsPt d S )Nr rF r r0 r )
|
r ra r r# r$ Ú
|
isinstancer! r. r r )r rc rL r( r r r Ú test_constructor_no_pandas_array s
|
z<TestDataFrameBlockInternals.test_constructor_no_pandas_arrayc
|
C s t ddddgddddgd }tj tjddd dgtd
|
¡|d< t ddddgddddgtj tjddd dgtd
|
¡d }t|d jj d t
|
kst t|d jj d t
|
ks²t t ||¡ d S ) Nr rF r rS r} r~ r r )r} r~ r )r} r~ r r )r r ZarraysZ PandasArrayr% ra rT Útyper! r. r r r# r$ )r r Zdf2r r r Ú!test_add_column_with_pandas_array s "
|
|
ýÿ z=TestDataFrameBlockInternals.test_add_column_with_pandas_arrayN)Ú__name__Ú
|
__module__Ú __qualname__r r) r/ r8 rE rG rM r] rd rp rt r{ r| r r r r r r r r r % s" C !- r c C sd t dtddd gddi}|d jddd t|jjd jtsDt| s`| ¡
|
¡
|
¡ dks`td S ) Nr} r rF Úcategoryr T)Zinplacer ) r r Zfillnar r! r. rA r r Zisnullr )rC r r r r Ú+test_update_inplace_sets_valid_block_values s
|
r c C s t } tdgtd| d< tdgtd| d< | d dk | | d dk d| jd< t dgdgdtd}t | |¡ | jd dkstd S ) Nr} r Úcol1r Úcol2r0 )r r )r r )r r rT Úatr# r$ r )r r( r r r Ú$test_nonconsolidated_item_cache_take§ s
|
r ) r r ri Únumpyr% r> Z pandas.errorsr Z pandas.util._test_decoratorsÚutilZ_test_decoratorsÚtdZpandasr r r r r r r
|
Zpandas._testingZ_testingr# Zpandas.core.internalsr r Z skip_array_manager_invalid_testZ
|
pytestmarkr r r r r r r Ú<module> s u
|
