U
|
ý°dG1 ã @ sþ d dl Zd dlZd dlmZ d dlmZ d dlZd dlm Z m
|
Z
|
d dl m Z d+ddZdd Zd
|
d Zd d Zdd Zdd Zdd Zdd Zdd Zdd Zdd Zej d d dg¡d d Zej d d dg¡d!d" Z d#d$ Z d%d& Z d'd( Z d)d* Z dS ),é N)ÚPY311)ÚDatetimeTZDtype)ÚCategoricalIndexÚ
|
MultiIndexFc C sP t | t |kstt| |D ],\}}t |¡}t |¡}tj|||d q d S )N©Ú check_dtype)ÚlenÚAssertionErrorÚzipÚnpZasarrayÚtmZassert_numpy_array_equal)ÚactualÚexpectedr ZactÚexp© r ú^d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexes/multi/test_get_set.pyÚassert_matching s
|
|
|
r c C s~ ddg| _ | d¡dks t| d¡dks.td}tjt|d | d¡ W 5 Q R X tjtdd | d¡ W 5 Q R X d S )Né r z/Too many levels: Index has only 2 levels, not 3©Úmatché zLevel fourth not foundZfourth)ÚnamesÚ_get_level_numberr ÚpytestÚraisesÚ
|
IndexErrorÚKeyError)ÚidxÚmsgr r r Ú test_get_level_number_integer s
|
r c C sh t jdddgdddgtjdddd gd
|
d d gd } t t d¡t d¡tddd¡}t || j
|
¡ d S )Nr r é ÚaÚbÚcÚ20200101ÚUTC©ZperiodsÚtzÚintÚstringÚdt©r Úint64ÚOÚutc©r' )r( r) r* © r Ú from_productÚpdÚ
|
date_rangeÚSeriesr Údtyper r Úassert_series_equalÚdtypes©Z idx_multityper r r r Útest_get_dtypes% s þýÿr9 c C s^ t dddgdddgtjdddd g¡} t t d
|
¡t d ¡td d d¡}t || j
|
¡ d S )Nr r r r! r" r# r$ r% r& r, r- r. r/ )Zlevel_0Zlevel_1Zlevel_2r0 r8 r r r Ú test_get_dtypes_no_level_name5 s ýÿýÿr: c C sp t jdddgdddgtjdddd gd
|
d
|
d
|
gd j} tjt d ¡t d ¡tddgd
|
d
|
d
|
gd}t
|
| |¡ d S )Nr r r r! r" r# r$ r% r& ÚAr+ r, r- r. r/ )Úindex) r r1 r2 r3 r7 r4 r r5 r r r6 )Úresultr r r r Ú%test_get_dtypes_duplicate_level_namesH s ýúþr> c C sT | }t jtdd |j d¡ W 5 Q R X t jtdd |j d¡ W 5 Q R X d S )NzToo many levelsr r znot a valid level numberéýÿÿÿ)r r r r< r )Z multiindex_dataframe_random_dataÚframer r r Ú#test_get_level_number_out_of_boundsY s
|
rA c C s | j | jkstdd |D }| |¡}| j|ks6t|j|ksDtd}tjt|d | || ¡ W 5 Q R X dd |D }|j|dd}|d kst|j|ks¦t| j|d dd }| j|ksÆt|j|d |d
|
gksàt|j|d ddd }|d kst|j|d |d
|
gks t| j|dd
|
gd }| j|ks@t|j|ksPt|j|dd
|
gdd }|d ksrt|j|kstd S ) Nc S s g | ] }|d qS )ZSUFFIXr ©Ú.0Únamer r r Ú
|
<listcomp>e s z)test_set_name_methods.<locals>.<listcomp>z9Length of names must match number of levels in MultiIndexr c S s g | ] }|d qS )ZSUFFIX2r rB r r r rE l s T)Úinplacer ©Úlevelr ©rH rF )ÚrenameÚ set_namesr r r r Ú
|
ValueError)r Z index_namesZ new_namesÚindr Z
|
new_names2Úresr r r Útest_set_name_methodsb s0
|
rO c C s | j }dd |D }| j}|\}}dd |D }dd |D }||g}d}tjt|d || _ W 5 Q R X trrdnd}tjt|d || _W 5 Q R X d S ) Nc S s g | ]}d d |D qS )c S s g | ] }|d qS ©r! r ©rC Zlevr r r rE s z=test_set_levels_codes_directly.<locals>.<listcomp>.<listcomp>r ©rC rH r r r rE s z2test_set_levels_codes_directly.<locals>.<listcomp>c S s g | ]}|d d qS ©r r r ©rC Úxr r r rE s c S s g | ]}|d d qS ©r r rT r r r rE s zCan't set attributer z5property 'codes' of 'MultiIndex' object has no setterzcan't set attribute)ÚlevelsÚcodesr r ÚAttributeErrorr )r rW Ú
|
new_levelsrX Ú major_codesÚ minor_codesÚ new_codesr r r r Ú test_set_levels_codes_directly s ÿýr^ c C sÎ | j }dd |D }| |¡}t|j | t| j | | j|d dd}t|j |d |d g t| j | | j|d dd}t|j |d |d g t| j | | j|ddgd}t|j | t| j | | ¡ }tjtdd | jdgdd W 5 Q R X t| j |j d d
|
tjtdd | jddd d d dgdd W 5 Q R X t| j|jd d
|
tjt dd | jddd W 5 Q R X t| j |j d d
|
tjt dd | jddd W 5 Q R X t| j|jd d
|
d S )Nc S s g | ]}d d |D qS )c S s g | ] }|d qS rP r rQ r r r rE ¡ s z.test_set_levels.<locals>.<listcomp>.<listcomp>r rR r r r rE ¡ s z#test_set_levels.<locals>.<listcomp>r rG r z^Onr r# Tr r r é é z^Levelsz^Codes)
|
rW Ú
|
set_levelsr Úcopyr r rL Ú set_codesrX Ú TypeError)r rW rZ Úind2Zoriginal_indexr r r Útest_set_levels s6
|
$rf c C sD | j }|\}}dd |D }dd |D }||g}| |¡}t|j | t| j | | j|d dd}t|j |d |d g t| j | | j|d dd}t|j |d |d g t| j | | j|ddgd}t|j | t| j | t dd tdD ¡}td d
|
d
|
}t d d |D ¡}|j|dd }| |¡s@td S ) Nc S s g | ]}|d d qS rS r rT r r r rE Ï s z"test_set_codes.<locals>.<listcomp>c S s g | ]}|d d qS rV r rT r r r rE Ð s r rG r c S s g | ] }d |fqS ©r r ©rC Úir r r rE ç s é é éÿÿÿÿc S s g | ] }d |fqS rg r rh r r r rE é s )rX rH )rX rc r r Ú from_tuplesÚrangeÚequalsr ) r rX r[ r\ r] re rM r r= r r r Útest_set_codesÊ s,
|
rp c C s | j | j }}| j}tjtdd | |d g¡ W 5 Q R X tjtdd | |d g¡ W 5 Q R X tjtdd | |d g¡ W 5 Q R X tjt dd | |d ¡ W 5 Q R X tjt dd | |d ¡ W 5 Q R X tjt dd | |d ¡ W 5 Q R X tjt dd | j|d ddgd W 5 Q R X tjt dd | j|dd W 5 Q R X tjt dd | j|d ddgd W 5 Q R X tjt dd | j|dd W 5 Q R X tjtdd | j|d ddgd W 5 Q R X tjt d
|
d | j|dd W 5 Q R X d S ) NzLength of levelsr r zLength of codeszLength of nameszlist of lists-likez list-liker rG zNames must be a)
|
rW rX r r r rL ra rc rK rd )r rW rX r r r r Ú%test_set_levels_codes_names_bad_inputð s4 rq rF Tc C sT t ddggddggdgd}t ddgg¡}|jdd| d}| rD|}t ||¡ d S )Nr r Úfirst)rW rX r rI )r r1 rK r Úassert_index_equal)rF r Úmr= r r r Ú test_set_names_with_nlevel_1 s ru Úorderedc C s t tdddddgg¡}ttd| d}|j|dd}t |ddddgg|jd }t ||¡ | d¡}ttd
|
|j |j
|
d }t ||¡ d S ) NZxyzxr r r r Zbac)rv rG )rW rX Zbacb)Ú
|
categoriesrv ) r Ú from_arraysÚlistr ra rX r rs Zget_level_valuesrw rv )rv r< Zcidxr= r Z
|
result_lvlZ expected_lvlr r r Útest_set_levels_categorical. s
|
ÿrz c C s¨ ddddddg} dddgd }t j| |gddgd }tjtj d
|
d ¡d d ddg|d}| ¡ }|jd kslt |j
|
j dks|t d|j d< |jd kst |j
|
j dks¤t d S )NZhansÚgretheÚ1Ú2Ú3r ÚNameÚNumberr+ é r_ ÚoneZtwoZthreeZfour)Úcolumnsr< )r r gö(\ÂÕX@))r{ Ú4r ) r rx r2 Z DataFramer ÚrandomZrandnZ
|
sort_indexZ_is_copyr r< r Úat)Zlev1Zlev2r Zdfr r r Útest_set_value_keeps_names? s
|
ÿ
|
r c C st dddg} dgd }t j| |gddgd}|jttdd d
|
gdd }dddg}t j||gddgd}t ||¡ d S ) Nr r r ZblackÚsizeÚcolorr+ r~ r} r| rG )r rx ra Úmapr( r rs )ÚsizesÚcolorsr< r= Zexpected_sizesr r r r Ú test_set_levels_with_iterableO s
|
|
|
r c C sN t jg gdgd} | jt g ¡dd}t jt g ¡gdgd}t ||¡ d S )Nr; r+ r rG )r rx ra r2 Z DatetimeIndexr rs )Zmidxr= r r r r Útest_set_empty_level\ s r c C sx t jddgddgd} tjtdd | dd d
|
gd ¡ W 5 Q R X tjtdd | d d gd d ggd ¡ W 5 Q R X d S ) N)r r )r r ZfooÚbarr+ zpositional argumentsr r! r" r# r r )r rm r r rd ra rc )r r r r Ú test_set_levels_pos_args_removald s þû r )F)!Únumpyr r Z pandas.compatr Zpandas.core.dtypes.dtypesr Zpandasr2 r r Zpandas._testingZ_testingr r r r9 r: r> rA rO r^ rf rp rq ÚmarkZ parametrizeru rz r r r r r r r r Ú<module> s2
|
|
"-&/
|
|
|
