U
|
ý°dU" ã @ sä d dl Z d dlZd dlZd dlmZ d dlZd dlmZm Z m
|
Z
|
m Z d dl m Z dd Zdd Zdd Zd
|
d Zd d Zejjdd Zdd Zdd Zdd Zdd Zdd Zdd Z d d Z d d Z d d! Z d"d# Z d$d% Z!dS )&é N)Ú'construct_1d_object_array_from_listlike)ÚIndexÚ IntervalIndexÚ
|
MultiIndexÚ
|
RangeIndexc C sò t ddg¡} | jd jdks"t| jd jdks6tt dgtdg¡} | jd jdks^tt dgtdg¡} | jd jd kstt dgtd
|
g¡} | jd jd ks®tt dgtd g¡} | jd dk ¡ sØt| jd dk ¡ sîtd S ) N)ÚAé )r é r Zint8r Úaé( i Úint16i@ Úint32éè )r Ú from_tuplesÚcodesÚdtypeÚAssertionErrorÚ from_productÚrangeÚall)Úi© r ú`d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexes/multi/test_integrity.pyÚtest_labels_dtypes s r c C s dt d¡fdt jfdt d¡fdt d¡fdt d¡fdt d¡fg} t | ¡}t| }t |j|¡ t |jd d
|d d
j¡ d S ) Nr z
|
2000-01-01r é z
|
2000-01-03z
|
2000-01-04z
|
2000-01-02é ) ÚpdÚ TimestampZNaTr r r ÚtmÚassert_numpy_array_equalÚvalues)ZtuplesÚresultÚexpectedr r r Útest_values_boxed$ s ú
|
r# c C sÚ t dd¡} t | ¡}tj| dd}t ||g¡}|j}t dd |D ¡}t ||¡ t dd |D ¡}t ||¡ |d d
j}t d d |D ¡}t ||d d
¡ t d
|
d |D ¡}t ||d d
¡ d S ) Nl ÈNÎZol ÈNÎZoz
|
US/Central)Útzc S s g | ] }|d qS ©r r ©Ú.0Úxr r r Ú
|
<listcomp>> s z8test_values_multiindex_datetimeindex.<locals>.<listcomp>c S s g | ] }|d qS ©r r r& r r r r) A s r c S s g | ] }|d qS r% r r& r r r r) G s c S s g | ] }|d qS r* r r& r r r r) J s ) ÚnpÚaranger Z DatetimeIndexr Ú from_arraysr r Úassert_index_equal)ÚintsZnaiveZawareÚidxr! ÚouterÚinnerr r r Ú$test_values_multiindex_datetimeindex4 s
|
r3 c C sà t dd¡} tj| dd}t | |g¡}|j}tdd |D }t |t| t j
|
d¡ t dd |D ¡}t ||¡ |d d
j}td
|
d |D }t |t| d d
t j
|
d¡ t d d |D ¡}t ||d d
¡ d S ) Ni× iÜ ÚD)Úfreqc S s g | ] }|d qS r% r r& r r r r) V s z6test_values_multiindex_periodindex.<locals>.<listcomp>©r c S s g | ] }|d qS r* r r& r r r r) Y s r c S s g | ] }|d qS r% r r& r r r r) _ s c S s g | ] }|d qS r* r r& r r r r) b s ) r+ r, r Z PeriodIndexr r- r r r r. Úint64)r/ Zpidxr0 r! r1 r2 r r r Ú"test_values_multiindex_periodindexN s r8 c C s¤ t td} t td}t d¡}t tdd¡}t| |g||gd}t dddddddddg ¡}t dddddddddg ¡}t| |g||gd}|jd ks td S )
|
Nip é
|
iX )Úlevelsr r r r r F) Úlistr r+ r, Úrepeatr ÚarrayZ is_uniquer )Z
|
major_axisZ
|
minor_axisZ major_codesZ minor_codesÚindexr r r Útest_consistencyf s
|
ÿ ÿr? c C s t jt d¡t d¡gddgd} | | j¡}t |tjt| dd¡ ddt| d t| d fD ] }| | | ¡}||ksdt
|
qdd S )
|
Nr ZoneZtwo©ÚnamesZintpr6 r r r ) r r r+ r, Z get_indexerr r r ÚlenÚget_locr )r> r! r r r r Útest_hash_collisions} s ÿ rD c C s d S )Nr r r r r Ú test_dims s rE c C s¾ ddgt d¡t d¡gg} tj| ddgd}dd g}d
|
}tjt|d |j|d d W 5 Q R X d }tjt|d |j||d W 5 Q R X d}tjt|d |j|dd W 5 Q R X d S )Nr ÚBz
|
2011-01-01z
|
2011-01-02ÚstrÚdtr@ r r z1take\(\) got an unexpected keyword argument 'foo'©Úmatch)Úfooz$the 'out' parameter is not supported)Úoutz%the 'mode' parameter is not supportedZclip)Úmode) r r r r ÚpytestÚraisesÚ TypeErrorZtakeÚ
|
ValueError)Úvalsr0 ÚindicesÚmsgr r r Útest_take_invalid_kwargs s rU c C s, d}t jt|d t | ¡ W 5 Q R X d S )Nz"isna is not defined for MultiIndexrI )rN rO ÚNotImplementedErrorr Zisna)r0 rT r r r Útest_isna_behavior¢ s rW c C sÜ t jdt ddgtdg¡dgd} tjtdd | jd W 5 Q R X tjtd d | jd
|
W 5 Q R X t jdt ddgtd g¡dgd}tjtdd |jd W 5 Q R X tjtd d |jd
|
W 5 Q R X d S ) Nr r i ¡ Údest)r> Úcolumnsz ^\(-1, 0\)$rI ))éÿÿÿÿr rX z
|
^\(3, 0\)$))r r rX i!¡ ) r Ú DataFramer r r rN rO ÚKeyErrorÚloc)Zdf_below_1000000Zdf_above_1000000r r r Útest_large_multiindex_error« s$ ÿ ÿr^ c C s^ t td} tj| | dt dd | D ¡d}d}tjt|d |d ¡ W 5 Q R X d S ) Ni@B )r
|
Úbc S s g | ] }||fqS r r r& r r r r) Á s z7test_million_record_attribute_error.<locals>.<listcomp>©r> z&'Series' object has no attribute 'foo'rI r
|
)
|
r; r r r[ r r rN rO ÚAttributeErrorrK )ÚrÚdfrT r r r Ú#test_million_record_attribute_error½ s ÿrd c C s | d }| |¡dkstd S )Nr T)Z$_can_hold_identifiers_and_holds_namer )r0 Úkeyr r r Útest_can_hold_identifiersÉ s rf c C sð | j | j }}t d¡}tjt|d |d |d< W 5 Q R X tjt|d |d d |d d< W 5 Q R X tjt|d |d |d< W 5 Q R X tjtdd |d d |d d< W 5 Q R X | j}tjt|d |d |d< W 5 Q R X d S )Nz#does not support mutable operationsrI r z#assignment destination is read-only) r: r ÚreÚcompilerN rO rP rQ rA )r0 r: r Z mutable_regexrA r r r Útest_metadata_immutableÎ s
|
ri c C sR t dddddgdddddgg¡} | js,t| ddgdddgg¡} | jrNtd S )Nr rF r r r )r r- Zis_monotonic_increasingr Z
|
set_levels)Úindr r r Ú$test_level_setting_resets_attributesá s "
|
rk c C s t t d¡ d¡¡} t t d¡ d¡¡}t j| ¡ | ¡ ddd}ddg|j_t | t jt d¡t d¡dt
|
j t d t d gddgd
|
d }t j||d d |j d¡}ttjd tjddd d ¡}t ||¡ |j d¡}tt tjd tjdd ¡dd}t ||¡ d S )Néd )r9 r9 )Údf1Údf2r )ZaxisZfizzZbuzz)rn rm r9 r@ r` T)Z
|
check_liker6 )Úname)r r[ r+ r, ZreshapeÚconcatÚstackr> rA rG r r r r Zassert_frame_equalZget_level_valuesr r7 r< r. Ztile)rm rn rc r" r! r r r Ú%test_rangeindex_fallback_coercion_bugé s þ rr c C st | ¡ }t| rd| | d ¡ | ¡ }| j dd}t| ttfsL||ksLt| jdkrp||ksptn |dksptd S )Nr T)ÚdeepÚobject)Z memory_usagerB rC Ú
|
isinstancer r r Z inferred_type)r0 r! Zresult2Zresult3r r r Útest_memory_usage s
|
rv c C s | j dkstd S )Nr )Znlevelsr )r0 r r r Ú test_nlevels s rw )"rg Únumpyr+ rN Zpandas.core.dtypes.castr Zpandasr r r r r Zpandas._testingZ_testingr r r# r3 r8 r? ÚmarkZslowrD rE rU rW r^ rd rf ri rk rr rv rw r r r r Ú<module> s0
|
|
