
    bMh7                         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	m
Z d Zd Zd Zd Zej                            dd	d
g          d             Zd Zd Zd Zd Zd ZdS )    )datetimeN)Index
MultiIndexc                 8   |                      dd          }|                    |           sJ |d         dk    sJ |                      dd          }t          t          | j        d                   dgz   d          }t          j        |j        d         |           |j        ddgk    sJ t          t          | j        d                   d	gz   d          }t          j        |j        d         |           |d         dk    sJ d
}t          j	        t          |          5  |                      dd           d d d            n# 1 swxY w Y   t          j        g dg dgg d          }|                    ddgd           |d                             d          }d|j        d<   d|j        d<   d|j        d<   d|j        d<   d|j        d <   d!|j        d"<   d|j        d#<   d|j        d$<   d|j        d%<   d|j        d&<   d|j        d'<   d!|j        d(<   t          j        g dg dg d)g d*g d+g d,g d-g d.gg d          }|                    ddgd           t          j        ||d/0           t          j        ||d                    d S )1Nr   )bartwo)abcthreer	   first)namesecond   r
   z/Item must have length equal to number of levelsmatch)foo2)abr   )r   dr   )1st2nd3rd)columnsr   r   T)inplacer   )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   r   )r   r   r   )r   r   r    )r   r   r"   )r   r&   r$   )r   r   r'   F)check_dtype)insertequal_levelsr   listlevelstmassert_index_equalnamespytestraises
ValueErrorpd	DataFrame	set_indexcopylocassert_frame_equalassert_series_equal)idx	new_indexexp0exp1msglefttsrights           g/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/pandas/tests/indexes/multi/test_reshape.pytest_insertrD      s6   

1n--I!!#&&&&&Q<>)))) 

1.//Icjm$$w.W===D)*1-t444?w11111cjm$$y0x@@@D)*1-t444Q<+++++ <C	z	-	-	- ! !

1i   ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! <6@U@U@UVVVDNNE5>4N000	e		t		$	$B"#DH"$DH"#DH"#DH"#DH"#DHBF:BF8BF:BF8BF:BF8LMMMMMMNNMMMMMMMM		
 &%%  E 
OOUENDO111 $599992uU|,,,,,s   &E		EEc                     d t          d          D             d t          d          D             z   ddgz   } t          j        t          j        ddd	          t          j        | d d
                             }d	|j        d<   d|j        d<   t          j        t          j        ddd          t          j        |                     }t          j	        ||           d S )Nc                     g | ]}d |fS )test1 .0is     rC   
