U
|
ý°d{ ã @ sl d dl Zd dlZd dlm mZ d dlmZ d dl Z
|
d dl m Z m Z m Z mZ d dlmZ G dd dZdS )é N)ÚPerformanceWarning)Ú DataFrameÚIndexÚ
|
MultiIndexÚSeriesc @ st e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dd Z dd Zdd ZdS ) ÚTestMultiIndexBasicc C s t ddddgddddgtj d¡d dd g¡}t t¡ |jd
|
W 5 Q R X |j d dd dg }t t¡ |jd W 5 Q R X d S )Nr é ÚxÚzÚyé )ÚjimÚjoeZjolier r )r r
|
é é )r )
|
r ÚnpÚrandomZrandÚ set_indexÚtmZassert_produces_warningr ÚlocZiloc©ÚselfÚdf© r úgd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexing/multiindex/test_multiindex.pyÚtest_multiindex_perf_warn s
|
|
|
ýÿú z-TestMultiIndexBasic.test_multiindex_perf_warnc C sp d}t j}dt _tt |¡t dg| t |¡f¡}|d dksFt|d dksVt|d d ksft|t _d S )
|
Ni' i N Úa)r é r )r é r )r é r )Ú_indexZ _SIZE_CUTOFFr r Zaranger Ú from_arraysÚAssertionError)r ÚnZ
|
old_cutoffÚsr r r Ú(test_indexing_over_hashtable_size_cutoff" s &z<TestMultiIndexBasic.test_indexing_over_hashtable_size_cutoffc C s¤ t ddtjdgddddgdd tjd
|
gd }|jd d gdd}t ddtjdgddddgdd tjd
|
gd tddtjdgd dtddddgd dgd}t ||¡ d S )NZR1ZR2ZR4ÚC1ÚC2ÚC3ÚC4é
|
é é ©r ÚbÚcr r. F)Zdrop©Úname©Úindex)r r Únanr r r Úassert_frame_equal)r r ÚresultÚexpectedr r r Útest_multi_nan_indexing1 s"
|
ýÿ
|
ýþú z+TestMultiIndexBasic.test_multi_nan_indexingc C s t tjtjtjtjgddddgddtjdgd}| d d
|
g¡}t d ddtjdgittjtjtjtjgd d tddddgd
|
d gd }t ||¡ d S )Nr& r' r( r) r* r+ r, r- r r. r/ r0 r2 ) r ÚpdZNaTr r4 r r r r5 )r r r7 r r r Ú"test_exclusive_nat_column_indexingH s
|
ýÿ ÿþü z6TestMultiIndexBasic.test_exclusive_nat_column_indexingc C sØ t dddg¡}tdddg}tj||gddgd}tdd d
|
gtjtjtjgd |d }t|d ddtjgd |d }|jdd}d|j |d dfdf< t
|
||¡ |jdd}d|j |d dfgdf< t
|
||¡ d S )NZ20190101Z20190102r r/ Zindex1Úindex2©Únamesr r r )Úc1Úc2r2 r> ç ð?T)Údeepr r? ) r9 Ú to_datetimer r r! r r r4 Úcopyr r r5 )r ÚdtiÚidxÚmir r7 Zdf2Zdf3r r r Ú test_nested_tuples_duplicates^ s $ z1TestMultiIndexBasic.test_nested_tuples_duplicatesc C sr t tddd}tjddddd}t ||g¡}ttj d d¡|d
|
}|j
|
d j }t ||¡ |j|jksntd S ) Nr ÚAr0 z
|
2020-01-01r ÚDÚB)ZperiodsÚfreqr1 é r2 r )r Úranger9 Z
|
date_ranger Z from_productr r r Zrandnr r3 r Zassert_index_equalrK r" )r rE rD rF r r6 r r r Ú2test_multiindex_with_datatime_level_preserves_freqq s zFTestMultiIndexBasic.test_multiindex_with_datatime_level_preserves_freqc C sd dddg}dddg}t |||d}|jdd gd
|
d t d |itj||gd dd}t ||¡ d S )Ny ð? @y @ Ày $@ ð¿r r r )r r r
|
r r T)Zinplacer
|
)r r r< r2 )r r r r! r r5 )r Z complex_dataZnon_complex_datar6 r7 r r r Útest_multiindex_complex{ s"
|
|
ýÿþþz+TestMultiIndexBasic.test_multiindex_complexc C sV t dddg¡}t|d}|jddidd}t dddg¡}t|d}t ||¡ d S ) N)rH Úcat)rJ rP r2 rH ÚAppler )r3 Úlevel)rQ rP )r Ú from_tuplesr Úrenamer r5 )r rF r Úmi2r7 r r r Ú&test_rename_multiindex_with_duplicates s
|
|
z:TestMultiIndexBasic.test_rename_multiindex_with_duplicatesc
|
C sÌ t dtjgtjtjgg¡}t tjdgtjtjgg¡}tddg|d}tddg|d}| |¡\}}t ddtjgtjtjtjgg¡}tdtjdg|d}ttjddg|d} t ||¡ t || ¡ d S )Nç @T@g T@r r r2 r@ g @©r r! r r4 r Zalignr Úassert_series_equal©
|
r Zmi1rU Zser1Zser2Zresult1Zresult2rF Z expected1Z expected2r r r Ú2test_series_align_multiindex_with_nan_overlap_only s " zFTestMultiIndexBasic.test_series_align_multiindex_with_nan_overlap_onlyc
|
C s¾ t dtjgtjtjgg¡}t tjdgtjtjgg¡}tddg|d}tddg|d}| |¡\}}t dtjgtjtjgg¡}tddg|d}tddg|d} t ||¡ t || ¡ d S )NrW r r r2 rX rZ r r r Ú%test_series_align_multiindex_with_nan© s z9TestMultiIndexBasic.test_series_align_multiindex_with_nanc C s4 t ddgddgg ddg¡j ¡ }|dks0td S )Nr r r )r r r3 Znuniquer" )r r# r r r Útest_nunique_smoke¸ s $z&TestMultiIndexBasic.test_nunique_smokec C sR t tddgt ddgg¡jddddg tddddgt ddddgg¡¡ d S )Nr r r r. )r rY r r r! r )r r r r Ú test_multiindex_repeated_keys½ s
|
ÿ üz1TestMultiIndexBasic.test_multiindex_repeated_keysc C sL t dddgdddgddd gd
|
¡}tjtd d |d g W 5 Q R X d S )Nr r r r r r é é r ))Zfoo)Úbar)NÚ missing_key)Úmatch)rb )r Ú from_dictÚpytestZraisesÚKeyErrorr r r r Ú#test_multiindex_with_na_missing_keyÆ s ýÿz7TestMultiIndexBasic.test_multiindex_with_na_missing_keyN)Ú__name__Ú
|
__module__Ú __qualname__r r% r8 r: rG rN rO rV r[ r\ r] r^ rg r r r r r s
|
|
r )Únumpyr re Zpandas._libs.indexZ_libsr3 r Z pandas.errorsr Zpandasr9 r r r r Zpandas._testingZ_testingr r r r r r Ú<module> s
|
