U
|
ý°d
ã @ s¦ d dl m Z d dlZd dlZd dlZd dlZd dlmZmZ d dl m
|
Z dd Z dd Z dd Zd
|
d Zej d d dg¡dd Zdd Zdd Zdd Zdd ZdS )é )ÚdatetimeN)ÚIndexÚ
|
MultiIndexc C s& | dd¡}| | ¡st|d dks*t| dd¡}tt| jd dg dd}t |jd |¡ |jddgksvttt| jd d g dd}t |jd |¡ |d dks´td
|
}t j
|
t |d | dd ¡ W 5 Q R X t j d ddgdddggdddgd}|jddgdd |d jdd}d|jd< d|jd< d|jd < d |jd < d |jd < d!|jd"< d|jd#< d|jd$< d|jd%< d |jd&< d |jd'< d!|jd(< t j d ddgdddgdd)dgdd dgdddgd d)d gd d*d gd d d!ggdddgd}|jddgdd tj||d+d, t ||d ¡ d S )-Nr )ÚbarZtwo)ÚabcÚthreer Úfirst)ÚnameÚsecondé r z/Item must have length equal to number of levels©Úmatch)Zfoo2ÚaÚbÚdZ1stZ2ndÚ3rd)ÚcolumnsT)Zinplace)Údeepé )©r Úxr éÿÿÿÿ)©r r r é )©r r r é )©r r r é )©r Úwr é )©r r r r r r r r r! r r F)Z check_dtype)ÚinsertZ equal_levelsÚAssertionErrorr ÚlistÚlevelsÚtmÚassert_index_equalÚnamesÚpytestÚraisesÚ
|
ValueErrorÚpdZ DataFrameZ set_indexÚcopyÚlocZassert_frame_equalÚassert_series_equal)ÚidxZ new_indexZexp0Zexp1ÚmsgÚleftÚtsÚright© r5 ú^d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexes/multi/test_reshape.pyÚ test_insert sV $
|
|
|
|
|
|
|
|
|
|
|
|
ø
|
õ r7 c C s dd t dD dd t dD ddg } t t dd d
|
¡t | d d
¡¡}d
|
|jd< d |jd< t t dd d ¡t | ¡¡}t ||¡ d S )Nc S s g | ] }d |fqS )Ztest1r5 ©Ú.0Úir5 r5 r6 Ú
|
<listcomp>N s z test_insert2.<locals>.<listcomp>r c S s g | ] }d |fqS )Ztest2r5 r8 r5 r5 r6 r; O s r )Útesté )r< é r é
|
é éþÿÿÿé é )
|
Úranger, ZSeriesÚnpZlinspacer Ú from_tuplesr. r& r/ )r0 r2 r4 r5 r5 r6 Ú test_insert2K s ÿþÿ$
|
|
rG c C s | d d
| dd
¡}| | ¡s(t| d d
| dd
| dd
g}|d |dd
¡}| | ¡snt| g ¡}| | ¡std S )Nr r r )ÚappendÚequalsr# )r0 ÚresultZfoosr5 r5 r6 Ú test_append] s "
|
rK c C s¢ t dddg} tjddddd}t d d
|
d g}t | |g¡}t | ||g¡}| |¡}t d¡}d| t d d d ¡fd| t d d d¡fd| t d d d¡fg}t dddg| }t
|
||¡ | | ¡}t |dddg }t
|
||¡ | |¡}t | | ¡| |¡g¡}t
|
||¡ | |¡}t
|
||¡ | |¡}t t jd| t d d d ¡d fd| t d d d¡d
|
fd| t d d d¡d fg| tdd ¡}t
|
||¡ d S )Ngñ?g333333ó?gÍÌÌÌÌÌô?z
|
2011-01-01ÚDr z
|
Asia/Tokyo)ÚfreqZperiodsÚtzÚAÚBÚCiÛ r r )Zdtype)r r, Z
|
date_ranger Ú from_arraysrH ÚpytzÚtimezoneZlocalizer r& r' Z _simple_newrE ÚarrayÚobject) Zidx1Zidx2Zidx3Zmidx_lv2Zmidx_lv3rJ rN Zexpected_tuplesÚexpectedr5 r5 r6 Útest_append_indexj sF
|
|
ý
|
|
|
|
ýûù ö rX z name, expr )ÚcNc C st t jddgddggddgd}t jdgdggd| gd}| |¡}t jdddgdddggd|gd}t ||¡ d S ) Nr r r r r r ©r( r ©r rR rH r& r' )r ÚexpÚmidxÚmidx2rJ rW r5 r5 r6 Útest_append_names_match s
|
|
"r_ c C sp t jddgddggddgd} t jdgdggd d
|
gd}| |¡}t jdddgdddggd d}t ||¡ d S ) Nr r r r r r rZ r r Úyr[ )r] r^ rJ rW r5 r5 r6 Ú test_append_names_dont_match£ s
|
|
ra c C s\ d} dddg}t ddg¡}tj||g|d}tj|| | ¡g|d}t | | ¡|¡ d S )Nr r r Zfoor rZ )rE rU r Z from_productÚrepeatr& r' )ZrepsZnumbersr( ÚmrW r5 r5 r6 Ú test_repeat¬ s
|
rd c C s2 | dd
}| dd
| d| d ¡¡s.td S )Nr r r )rI r" r# )r0 rJ r5 r5 r6 Útest_insert_base¶ s re c C s | dd
}| d¡}| |¡s$t|j|jks4t| d d
}| d¡}| |¡sXt|j|jkshtd}tjt|d | t| ¡ W 5 Q R X d S )Nr r r z/index 6 is out of bounds for axis 0 with size 6r )ÚdeleterI r# r r) r* Ú
|
IndexErrorÚlen)r0 rW rJ r1 r5 r5 r6 Útest_delete_base½ s
|
|
ri )r ÚnumpyrE r) rS Zpandasr, r r Zpandas._testingZ_testingr& r7 rG rK rX ÚmarkZ parametrizer_ ra rd re ri r5 r5 r5 r6 Ú<module> s < /
|
|
|