<listcomp>z test_insert2.<locals>.<listcomp>N   s    (((!'1(((    r$   c                     g | ]}d |fS )test2rH   rI   s     rC   rL   z test_insert2.<locals>.<listcomp>O   s    
*
*
*AGQ<
*
*
*rM   r'   )test   )rP      r   
            )
ranger4   Seriesnplinspacer   from_tuplesr8   r.   r:   )r;   r@   rB   s      rC   test_insert2r]   K   s     	)(uQxx(((
*
*q
*
*
*	+
&	'  9R[B++Z-CCH-M-MNNDDH\DH\Ibk!R,,j.DS.I.IJJE4'''''rM   c                    | d d                              | dd                    }|                    |           sJ | d d         | dd         | dd          g}|d                              |dd                    }|                    |           sJ |                      g           }|                    |           sJ d S )Nr    r   r   )appendequals)r;   resultfooss      rC   test_appendrc   ]   s    !W^^CG$$F==GS1Xs122w'D!W^^DH%%F== ZZ^^F==rM   c                     t          g d          } t          j        dddd          }t          g d          }t          j        | |g          }t          j        | ||g          }|                     |          }t          j        d          }d|                    t          d	d
d
                    fd|                    t          d	d
d                    fd|                    t          d	d
d                    fg}t          g d|z             }t          j        ||           |                    |           }t          |g dz             }t          j        ||           |                    |          }t          j        |                     |           |                    |          g          }t          j        ||           |                    |          }t          j        ||           |                    |          }t          j        t          j        d|                    t          d	d
d
                    dfd|                    t          d	d
d                    dfd|                    t          d	d
d                    dfg|z   t                    d           }t          j        ||           d S )N)皙?333333??z
2011-01-01Dr    z
Asia/Tokyo)freqperiodstz)ABCre   i  r   rf   r   rg   rl   rm   rn   )dtype)r   r4   
date_ranger   from_arraysr_   pytztimezonelocalizer   r.   r/   _simple_newrZ   arrayobject)	idx1idx2idx3midx_lv2midx_lv3ra   rk   expected_tuplesexpecteds	            rC   test_append_indexr   j   s   !!D=C|LLLD!!D%tTl33H%tT4&899H[[""F 
|	$	$B	bkk(4A..//0	bkk(4A..//0	bkk(4A..//0O
 ___677H&(+++__T""F_677H&(+++__X&&F%t{{4'8'8$++d:K:K&LMMH&(+++__X&&F&(+++__X&&F 
bkk(4A"6"677=bkk(4A"6"677=bkk(4A"6"677=
  	
 	
 	
 	 H &(+++++rM   z	name, expr!   )cNc                 
   t          j        ddgddggddg          }t          j        dgdggd| g          }|                    |          }t          j        g d	g d
gd|g          }t          j        ||           d S )Nr   r   r    r"   r   r   r0   r$   r   r   r    r    r"   r$   r   rq   r_   r.   r/   )r   expmidxmidx2ra   r~   s         rC   test_append_names_matchr      s     !Aq6Aq6"23*EEED"QC!:c4[AAAE[[F%yyy)))&<S#JOOOH&(+++++rM   c                     t          j        ddgddggddg          } t          j        dgdggd	d
g          }|                     |          }t          j        g dg dgd           }t          j        ||           d S )Nr   r   r    r"   r   r   r   r$   r   yr   r   r   )r   r   ra   r~   s       rC   test_append_names_dont_matchr      s    !Aq6Aq6"23*EEED"QC!:c3Z@@@E[[F%yyy)))&<DIIIH&(+++++rM   c                     t           j                            g d          } t           j                            g d          }t          j        | | g          }t          j        ||g          }|                    |          }t          j        t          j        dd          t          j        dd          ft          j        dd          t          j        dd          ft          j        dd          t          j        dd          ft          j        dd          t          j        dd          ft          j        dd          t          j        dd          ft          j        dd          t          j        dd          ft          j        dd          t          j        dd          ft          j        dd          t          j        dd          fg          }t          j	        ||           d S )	N)              ?       @)      ?      ?      @r   r   r   r   r   r   )
r4   IntervalIndexfrom_breaksr   from_productr_   r\   Intervalr.   r/   )ivl1ivl2mi1mi2ra   r~   s         rC   'test_append_overlapping_interval_levelsr      s   ''88D''88D

!4,
/
/C

!4,
/
/CZZ__F%[c""BKS$9$9:[c""BKS$9$9:[c""BKS$9$9:[c""BKS$9$9:[c""BKS$9$9:[c""BKS$9$9:[c""BKS$9$9:[c""BKS$9$9:		
 H &(+++++rM   c                     d} g d}t          j        ddg          }t          j        ||g|          }t          j        ||                    |           g|          }t          j        |                    |           |           d S )Nr   r   foor   r   )rZ   rv   r   r   repeatr.   r/   )repsnumbersr0   mr~   s        rC   test_repeatr      s    DiiGHeU^$$E% 0>>>A&d1C1C'DERRRH!((4..(33333rM   c                     | dd         }| dd                              |                    d| d                             sJ d S )Nr   r"   r   )r`   r*   )r;   ra   s     rC   test_insert_baser      sI    1XF qs8??6==CF334444444rM   c                    | dd          }|                      d          }|                    |          sJ |j        |j        k    sJ | d d         }|                      d          }|                    |          sJ |j        |j        k    sJ d}t          j        t
          |          5  |                      t          |                      d d d            d S # 1 swxY w Y   d S )Nr   r   r   z/index 6 is out of bounds for axis 0 with size 6r   )deleter`   r   r1   r2   
IndexErrorlen)r;   r~   ra   r?   s       rC   test_delete_baser      s*   122wHZZ]]F==""""";(-''''3B3xHZZ^^F==""""";(-''''
;C	z	-	-	-  

3s88                 s   .#CC"%C")r   numpyrZ   r1   rr   pandasr4   r   r   pandas._testing_testingr.   rD   r]   rc   r   markparametrizer   r   r   r   r   r   rH   rM   rC   <module>r      sD                               9- 9- 9-x( ( ($
 
 
,, ,, ,,^ z;&?@@, , A@,, , ,, , ,,4 4 45 5 5    rM   