U
|
ý°d}È ã @ sÚ d dl mZm Z d dlZd dlZd dlZd dlZd dlmZ d dl m
|
Z
|
d dl m m Z d dlmZ d dlZd dlmZmZmZmZmZmZmZmZmZ d dl m Z d dl m m!Z" d dl#m$Z$m%Z%m&Z& d dl'm(Z(m)Z)m*Z* d d l+m,Z,m-Z- ej.Z/ej0e-e*gd
|
d d Z1ej0d d Z2dd Z3dd Z4dZ5dd e-fddZ6d0ddZ7d1ddZ8ej0dd Z9G d d d Z:G d d d Z;d d! Z<G d"d# d#Z=G d$d% d%Z>G d&d' d'Z?G d(d) d)Z@d*d+ ZAd,d- ZBd.d/ ZCdS )2é )ÚdateÚdatetimeN)ÚBlockPlacement)ÚIS64)Ú is_scalar) Ú CategoricalÚ DataFrameÚ DatetimeIndexÚIndexÚ IntervalIndexÚSeriesÚ TimedeltaÚ TimestampÚ period_range)Ú DatetimeArrayÚ SparseArrayÚTimedeltaArray)Ú BlockManagerÚSingleBlockManagerÚ
|
make_block)Úensure_block_shapeÚ new_block©Úparamsc C s | j S )zS
|
Fixture to test both the internal new_block and pseudo-public make_block.
|
©Úparam)Úrequest© r ú\d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/internals/test_internals.pyÚ block_maker3 s r c C s t dS )Nza: f8; b: object; c: f8; d: object; e: f8;f: bool; g: i8; h: complex; i: datetime-1; j: datetime-2;k: M8[ns, US/Eastern]; l: M8[ns, CET];)Ú
|
create_mgrr r r r Úmgr; s ÿr! c C sX t | j|j¡ | j|jks tt| jts0tt|jts@tt | jj|jj¡ d S ©N) ÚtmÚassert_numpy_array_equalÚvaluesÚdtypeÚAssertionErrorÚ
|
isinstanceÚmgr_locsr Úas_array)ÚleftÚrightr r r Úassert_block_equalD s
|
r- c C s: t | d ¡}t jjj|| |jfdt| d d ¡ S )Nr )r é )ÚxÚshapeÚstrides)ÚnpÚarangeÚlibZ stride_tricksZ
|
as_stridedÚitemsizeÚlenÚcopy)r0 Úarrr r r Úget_numeric_matL s ÿr9 é
|
c C sX t |}t|}|dkr tf}|f| }t|}| dkrJ| | ¡| }nü| dkrhd| | ¡| }nÞ| dkrt dd | ¡ | D |¡}n´| dkr®tj|tj d }n| d
|
krÈ|d d ¡}n~|
|
d ¡r@t d| ¡} | dk söt d| | ¡ d }
|
|dkst dtt t¡d |
|
dj}t|t|d}n| dkrZ|d d¡}nì| dkrtddddddddddg
|
}nÄ| d krªtd d d d d d d d d d g
|
}n| d!kr6|d" d#krÆttd$d% |dd"
D sæt | d&¡rútj} nd'} t| | ddd| dd(| d)g
|
| d*}|j ¡ } | |d 7 } ntd+| d,|||t|d-S ).a
|
Supported typestr:
|
|
* float, f8, f4, f2
|
* int, i8, i4, i2, i1
|
* uint, u8, u4, u2, u1
|
* complex, c16, c8
|
* bool
|
* object, string, O
|
* datetime, dt, M8[ns], M8[ns, tz]
|
* timedelta, td, m8[ns]
|
* sparse (SparseArray with fill_value=0.0)
|
* sparse_na (SparseArray with fill_value=np.nan)
|
* category, category2
|
|
N)ÚfloatÚf8Zf4Úf2ÚintÚi8Úi4Úi2Úi1ZuintÚu8Zu4Úu2Úu1)ÚcomplexZc16Zc8y ð?)ÚobjectÚstringÚOc S s g | ]}d |dqS )ÚAÚdr )Ú.0Úir r r Ú
|
<listcomp>
s z create_block.<locals>.<listcomp>)ÚbÚbool©r& )r ÚdtúM8[ns]g eÍÍArS zM8[nszM8\[ns,\s*(\w+\/?\w*)\]zincompatible typestr -> r r. z)must have only 1 num items for a tz-awarei Ê;)Útz©Úndim)Ú timedeltaÚtdúm8[ns]rY )Úcategoryé é é )Z category2ÚarO ÚcrK )ÚsparseZ sparse_naéÿÿÿÿr: c s s | ]}|d kV qdS )r. Nr )rL Úsr r r Ú <genexpr> s z create_block.<locals>.<genexpr>Z_naç é é ©Ú
|
fill_valuezUnsupported typestr: "ú"©Ú placementrV ) r r6 ÚNr9 Úastyper2 ÚreshapeÚravelÚonesÚbool_Ú
|
startswithÚreÚsearchr' Úgroupsr r3 Ú_datar r ÚNotImplementedErrorÚallÚendswithÚnanr Z sp_valuesÚviewÚ
|
ValueError) Útypestrrk Ú
|
item_shapeÚ
|
num_offsetÚmakerZ num_itemsr0 Úmatr% ÚmrT rh r8 r r r Ú create_blockV sX
|
"
|
|
|
|
þ
|
r c C s0 |d kr t }tt| td|ddtt |¡S )Nr r )rk r~ )rl r r Úslicer
|
r2 r3 )r} Únum_rowsr r r Úcreate_single_mgr® s þr c C sF |dkrt f}d}g }i }| d¡D ] }| ¡ }t|s:q$| d¡ddd
\}}| ¡ }| ¡ d¡}| |¡ tt t|¡| }z|| |¡ W n t k
|
r¶ |||< Y nX |t|7 }q$t
|
|}g } d}
|
| ¡ D ]8\}}| d¡d } | t | |||
|
d¡ |
|
t|7 }
|
qÞt| d d
|
d } tt| |gd d |D S )aU
|
Construct BlockManager from string description.
|
|
String description syntax looks similar to np.matrix initializer. It looks
|
like this::
|
|
a,b,c: f8; d,e,f: i8
|
|
Rules are rather simple:
|
|
* see list of supported datatypes in `create_block` method
|
* components are semicolon-separated
|
* each component is `NAME,NAME,NAME: DTYPE_ID`
|
* whitespace around colons & semicolons are removed
|
* components with same DTYPE_ID are combined into single block
|
* to force multiple blocks with same dtype, use '-SUFFIX'::
|
|
'a:f8-1; b:f8-2; c:f8-foobar'
|
|
Nr ú;ú:r[ ú,ú-)r~ r c S s
|
| j d S ©Nr )r) )rO r r r Ú<lambda>ð ó z create_mgr.<locals>.<lambda>)Úkeyc S s g | ]}t t |¡qS r )r
|
r2 r3 )rL Únr r r rN ó s z create_mgr.<locals>.<listcomp>)rl ÚsplitÚstripr6 Ú partitionÚextendÚlistr2 r3 ÚKeyErrorr
|
ÚitemsÚappendr Úsortedr Útuple) Údescrr~ ÚoffsetZ mgr_itemsZblock_placementsrK ÚnamesZblockstrrk Úblocksr r} Zsblocksr r r r ¸ sL
|
ÿÿþr c C s t ddddgS )Nr; r r[ r] )r r r r r Úfblock÷ s r c @ s e Zd Zdd Zej dddddggdd ggd
|
d d ggd dggg¡dd Zdd Zdd Z dd Z
|
dd Z dd Z dd Z d S ) Ú TestBlockc C s t ddg}|jtjks td S )Nr@ r )r r& r2 Úint32r' )ÚselfZ
|
int32blockr r r Útest_constructorý s zTestBlock.test_constructorz typ, datar; r r[ r] rF é rG r. r\ rP re c C s t ||}tt |¡| d S r" )r r- r# Úround_trip_pickle)r¡ ÚtypÚdataÚblkr r r Ú test_pickle s
|
|
zTestBlock.test_picklec C s6 t |jtstt |jjtjdddgtj d¡ d S )Nr r[ r] rQ )
|
r( r) r r' r# r$ r* r2 ÚarrayÚintp©r¡ r r r r Ú test_mgr_locs s
|
ÿzTestBlock.test_mgr_locsc C s> |j |jj kst|j|jjks$tt|t|jks:td S r" )r0 r% r' r& r6 r« r r r Ú
|
test_attrs s zTestBlock.test_attrsc C s" | ¡ }||k stt|| d S r" )r7 r' r- )r¡ r Zcopr r r Ú test_copy s zTestBlock.test_copyc C sÄ | ¡ }|j}| d¡d }|j|ks*t||k s6tt |jjtjddgtj d¡ |j
|
d dk ¡ rlt|j
|
d dk ¡ st| ¡ }|j}| d¡}t |dksªt|j|ks¸tt |d jjtjdgtj d¡ t |d jjtjdgtj d¡ |j
|
d dk ¡ rt|d j
|
d dk ¡ s0t| ¡ }| d¡}t |dksTtt |d jjtjddgtj d¡ |d j
|
d dk ¡ st| ¡ }t jtd d | d¡ W 5 Q R X d S )Nr r[ r] rQ r. ©Úmatchr\ )r7 r) Údeleter' r# r$ r* r2 r© rª r% rx r6 ÚpytestÚraisesÚ
|
IndexError)r¡ r ZnewbZlocsÚnbr r r Ú test_delete sJ ÿ
|
|
ÿ
|
ÿ
|
|
ÿ zTestBlock.test_deletec C sö t jddd dd¡ d¡}t|}|jjd }t|jt s@t
|
| d¡}t |d ksZt
|
t|d jt snt
|
t|d jt st
|
t| d
|
¡}|jjd }t|jt s¬t
|
| dd g¡}t |d ksÊt
|
t|d jt sÞt
|
t|d jt sòt
|
d S ) Né r? rQ re r] rY r r. r[ rS r\ )r2 r3 rn r{ r Ú_mgrr r( r% r r' r± r6 r )r¡ r8 Údfr§ rµ r r r Útest_delete_datetimelikeH s
|
z"TestBlock.test_delete_datetimelikec C sº t j dd¡}t|dddgdd}| ¡ }d|d d
< |jdk ¡ sHtt|dksXtt|dg dgddt|dg dgddt|dg dgddg}t ||D ]\}}t
|
|| q¢d S ) Nr\ r] r. rf r[ rj iñØÿÿr ) r2 ÚrandomÚrandnr Ú_splitr% rx r' r6 Úzipr- )r¡ r% r§ ÚresultÚexpectedÚresÚexpr r r Ú
|
test_split] s ýzTestBlock.test_splitN)Ú__name__Ú
|
__module__Ú __qualname__r¢ r² ÚmarkÚ parametrizer¨ r¬ r r® r¶ rº rà r r r r r ü s
|
üþ
|
*r c @ sê e Zd Zdd Zdd Zdd Zej ddd g¡d
|
d Z d d Z
|
dd Z dd Z dd Z dd Zdd Zdd Zej ddejfd ejfg¡d d Zej dd ejfd ejfd!ejfg¡d"d# Zd$d% Zd&d' Zej d(d)d*d+d,d-g¡d.d/ Zd0d1 Z d2d3 Z d4d5 Z ej dd6d7d8d9d:d;d<d=d>d?d@dAdBdCdDdEdFdGg¡dHdI Z dJdK Z dLdM Z!dNdO Z"dPdQ Z#dRdS Z$ej ddTdUg¡dVdW Z%ej ddXdYdZd[d\g¡d]d^ Z&d_d` Z'ej dadbdcdbdddegdfg¡dgdh Z(didj Z)dkdl Z*dmS )nÚTestBlockManagerc C s* t d}|jdkstt|dks&td S )Nza,b,c: f8-1; d,e,f: f8-2r[ rf )r Únblocksr' r6 ©r¡ r! r r r r r s zTestBlockManager.test_attrsc C s¶ t d}|j|j }}tt dg¡|d _tt dg¡|d _d}tjt |d t
|
||}| ¡ W 5 Q R X tt dg¡|d _tt dg¡|d _t
|
||}| d¡ d S )Nz a:bool; a: f8r r. zGaps in blk ref_locsr¯ ) r Úaxesr r r2 r© r) r² r³ r' r Z_rebuild_blknos_and_blklocsÚiget)r¡ Ztmp_mgrrÌ r Úmsgr! r r r Ú test_duplicate_ref_loc_failurew s
|
|
z/TestBlockManager.test_duplicate_ref_loc_failurec C sR t |¡}t t|t|¡ t|ds,tt|ds:t|jrDt|jrNtd S )NÚ_is_consolidatedÚ_known_consolidated)r# r¤ Úassert_frame_equalr Úhasattrr' rÐ rÑ )r¡ r! Úmgr2r r r r¨ s
|
|
z TestBlockManager.test_pickleÚ
|
mgr_stringza,a,a:f8z a: f8; a: i8c C s* t |}t |¡}t t|t|¡ d S r" )r r# r¤ rÒ r )r¡ rÕ r! rÔ r r r Útest_non_unique_pickle s
|
z'TestBlockManager.test_non_unique_picklec C sP t d}t |¡}t t|t|¡ td}t |¡}t t|t|¡ d S )Nú a: categoryrZ )r r# r¤ rÒ r r Zassert_series_equalr )r¡ r! rÔ ZsmgrZsmgr2r r r Ú test_categorical_block_pickle s
|
|
z.TestBlockManager.test_categorical_block_picklec C s¨ t td}tj dd¡}t| ¡ tjdtjd|j d}t
|
|f|t t d¡gd}t | d¡ ¡ |d ¡ t | d¡ ¡ |d ¡ t | d¡ ¡ |d ¡ d S ) NÚabcr\ rQ )r% rk rV )r rÌ r r. r[ )r
|
r r2 r» Zrandr r7 r3 rª rV r r# Úassert_almost_equalrÍ Úinternal_values)r¡ Úcolsr% Úblockr! r r r Ú test_iget§ s ý zTestBlockManager.test_igetc C sÜ t ddd}| t|jdt dgd ¡¡ | dt dgd ¡¡ t | d ¡
|
¡ t d gd ¡¡ t | d¡
|
¡ tjdgd tj d
|
¡ t | d ¡
|
¡ t d gd ¡¡ t | d¡
|
¡ tjdgd tj d
|
¡ d S ) Nz
|
a,b,c: int©r\ ©r~ rK Úfoor\ r. Úbarr rQ r[ ) r Úinsertr6 r r2 r© Úisetr# r$ rÍ rÛ Úobject_rË r r r Útest_setµ s " ÿ" ÿzTestBlockManager.test_setc C s | t|jdtjttd¡ | |j d¡t dt¡¡ |j d¡}|
|
|¡j tj ks\t | ¡ }| |j d¡t dt¡¡ |j d¡}|
|
|¡j tj ks¢t | t|jdtj t¡ t¡¡ |j d¡}|
|
|¡j tjksæt | |j d¡tj t¡¡ |
|
|¡j tjkst d S )NÚbazrQ rá Zquux)rã r6 r r2 Úzerosrl rP rä Úget_locÚrepeatrÍ r& rå r' Ú consolidater» r¼ rm r> Úint_Zfloat_)r¡ r! ÚidxrÔ r r r Útest_set_change_dtypeà s " z&TestBlockManager.test_set_change_dtypec C sD |j dd}t|j|jD ]T\}}t |j|j¡ t|jtjrV|jj |jj ksnt
|
q|jj j |jj j kst
|
q| ¡ |j dd}t|j|jD ]¬\}}|j}|j}t ||¡ t|tjrÌ|j }|j }n|j j }|j j }t|t r
|
|d krú|d ks>||k s>t
|
qt|tjs(||k s>t
|
q|d kr:|d kst
|
qd S )NF)ÚdeepT)r7 r¾ r r# Z assert_equalr% r( r2 ÚndarrayÚbaser' Z_ndarrayZ_consolidate_inplacer ) r¡ r! Úcpr§ Zcp_blkZbvalsZcpvalsZlbaseZrbaser r r r® Ö s, "zTestBlockManager.test_copyc C s t d}| ¡ jtjks td S )Nza: sparse-1; b: sparse-2)r r* r& r2 Úfloat64r' rË r r r Ú test_sparseû s z TestBlockManager.test_sparsec C s, t d}t|jdkstt|ts(td S )Nz a: sparse-1; b: sparse-2; c: f8r\ )r r6 r r' r( r rË r r r Útest_sparse_mixedÿ s z"TestBlockManager.test_sparse_mixedzmgr_string, dtypez c: f4; d: f2úc: f4; d: f2; e: f8c C s t |}| ¡ j|kstd S r" ©r r* r& r' ©r¡ rÕ r& r! r r r Útest_as_array_float s z$TestBlockManager.test_as_array_floatza: bool-1; b: bool-2z%a: i8-1; b: i8-2; c: i4; d: i2; e: u1zc: i4; d: i2; e: u1c C s t |}| ¡ j|kstd S r" r÷ rø r r r Útest_as_array_int_bool s z'TestBlockManager.test_as_array_int_boolc C s t d}| ¡ jdkstd S )Nz h: datetime-1; g: datetime-2rS r÷ rË r r r Útest_as_array_datetime s z'TestBlockManager.test_as_array_datetimec C sF t d}| d¡jdks t| d¡jdks0t| ¡ jdksBtd S )Nz%h: M8[ns, US/Eastern]; g: M8[ns, CET]r zdatetime64[ns, US/Eastern]r. zdatetime64[ns, CET]rG )r rÍ r& r' r* rË r r r Útest_as_array_datetime_tz s z*TestBlockManager.test_as_array_datetime_tzÚtÚfloat16Úfloat32ró r Úint64c C sL t d}t |¡}| |¡}| d¡jj|ks2t| d¡jj|ksHt| d¡jj|ks^tt d}t |¡}|j|dd}| d¡jj|kst| d¡jj|ksªt| d ¡jj|ksÀt| d
|
¡jj|ksÖt| d¡jjtjksît| d¡jjtjkst|tjkr0| d ¡jjtj ksHtn| d ¡jj|ksHtd S ) Nrö r r. r[ z6a,b: object; c: bool; d: datetime; e: f4; f: f2; g: f8Úignore)Úerrorsr] re rf r\ )
|
r r2 r& rm rÍ Útyper' rå r Ú
|
datetime64)r¡ rý r! Ztmgrr r r Ú test_astype" s$
|
|
|
z TestBlockManager.test_astypec C sP dd }t d}|jdd}||| t d}| dtjdgt tjd ¡ | d
|
tjd gt tjd ¡ | d tjd gt tjd ¡ |jdd}| d¡jtjksªt | d
|
¡jtjksÀt | d ¡jtjksÖt | d¡jtj
|
ksìt | d¡jtj kst t d}| dtjdgt tjd ¡ | d
|
tjd gt tjd ¡ | d tjd gt tjd ¡ |jdd}| d¡jtjkst | d
|
¡jtjks¢t | d ¡jtjksºt | d¡jtj ksÒt | d¡jtj ksêt | d¡jjst tj| d¡jtj
|
ks t | d¡jtj ks4t | d¡jtjksLt d S )Nc S s t | j}t |j}t|t|ks(t|D ]2}d}|D ] }|j|jk ¡ r8d} qVq8|s,tq,|D ]2}d}|D ] }|j|jk ¡ rpd} qqp|sdtqddS )z4compare the blocks, numeric compare ==, object don'tFTN)Úsetr r6 r' r% rx )Zold_mgrÚnew_mgrZ
|
old_blocksZ
|
new_blocksrO Úfoundrµ Úobr r r Ú_compare? s"
|
|
|
z/TestBlockManager.test_convert.<locals>._comparez f: i8; g: f8T©r7 z a,b,foo: object; f: i8; g: f8r Ú1rQ r. z2.r[ zfoo.r\ r] zEa,b,foo: object; f: i4; bool: bool; dt: datetime; i: i8; g: f8; h: f2re rf r£ é )r Úconverträ r2 r© rl rå rÍ r& r' r ró r rq r r rþ )r¡ r
|
r! r r r r Ú test_convert> s<
|
ÿ z TestBlockManager.test_convertc C sH t jtdd td W 5 Q R X t jtdd td W 5 Q R X d S )Nz need to splitr¯ úa: category; b: categoryúa: category2; b: category2)r² r³ r| r ©r¡ r r r Útest_invalid_ea_blockx s z&TestBlockManager.test_invalid_ea_blockc C sT dD ]J}t d| }| ¡ j|ks(tt d| d| }| ¡ j|kstqd S )N)r< r? rG rP rF rS rY za: z; b: r÷ )r¡ r& r! r r r Útest_interleave s
|
z TestBlockManager.test_interleave)r× r? )r r? )úa: category; b: category2rG )ú a: category2rG )r rG )úa: f8r< )ú a: f8; b: i8r< )ú a: f4; b: i8r< )úa: f4; b: i8; d: objectrG )úa: bool; b: i8rG )ú
|
a: complexrF )úa: f8; b: categoryrG )úa: M8[ns]; b: categoryrG )úa: M8[ns]; b: boolrG )úa: M8[ns]; b: i8rG )úa: m8[ns]; b: boolrG )úa: m8[ns]; b: i8rG )úa: M8[ns]; b: m8[ns]rG c C s² t d}| ¡ jdkstt d}| ¡ jdks4tt d}| ¡ jdksNtt d}| ¡ jdkshtt d}| ¡ jdkstt d }| ¡ jdkstt d
|
}| ¡ jdks¶tt d }| ¡ jdksÐtt d }| ¡ jd ksêtt d}| ¡ jdkstt d}| ¡ jdks"tt d}| ¡ jdks>tt d}| ¡ jdksZtt d}| ¡ jdksvtt d}| ¡ jdkstt d}| ¡ jdks®td S )Nr× r? r rG r r r< r r r r r rF r r r r r! r" r# r÷ rø r r r Útest_interleave_dtype s@ z&TestBlockManager.test_interleave_dtypec C sþ | |j d¡tj t¡¡ | |j d¡tj t¡¡ | |j d¡tj t¡¡ | |j d¡tj t¡¡ | |j d¡tj t¡¡ | ¡ }|jdks¢t | ¡
|
¡ }|jdks¼t t |j d j tsÒt t |j d j jtjt|jtjd ¡ d S )
|
NÚfrK rO ÚgÚhr] r. r rQ )rä r ré r2 r» r¼ rl rë rÊ r' Úget_numeric_datar( r r) r r# r$ r* r3 r6 rª )r¡ r! Zconsr r r Ú test_consolidate_ordering_issuesà s ÿz1TestBlockManager.test_consolidate_ordering_issuesc C sÈ t d}|jddddgdd}| ¡ r*tt |jtddddg¡ t | d¡
|
¡ | d¡
|
¡ ¡ t | d ¡
|
¡ | d
|
¡
|
¡ ¡ t | d¡
|
¡ | d ¡
|
¡ ¡ t | d ¡
|
¡ | d ¡
|
¡ ¡ d S ) Nz3a: f8; b: i8; c: f8; d: i8; e: f8; f: bool; g: f8-2r& r_ r^ rK r ©Úaxisrf r[ r. r\ ) r Ú reindex_axisZis_consolidatedr' r# Úassert_index_equalr r
|
rÚ rÍ rÛ )r¡ r! Ú reindexedr r r Útest_reindex_itemsÔ s( ÿ ÿ ÿ ÿz#TestBlockManager.test_reindex_itemsc C s¤ t ddd}| dtjdddgtjd¡ | ¡ }t |jt d d
|
d d g¡ t
|
| |j d
|
¡¡ ¡ | |j d
|
¡¡ ¡ ¡ |j|j d
|
¡t d ddg¡dd |rÐt
|
| |j d
|
¡¡ ¡ t dddg¡¡ n*t
|
| |j d
|
¡¡ ¡ t d ddg¡¡ |jdd}t |jt d d
|
d d g¡ |j|j d
|
¡t dddg¡dd |rvt
|
| |j d
|
¡¡ ¡ t dddg¡¡ n*t
|
| |j d
|
¡¡ ¡ t d ddg¡¡ d S )Nú[int: int; float: float; complex: complex;str: object; bool: bool; obj: object; dt: datetimerß rà re r. r[ r\ rQ r> r; rF rP ç Y@g i@g Àr@T©Úinplaceç ð?r g @@g @@g p§@)r rä r2 r© rå r( r# r- r r
|
rÚ rÍ ré rÛ )r¡ Úusing_copy_on_writer! ÚnumericZnumeric2r r r Útest_get_numeric_dataë sR ý þ
|
ýþþ
|
ýþþz&TestBlockManager.test_get_numeric_datac C sp t ddd}| dtjdddgtjd¡ | ¡ }t |jt dg¡ t
|
| |j d¡¡ ¡ | |j d¡¡ ¡ ¡ |jd t dddg¡dd
|
|rÂt | |j d¡¡ ¡ t dddg¡¡ n*t | |j d¡¡ ¡ t dddg¡¡ |jdd }| d t dddg¡¡ |rBt | |j d¡¡ ¡ t dddg¡¡ n*t | |j d¡¡ ¡ t dddg¡¡ d S ) Nr0 rß rà rf TFrQ rP r r2 r )r rä r2 r© rå Z get_bool_datar# r- r r
|
rÚ rÍ ré rÛ r$ )r¡ r5 r! ZboolsZbools2r r r Útest_get_bool_data s@ ý þ þþ þþz#TestBlockManager.test_get_bool_datac C s t td d S )Nu b,×: object)Úreprr r r r r Ú test_unicode_repr_doesnt_raiseF s z/TestBlockManager.test_unicode_repr_doesnt_raiseúa,b,c: i8-1; d,e,f: i8-2za,a,a: i8-1; b,b,b: i8-2c C s2 t |}t|jd d d
|j}| |¡s.td S )Nra )r r r rÌ Úequalsr' )r¡ rÕ Úbm1Zbm2r r r Ú test_equalsI s z TestBlockManager.test_equalsz a:i8;b:f8za:i8;b:f8;c:c8;d:bza:i8;e:dt;f:td;g:stringza:i8;b:category;c:category2zc:sparse;d:sparse_na;b:f8c C sN t |}t |j¡}|D ]0}tt||j}| |¡s:t| |¡stqd S r" ) r Ú itertoolsÚ permutationsr r r rÌ r< r' )r¡ rÕ ÚbmZ block_permsZbm_permZbm_thisr r r Ú(test_equals_block_order_different_dtypesR s z9TestBlockManager.test_equals_block_order_different_dtypesc C s. t ddd}| ¡ ¡ dddddgks*td S ) Nr< re )r
rd r4 ç @g @g @)r Úexternal_valuesÚtolistr' rË r r r Útest_single_mgr_ctore s z%TestBlockManager.test_single_mgr_ctorÚvaluer. ÚTruer[ r\ g @c C sL t d}dt|j d}tjt|d |jdgdg|d W 5 Q R X d S )Nr; z9For argument "inplace" expected type bool, received type Ú.r¯ r. r[ r2 )r r rÄ r² r³ r| Z replace_list)r¡ rG r= rÎ r r r Útest_validate_bool_argsi s
|
ÿz(TestBlockManager.test_validate_bool_argsc C s t d}| dt dg¡¡ t |jtjddddgtr:dndd¡ t |jtjddddgtrbdndd¡ t |j
|
dks~t d S )Núa,b,c: i8; d: f8r r. r r rQ r[ ) r Ú_iset_split_blockr2 r© r# r$ Úblklocsr Úblknosr6 r r' ©r¡ rA r r r Útest_iset_split_blockt s ÿ ÿz&TestBlockManager.test_iset_split_blockc C s t d}| dt dg¡t ttdg¡¡ t |jtjddddgt rLdndd¡ t |j
|
tjddddgt rtdndd¡ t |j d kst d S )
|
NrK r r: r. r r rQ r[ r\ )r rL r2 r© r Úranger# r$ rM r rN r6 r r' rO r r r Ú test_iset_split_block_values s & ÿ ÿz-TestBlockManager.test_iset_split_block_valuesN)+rÄ rÅ rÆ r rÏ r¨ r² rÇ rÈ rÖ rØ rÞ ræ rî r® rô rõ r2 rÿ ró rù rq r r rú rû rü r r r r r$ r) r/ r7 r8 r: r> rB rF rJ rP rR r r r r rÉ q s
|
%þ
|
ýþ
|
|
:îþ
|
%2) ÿ
|
ûþ
|
|
|
|
rÉ c C s | j dkr| ¡ S | ¡ jS )Nr. )rV rD r* ÚT)r! r r r Ú _as_array s
|
rT c @ sÔ e Zd ZedeedeedefdedefdedefdedefdgZej de¡d d
|
Z
|
ej de¡d d Z ej de¡ej d de j dg¡dd Zej de¡ej d de j dg¡dd ZdS )Ú TestIndexingr< r? za,b,c,d,e,f: f8rà za,b,c,d,e,f: i8z a,b: f8; c,d: i8; e,f: stringza,b: f8; c,d: i8; e,f: f8r! c C s dd }|j dks t|j t|j D ]^}|||td |||td |||td |||tdd |||tddd |j dk r&|||tjg tjd
|
|||tj|j| tjd
|
|||tj |j| tjd
|
|j| dkr.|||t
|
|j| ¡d dk |||tjd d d gtjd
|
|||g |||t t|j| |j| dkr&|||dddg |||d d dg q&d S )Nc S sê t | }t|tjrb| j| }t|rbt|rbt|t|krbt |tjt|t| tdg¡}t|t r|| j
|
||d}n"| j dkr|dkr| |¡}nd S t d f| |f }t j|| t |dd t | j| | |j| ¡ d S )NrQ r* r. r F©Z check_dtype)rT r( r2 rð rÌ r6 Z concatenaterè rP r Z get_slicerV Z getitem_mgrr# r$ r- )r! r+ Zslobjr ÚaxZslicedZ mat_slobjr r r Úassert_slice_ok¨ s&
|
ÿ
|
ÿz4TestIndexing.test_get_slice.<locals>.assert_slice_okr[ r\ éd r. r] r éþÿÿÿrQ TFra éýÿÿÿ) rV r' rQ r r2 r© rq rp r0 rè r3 r )r¡ r! rX rW r r r Útest_get_slice¦ s0
|
ÿ zTestIndexing.test_get_slicec C s dd }t |jD ]x}|||g d |||dddgd |||tt |j| d |j| dkr|||dddgd |||dd d
|
gd qd S ) Nc S sT t | }| ||¡}tjt |||¡t |dd t | j| |¡|j| ¡ d S )NFrV )rT Ztaker# r$ r2 r- rÌ )r! r+ Úindexerr Ztakenr r r Úassert_take_okä s ÿz.TestIndexing.test_take.<locals>.assert_take_ok)r] r r\ r. r[ ra rZ r[ )rQ rV r r0 )r¡ r! r^ rW r r r Ú test_takeâ s zTestIndexing.test_takerh Nr1 c C s dd }t |jD ]î}|||tg | ||||j| | ||||j| dddg | |||tdddg| |||td|j| d dg| |j| dkr||||j| d d
| ||||j| dd d
| ||||j| dd
|
d dd
|
d g | qd S ) Nc S s` t | }| j| |¡}| j|||d}tjtj||||dt |dd t |j| |¡ d S ©Nrg FrV ) rT rÌ Zget_indexer_forr, r# r$ ÚalgosÚtake_ndr- )r! r+ Ú
|
new_labelsrh r r] r. r r r Úassert_reindex_axis_is_okù s ýzATestIndexing.test_reindex_axis.<locals>.assert_reindex_axis_is_okr rá râ rç r\ r[ ra r. r[ )rQ rV r
|
rÌ r0 )r¡ rh r! rd rW r r r Útest_reindex_axisö s* ÿ ÿz TestIndexing.test_reindex_axisc
|
C s dd }t |jD ]z}|||tg tjg tjd| ||||j| t |j| ¡| |||tdg|j| t |j| ¡| ||||j| d d d
t |j| ¡| ||||j| t |j| ¡d d d
| |||tdddgt dddg¡| |||tdddgt dddg¡| |||td|j| d dgt dddg¡| |j| d kr|||tdddgt dd
|
d g¡| qd S ) Nc S sV t | }tj||||d}| j||||d}tj|t |dd t |j| |¡ d S r` )rT ra rb Zreindex_indexerr# r$ r- rÌ )r! r+ rc r] rh r Z reindexed_matr. r r r Ú assert_reindex_indexer_is_ok s ÿ ÿzGTestIndexing.test_reindex_indexer.<locals>.assert_reindex_indexer_is_okrQ rá ra râ rç r r\ r. r[ ) rQ rV r
|
r2 r© rª rÌ r3 r0 )r¡ rh r! rf rW r r r Útest_reindex_indexer s ÿ ÿû ÿ ÿ ÿ ÿû ûz!TestIndexing.test_reindex_indexer)rÄ rÅ rÆ r rl r ZMANAGERSr² rÇ rÈ r\ r_ r2 rz re rg r r r r rU s"
|
ù
|
|
;
|
rU c @ s e Zd Zej dedddfeddddfeddddfeddddfeddddfg¡dd Zej d
|
edddedddg¡d d Zd d Z ej d
|
eddeddedddedddeddeddeddedddedddedddg
|
¡dd Z
|
ej d
|
eddeddeddedddedddg¡dd Z ej ddgedddfdgedddfdddgedddfdddgedddfddgedddfddgedddfg¡dd Z ej d g dgdd d gd gdgddddgd dddgdddgdddgg ¡d!d" Z ej dedddddgfeddg feddg fg¡d#d$ Zej d%edddddgfeddg feddg feddddddgfg¡d&d' Zd(d) Zej d*edddg feddddddgfedddddddgfdddgddddgfedddg fedddd+d,d-gfeddddd-d,d+gfdddgdd+d,d.gfedddg feddddddgfdddgddddgfg ¡d/d0 Zej d1edddddgg¡d2d3 ZdS )4ÚTestBlockPlacementz slc, expectedr r] r[ r\ r. ra c C s t t||kstd S r" )r6 r r' ©r¡ ÚslcrÀ r r r Útest_slice_lenP s z!TestBlockPlacement.test_slice_lenrj c C s* d}t jt|d t| W 5 Q R X d S )Nzslice step cannot be zeror¯ ©r² r³ r| r ©r¡ rj rÎ r r r Útest_zero_step_raises] s z(TestBlockPlacement.test_zero_step_raisesc C s. t ddd}t|}|jt dd dks*td S )Nr\ ra rZ )r r r] r' )r¡ rj Zbpr r r Ú!test_slice_canonize_negative_stopc s z4TestBlockPlacement.test_slice_canonize_negative_stopNr: c C s* d}t jt|d t| W 5 Q R X d S )Nzunbounded slicer¯ rl rm r r r Útest_unbounded_slice_raisesj s z.TestBlockPlacement.test_unbounded_slice_raisesrY c C s t |jrtd S r" ©r Z is_slice_liker' )r¡ rj r r r Útest_not_slice_like_slices s z-TestBlockPlacement.test_not_slice_like_sliceszarr, slcée re é éÈ c C s t |j|kstd S r" )r Úas_slicer' )r¡ r8 rj r r r Ú test_array_to_slice_conversion s z1TestBlockPlacement.test_array_to_slice_conversionr8 rZ r[ éöÿÿÿc C s t |jrtd S r" rq )r¡ r8 r r r Útest_not_slice_like_arrays s z-TestBlockPlacement.test_not_slice_like_arraysc C s t t||kstd S r" )r r r' ri r r r Útest_slice_iter¬ s z"TestBlockPlacement.test_slice_iterzslc, arrc C s" t t|jtj|tjd¡ d S )NrQ )r# r$ r r* r2 Úasarrayrª )r¡ rj r8 r r r Ú test_slice_to_array_conversion³ s
|
ÿz1TestBlockPlacement.test_slice_to_array_conversionc C sz t tdd}| d¡jtdddks*t| t d¡¡jtdddksLtt| t ddd¡¡dddddgksvtd S )Nr re r. rf r: r[ ra )r r Úaddrv r' r2 r3 r )r¡ Zbplr r r Útest_blockplacement_addÁ s "z*TestBlockPlacement.test_blockplacement_addzval, inc, expectedé é é é c C s t t| |¡|kstd S r" )r r r} r' )r¡ ÚvalÚincrÀ r r r Útest_blockplacement_add_intÇ s z.TestBlockPlacement.test_blockplacement_add_intr c C s0 d}t jt|d t| d¡ W 5 Q R X d S )Nziadd causes length changer¯ rx )r² r³ r| r r} )r¡ r rÎ r r r Ú"test_blockplacement_add_int_raisesÚ s z5TestBlockPlacement.test_blockplacement_add_int_raises)rÄ rÅ rÆ r² rÇ rÈ r rk rn ro rp rr rw ry rz r| r~ r
r r r r r rh O sÀ ûþ
|
|
|
|
|
|
|
|
ôþ
|
|
|
ûþ
|
|
úþ
|
|
|
÷þ
|
,þ
|
üþ
|
õþ
|
rh c
|
@ sÊ e Zd Zejdd dd dd dd dd dd dd d d gd
|
d d Zd d Zej de j
|
e j e j g¡dd Z ej de j
|
e j e j g¡dd Zdd Zdd ZedddZeedd d Zd S ) ÚTestCanHoldElementc C s | S r" r ©r/ r r r r ä r zTestCanHoldElement.<lambda>c C s | ¡ S r" )Z to_seriesr r r r r å r c C s | j S r" )rv r r r r r æ r c C s t | S r" )r r r r r r ç r c C s
|
| t¡S r" )rm rG r r r r r è r c C s
|
t | ¡S r" )r2 r{ r r r r r é r c C s | d S r r r r r r r ê r c C s | d d
S r r r r r r r ë r r c C s |j S )z¡
|
Functions that take an Index and return an element that should have
|
blk._can_hold_element(element) for a Block with this index's dtype.
|
r )r¡ r r r r Úelementâ s zTestCanHoldElement.elementc C sÌ t ddg}| g ¡stt |j ¡ ¡}| d ¡s8td |d< |d tjksRtt d¡t
|
dddg}|D ]}| |¡s~t||d< qlt ddd}| |¡r¢td}t j t|d ||d< W 5 Q R X d S )Nr r z
|
2010-10-10iÚ r: zLvalue should be a 'Timestamp', 'NaT', or array of those. Got 'date' instead.r¯ )r Ú_can_hold_elementr' Úpdr© r% ro ZNaTr2 r r r r² r³ Ú TypeError)r¡ rÝ r8 Úvalsr rÎ r r r Ú$test_datetime_block_can_hold_elementõ s
|
ÿz7TestCanHoldElement.test_datetime_block_can_hold_elementr& c C sB t jdddg|d}t |¡}t|jdgdd}| g ¡s>td S )Nr. r\ r] rQ r[ rU )r2 r© r Ú from_breaksr rv r r' )r¡ r& r r8 Úiir§ r r r Ú(test_interval_can_hold_element_emptylist s
|
z;TestCanHoldElement.test_interval_can_hold_element_emptylistc
|
C s t jddddg|d}t |¡}t|jdgdd}||}| ||d¡ | |¡sVttj|d d
d
|
d }||}| ||d ¡ | |¡rtt t dt dt dg¡}||}| ||d ¡ | |¡rÐtt t
|
dt
|
dt
|
dg¡} || }| ||d ¡ | |¡rtd S ) Nr. r\ r] é rQ r[ rU Tra Zneither)ÚclosedF) r2 r© r r r rv Úcheck_series_setitemr r' r r )
|
r¡ r& r r8 r r§ ÚelemZii2Zii3Zii4r r r Ú test_interval_can_hold_element s$
|
z1TestCanHoldElement.test_interval_can_hold_elementc C s: t dddd}t|j dd¡dgdd}| g ¡s6td S )NÚ2016r\ rJ ©ZperiodsÚfreqr. r[ rU )r r rv rn r r' )r¡ Úpir§ r r r Ú&test_period_can_hold_element_emptylist6 s z9TestCanHoldElement.test_period_can_hold_element_emptylistc C sx t dddd}||}| ||d¡ | d¡d d
}||}| ||d¡ | d ¡d d
}||}| ||d¡ d S )
|
Nr r\ rJ r TÚDra FÚS)r r ZasfreqZ to_timestamp)r¡ r r r Zpi2Zdtir r r Ú test_period_can_hold_element< s z/TestCanHoldElement.test_period_can_hold_elementr2 c C s2 |j jd }|r | |¡s.tn| |¡r.td S r )r¸ r r r' )r¡ Úobjr r3 r§ r r r Úcheck_can_hold_elementL s z)TestCanHoldElement.check_can_hold_element)Úindexr3 c C sl |j ¡ }t|dd}| |||¡ t|r6||d< n||d t|
< |rZ|j|kshtn|jt kshtd S )NFr r )
|
rv r7 r r r r6 r© r' r& rG )r¡ r r¡ r3 r8 Zserr r r r S s
|
|
z'TestCanHoldElement.check_series_setitemN)rÄ rÅ rÆ r² Úfixturer r rÇ rÈ r2 r Zuint64ró r r r r rP r r
|
r r r r r r á s* øÿ
|
|
|
r c @ s e Zd Zdd ZdS )ÚTestShouldStorec C sp t dddg}t|}|jjd }| |¡s0t| |d d
¡sFt| | ¡ ¡rXt| t |¡¡rltd S )NrJ ÚBÚCr ra ) r r r¸ r Z should_storer' Z
|
as_orderedr2 r{ )r¡ Úcatr¹ r§ r r r Ú test_should_store_categoricale s z-TestShouldStore.test_should_store_categoricalN)rÄ rÅ rÆ r§ r r r r r£ d s r£ c C sF t ddg¡}td}d}tjt|d | ||dd W 5 Q R X d S )Nr4 rC r[ z:Wrong number of dimensions. values.ndim != ndim \[1 != 2\]r¯ rU )r2 r© r r² r³ r| )r r% rk rÎ r r r Útest_validate_ndimu s
|
r¨ c C sf t dddddg} tdddg | ¡}ttdddg | ¡}|jjd jj|jjd jjksbtd S )Nr r. r[ r\ r] ) r
|
r Zreindexr r¸ r r) r] r' )rí r^ rO r r r Útest_block_shape~ s r© c C sÐ t j t ddg¡¡}| |tt||jd}|jj dks>t
|
| t krÌ|j dksTt
|
| |tt||j|jd}|jj dkst
|
|j dkst
|
| | ¡ tt||j|jd}|jj dks¾t
|
|j dksÌt
|
d S )Nr. r[ rU )rM ÚuF)r& rV )r ZarraysZ PandasArrayr2 r© r r6 rV r& Úkindr' r Z is_extensionZto_numpy)r r8 r¿ r r r Ú test_make_block_no_pandas_array s
|
ÿr¬ )N)N)Dr r r? rs Únumpyr2 r² Zpandas._libs.internalsr Z pandas.compatr Z pandas.util._test_decoratorsÚutilZ_test_decoratorsrX Zpandas.core.dtypes.commonr Zpandasr r r r r
|
r r r r r Zpandas._testingZ_testingr# Zpandas.core.algorithmsÚcoreZ
|
algorithmsra Zpandas.core.arraysr r r Zpandas.core.internalsr r r Z pandas.core.internals.blocksr r Z skip_array_manager_invalid_testZ
|
pytestmarkr¢ r r! r- r9 rl r r r r r rÉ rT rU rh r r£ r¨ r© r¬ r r r r Ú<module> sZ ,
|
|
X
|
|
|
?
|
u =
|
