3
Ud                 @   sz   d dl m Z  d dlZd dlZd dlZd dlZd dlmZmZ d dl	j
Zdd Zdd Zdd	 Zd
d Zdd Zdd ZdS )    )datetimeN)Index
MultiIndexc             C   s  | j dd*}|j| st|d d+ks*t| j dd,}tt| jd dg dd}tj|jd | |jddgksvttt| jd	 dg dd}tj|jd	 | |d d-kstd
}t	j
t|d | j dd. W d Q R X tjdddgddd	ggdddgd}|jddgdd |d jdd}d|jd0< d1|jd3< d|jd5< d|jd7< d|jd9< d|jd;< d|jd<< d=|jd>< d|jd?< d|jd@< d|jdA< d|jdB< tjdddgddd	gdddgdddCgdddgdddgdddgdddggdddgd}|jddgdd tj||dd tj||d  d d! tdD d"d! tdD  dDdEg } tjtjdd&d'tjj| d dF }d'|jdG< d(|jdH< tjtjdd(d)tjj| }tj|| d S )INr   bartwoabcthreefirst)namesecond   z/Item must have length equal to number of levels)matchfoo2abdZ1stZ2nd3rd)columnsT)Zinplace)deep   x         w   F)Zcheck_dtypec             S   s   g | ]}d |fqS )Ztest1 ).0ir   r   [/var/www/html/virt/lib64/python3.6/site-packages/pandas/tests/indexes/multi/test_reshape.py
<listcomp>I   s    ztest_insert.<locals>.<listcomp>c             S   s   g | ]}d |fqS )Ztest2r   )r   r   r   r   r   r    J   s    test      
            )r   r   )r   r   )r   r   )r   r   )r   r   r   )r(   r   r   r   )r*   r   r   r   )r+   r   r   r   )r,   r   r   r   )r-   r   r   r   )r.   r   )r   r   r)   )r   r   )r   r   )r   r   )r   r   )r   r   r)   )r!   r"   )r!   r#   )r!   r"   )r!   r#   )insertZequal_levelsAssertionErrorr   listZlevelstmassert_index_equalnamespytestraises
ValueErrorpdZ	DataFrameZ	set_indexcopylocZassert_frame_equalZassert_series_equalrangeZSeriesnpZlinspacer   from_tuples)idxZ	new_indexZexp0Zexp1msglefttsrightr   r   r   test_insert   s^    $












,&

rD   c             C   s   | d d j | dd  }|j| s(t| d d | dd | dd  g}|d j |dd  }|j| snt| j g }|j| std S )Nr   r   r   )appendequalsr1   )r?   resultZfoosr   r   r   test_appendX   s    "
rH   c        	   
   C   s  t dddg} tjddddd}t d	d
dg}tj| |g}tj| ||g}| j|}tjd}d|jt	dddfd|jt	dddfd|jt	dddfg}t dddg| }t
j|| |j| }t |dddg }t
j|| |j|}tj| j| |j|g}t
j|| |j|}t
j|| |j|}t jtjd|jt	dddd	fd|jt	dddd
fd|jt	ddddfg| tdd }t
j|| d S )Ng?g333333?g?z
2011-01-01Dr   z
Asia/Tokyo)freqZperiodstzABCi  r   r   )Zdtype)r   r9   Z
date_ranger   Zfrom_arraysrE   pytztimezoneZlocalizer   r3   r4   Z_simple_newr=   arrayobject)	Zidx1Zidx2Zidx3Zmidx_lv2Zmidx_lv3rG   rK   Zexpected_tuplesexpectedr   r   r   test_append_indexe   s<    





rT   c              C   s\   d} dddg}t jddg}tj||g|d}tj||j| g|d}tj|j| | d S )Nr   r   r   Zfoor   )r5   )r=   rQ   r   Zfrom_productrepeatr3   r4   )ZrepsZnumbersr5   mrS   r   r   r   test_repeat   s    
rW   c             C   s2   | dd }| dd j |jd| d s.td S )Nr   r   r   )rF   r0   r1   )r?   rG   r   r   r   test_insert_base   s    rX   c          
   C   s   | dd  }| j d}|j|s$t|j|jks4t| d d }| j d}|j|sXt|j|jkshtd}tjt|d | j t|  W d Q R X d S )Nr   r   z/index 6 is out of bounds for axis 0 with size 6)r   r)   r)   )deleterF   r1   r
   r6   r7   
IndexErrorlen)r?   rS   rG   r@   r   r   r   test_delete_base   s    

r\   )r   Znumpyr=   r6   rO   Zpandasr9   r   r   Zpandas._testingZ_testingr3   rD   rH   rT   rW   rX   r\   r   r   r   r   <module>   s   
L/
