
    PiQ                     
   d dl Z d dlZd dlZd dlZd dlmZmZmZm	Z	m
Z
mZ d dlmZ ej                            dddg          d             Zej                            ddd	g          d
             Zd Z G d d          ZdS )    N)	DataFrameDatetimeIndexIndex
MultiIndexSeries	Timestampzmsg,labels,level)zlabels \[4\] not found in level   a)zlabels \[7\] not found in level   bc                    t          j        g dg dgddg          }t          g d|          }t          g d|          }t	          j        t          |           5  |                    ||	           d d d            n# 1 swxY w Y   t	          j        t          |           5  |                    ||	           d d d            d S # 1 swxY w Y   d S )
N         r	         r
   r   names
         indexmatchlevel)r   from_arraysr   r   pytestraisesKeyErrordrop)msglabelsr    misdfs         x/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/pandas/tests/frame/methods/test_drop.py0test_drop_raise_exception_if_labels_not_in_levelr,      se    
	III 6sCj	I	I	IB|||2&&&A	<<<r	*	*	*B	xs	+	+	+ $ $	vU###$ $ $ $ $ $ $ $ $ $ $ $ $ $ $	xs	+	+	+ % %
e$$$% % % % % % % % % % % % % % % % % %s$    BBB*CCCzlabels,level)r	   r
   )r   r   c                 B   t          j        g dg dgddg          }t          g d|          }t          g d|          }|                    | |d	          }t          j        ||           |                    | |d	          }t          j        ||           d S )
Nr   r   r
   r   r   r   r   ignore)r    errors)r   r!   r   r   r%   tmassert_series_equalassert_frame_equal)r'   r    r(   r)   r*   
expected_sexpected_dfs          r+   test_drop_errors_ignorer5   %   s     
	III 6sCj	I	I	IB|||2&&&A	<<<r	*	*	*BeH==J1j)))''&h'??K"k*****    c            	         t          t          j                            d                              d          g dt          j        ddd                    } | j        g d	                                         } t          j
        t          d
          5  |                     ddg           d d d            d S # 1 swxY w Y   d S )Nr   )r   r   r
   r   c2012hr   )freqperiodscolumnsr   )r   r   r   r   not found in axisr   r
   r   )r   nprandomdefault_rngstandard_normalpd
date_rangeiloccopyr"   r#   r$   r%   )r*   s    r+   9test_drop_with_non_unique_datetime_index_and_invalid_keysrI   3   s    

	a  0088mFa888
 
 
B 
		#	#	%	%B	x':	;	;	;  
c
                 s   B;;B?B?c                      e Zd Zd Zd Zd Zd Zej        	                    d e
g dg dgddg           e
g dg dgdd	g           e
g dg dgdd	g                              d
dg           e
g dg dgddg                              d
dg          g          d             Zej        	                    dg dg dg          ej        	                    dg dgdgg          d                         Zej        	                    dg dg dg          ej        	                    dddgddgg          d                         Zej        	                    dg i  ej        g            eg d           eg            eg           g          d             Zd Zd Zd Zd Zd Zd Zd  Zd! Zej        	                    d"g d#          ej        	                    d$d%d&g          d'                         Zd( Zej        	                    d)d*d*gg          d+             Zd, Z d- Z!d. Z"d/ Z#ej        	                    d0dd	gd
fdgd1fg          d2             Z$d3 Z%d1S )4TestDataFrameDropc                    t          g dg dg dgg dg d          }d\  |j        _        |j        _        |                    d          }|                    d	d
          }|                                |                                }}|                    dd          }|J |                    d	d
d          }|J ||||fD ]&}|j        j        dk    sJ |j        j        dk    sJ 't          |j                  g dk    sJ d}t          j        t          |          5  |                    dg           d d d            n# 1 swxY w Y   t          j        t          |          5  |                    dgd
           d d d            n# 1 swxY w Y   |                    dgd          }	t          g dd          }
t          j        |	j        |
           |                    ddgd          }	t          ddgd          }
t          j        |	j        |
           |                    dgd
d          }	t          g dd          }
t          j        |	j        |
           |                    ddgd
d          }	t          d	dgd          }
t          j        |	j        |
           |                    g d          }	t          g dd          }
t          j        |	j        |
           d S )Nr   r   r	   r   r   r   r   r8   defr   r?   )firstsecondr   rQ   r   axisT)inplacerW   rX   rT   rU   z\['g'\] not found in axisr   gr.   r/   namer
   r9   rW   r/   rP   rR   )r   r   r]   r?   r%   rH   listr"   r#   r$   r   r0   assert_index_equal)selfr*   df_dropped_bdf_dropped_edf_inplace_bdf_inplace_ereturn_valueobjr&   droppedexpecteds              r+   test_drop_namesz!TestDataFrameDrop.test_drop_namesD   s   YY			999-!//#OO
 
 

 *;&rzwws||wwsw++%'WWYY		l#((d(;;####((1d(CC### ,lK 	0 	0C9>W,,,,;#x/////BJ???2222*]83/// 	 	GGSENNN	 	 	 	 	 	 	 	 	 	 	 	 	 	 	]83/// 	# 	#GGSEG"""	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# 	# ''3%'11w777
gmX666''3*X'66#s'222
gmX666''3%a'99x888
gox888''3*1X'>>#s(333
gox888 ''"X'..w777
gmX66666s$   *EEE3FFFc                 
   t          g dg dd          }t          j        |                    dd          |dg                    t          j        |                    ddgd          |g                     t          j        |                    g d	d
          |j        dgd d f                    t          j        |                    d
dgd          |j        ddgd d f                    t          j        t          d          5  |                    d           d d d            n# 1 swxY w Y   t          j        t          d          5  |                    dd           d d d            n# 1 swxY w Y   t          j        t          d          5  |                    ddg           d d d            n# 1 swxY w Y   t          j        t          d          5  |                    ddgd           d d d            n# 1 swxY w Y   t          j        t          d          5  |                    g dd           d d d            n# 1 swxY w Y   t          j        |                    dd          |           t          j        |                    d
dgd          |j        g dd d f                    t          j        |                    ddd          |           t          j        |                    ddgdd          |dg                    t          t          t          t          d          t          dd          t          d                              g d          }t          j        |                    dd          |dg                    t          j        |                    dd          |d                    t          j        |                    g           |           |                    t          g d                    }t          d          |_        t          j        |                    d d!          |j        d"gd d f                    t          j        |                    d d"gd
          |j        g d d f                    t          t          j                            d                              d#          t          d                    }||j        d
k              }|                    ||j        d
k             j        d$%          }|J t          j        ||           d S )&Nr   r   r   r	   )r   r   r   r   )ABrm   r   rV   rn   r?   )r   r   r   r   r   r   r   z\[5\] not found in axisr   r   z\['C'\] not found in axisCz#\['C', 'D', 'F'\] not found in axis)ro   DFr.   r[   r   r^   abc)r
   r
   r   r?   r
   r   )XYru   ru   rowsrv   )r   r   T)r'   rX   )r   r0   r2   r%   locr"   r#   r$   r_   ziprange	set_indexr   r?   rA   rB   rC   rD   r   r   )ra   simplenu_dfr*   ri   rf   s         r+   	test_dropzTestDataFrameDrop.test_drops   s   LLLAABB
fkk#Ak66uFFF
fkk3*9kEEvbzRRR
fkk)))!k<<fj!aaa>PQQQ
fkk1a&wk??QPQFTUTUTUIAVWWW]8+EFFF 	 	KKNNN	 	 	 	 	 	 	 	 	 	 	 	 	 	 	]8+GHHH 	% 	%KK!K$$$	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	% 	%]8+EFFF 	  	 KKA	  	  	  	  	  	  	  	  	  	  	  	  	  	  	 ]8+GHHH 	, 	,KKc
K+++	, 	, 	, 	, 	, 	, 	, 	, 	, 	, 	, 	, 	, 	, 	, ]8+QRRR 	1 	1KKaK000	1 	1 	1 	1 	1 	1 	1 	1 	1 	1 	1 	1 	1 	1 	1 	fkk!Hk==vFFF
KKAxK00&*YYY\2J	
 	
 	
 	fkk#AhkGGPPP
KKc
8K<<fcUm	
 	
 	

 U1XXuR||T%[[99::OOO
 
 
 	ejj1j55ucU|DDD
ejj9j==uSzJJJ
ejjnne444ooo 6 677U
ejj6j::EIseQQQh<OPPP
ejj#s!j<<eiAAA>NOOO I!!!$$44W==tE{{
 
 
 qk?wwbl&8$wGG###
b(+++++sZ   D00D47D4E::E>E> GGG*HHH6II #I c                 j   t          j        g dddg          }t          g dg|          }|j                                        sJ t          g dg dg d	g
          }|                    dddgd          }|                                }|j                                        rJ |                    dd                              t                    }t          j        |          5  |                    dd          }d d d            n# 1 swxY w Y   t          j        ||           d S )N))r
    )b1c1)b2c2r   r9   r   )r   r   r	   rt   )r
   r   r9   rP   )r   r   r   r   )r   r   r   r	   )r?   datar
   rP   )r   r?   valuesr   rV   )r   from_tuplesr   r?   _is_lexsortedpivot_tablereset_indexr%   astypefloatr0   assert_produces_warningr2   )ra   performance_warninglexsorted_milexsorted_dfnot_lexsorted_dfri   results          r+   "test_drop_multiindex_not_lexsortedz4TestDataFrameDrop.test_drop_multiindex_not_lexsorted   s    "-333C:
 
 
 !)))lCCC#1133333 %(((0B0B0BDVDVDV/W
 
 
 ,77Sz# 8 
 
 ,7799#+99;;;;;$$Sq$1188??'(;<< 	8 	8%**3Q*77F	8 	8 	8 	8 	8 	8 	8 	8 	8 	8 	8 	8 	8 	8 	8 	fh/////s   /DDDc                    t          g dg dg dgg dg d          }|                    d          }|                    d          }t          j        ||           |                    d	d
          }|                    d	          }t          j        ||           |                    dd
          }|                    d          }t          j        ||           |                    dgd          }|                    dg          }t          j        ||           |                    dgd                              d	gd
          }|                    dgd	g          }t          j        ||           d}t	          j        t          |          5  |                    dd           d d d            n# 1 swxY w Y   t	          j        t          |          5  |                    dd           d d d            n# 1 swxY w Y   d}t	          j        t          |          5  |                    d
           d d d            d S # 1 swxY w Y   d S )Nr   rM   rN   r8   rO   rS   r
   r   rP   r   rV   rt   rQ   )r'   rW   r   z2Cannot specify both 'labels' and 'index'/'columns'r   r   )r'   r   )r'   r?   z>Need to specify at least one of 'labels', 'index' or 'columns')r   r%   r0   r2   r"   r#   
ValueError)ra   r*   res1res2r&   s        r+   test_drop_api_equivalencez+TestDataFrameDrop.test_drop_api_equivalence   s   YY			999-!//#OO
 
 
 wws||wwSw!!
dD)))wwsw##wwsw##
dD)))wwcw**wwsw##
dD)))wwu1w%%wwcUw##
dD)))wwu1w%%**C5q*99wwcUSEw22
dD)))B]:S111 	+ 	+GG3cG***	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ 	+ ]:S111 	- 	-GG3G,,,	- 	- 	- 	- 	- 	- 	- 	- 	- 	- 	- 	- 	- 	- 	- O]:S111 	 	GGGOOO	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	s6   !GG	G	+HHH7II"Iactualr   r
   r   r   r   r   c                 l   t          |j        t                    rdnd }t          j        d          }t          j        t          |          5  |                    d|d           d d d            n# 1 swxY w Y   t          j        t          |          5  |j	                            d|d           d d d            n# 1 swxY w Y   |                    dd|d          }t          j        ||           |j	                            dd|d          }t          j        |j	        |           d S )	Nr   z"['c'] not found in axis"r   r9   r    rW   r   r.   )rW   r    r/   )
isinstancer   r   reescaper"   r#   r$   r%   Tr0   r2   )ra   r   r    r&   expected_no_errs        r+   "test_raise_on_drop_duplicate_indexz4TestDataFrameDrop.test_raise_on_drop_duplicate_index   s     j99Cti566]83/// 	2 	2KK5qK111	2 	2 	2 	2 	2 	2 	2 	2 	2 	2 	2 	2 	2 	2 	2]83/// 	4 	4HMM#UM333	4 	4 	4 	4 	4 	4 	4 	4 	4 	4 	4 	4 	4 	4 	4 ++cx+PP
ov666 (--!5-RR
o/88888s$   A33A7:A7CC
Cr   )r   r   r   drop_labelsr   c                     fd|D             }t          |                                        }t          j        |t          |                     d S )Nc                     g | ]}|v|	S  r   ).0ir   s     r+   
<listcomp>z:TestDataFrameDrop.test_drop_empty_list.<locals>.<listcomp>  s#    CCCa{.B.B!.B.B.Br6   r   r   r%   r0   r2   )ra   r   r   expected_indexframes     `  r+   test_drop_empty_listz&TestDataFrameDrop.test_drop_empty_list   sa     DCCCUCCC&&&++K88
eY^%D%D%DEEEEEr6   )r   r   r   r	   r   c                     t          j        t          d          5  t          |                              |           d d d            d S # 1 swxY w Y   d S )Nr@   r   r   )r"   r#   r$   r   r%   )ra   r   r   s      r+   test_drop_non_empty_listz*TestDataFrameDrop.test_drop_non_empty_list  s     ]8+>??? 	5 	5E"""''444	5 	5 	5 	5 	5 	5 	5 	5 	5 	5 	5 	5 	5 	5 	5 	5 	5 	5s   $AAAempty_listlikezdatetime64[ns]dtypec                     ddgddgd}t          d          t          d          g}t          ||          }|                                }|                    |          }t	          j        ||           d S )Nr   r   onetwo)column_acolumn_bz
2021-01-01r   )r   r   rH   r%   r0   r2   )ra   r   r   r   r*   ri   r   s          r+   2test_drop_empty_listlike_non_unique_datetime_indexzDTestDataFrameDrop.test_drop_empty_listlike_non_unique_datetime_index  s~     G%@@<(()L*A*ABt5))) 7799((
fh/////r6   c                    g dg dg dg}t          t          |           }t          j        |          }t	          t
          j                            d                              d          |          }|	                    dd	          }|	                    d
gd	          }t          j        ||           |	                    dgd	          }|	                    dgd	          }|	                    dgd	          }t          j        ||           |	                    dd	          }|	                    dgd	          }t          j        ||           |	                    dgd	          }|	                    dd	          }|	                    ddd          }|	                    ddgd	          }t          j        ||           d S )N)r
   topr   routine1r   routine2)r   ODr   result1result2r   )r   wxwyr   r   r   r   )r	   r   rt   r
   r   rV   )r
   r   r   r   )r   r   r   )r   r   r   r   r   )r   r   r   )r   r   r   )sortedry   r   r   r   rA   rB   rC   rD   r%   r0   r2   )ra   arraystuplesr   r*   r   ri   s          r+   test_mixed_depth_dropz'TestDataFrameDrop.test_mixed_depth_drop$  s   CCC===(((
 V%%&v..ry,,Q//??GGQVWWW1%%77M?733
h///%q))77/0q799=="5!6Q=??
h///,15577/0q799
h///77/0q799775q7))!!4477(*EFQ  
 
 	h/////r6   c           
      p   t          g dt          j        dddgg dg dd                              g d                                          }|                    d	d
          }t          ddgdgt          j        ddt          j        dfgg d                    }t          j	        ||           d S )N)r   r   r   r           g      ?g       @)r
   r   r9   r9   rl   rm   rn   ro   rp   )rm   rn   ro   r9   ro   r   r   r   rp   )r   r   r   r   r
   r   r>   )
r   rA   nanr{   
sort_indexr%   r   r   r0   r2   ra   r*   r   ri   s       r+   $test_drop_multiindex_other_level_nanz6TestDataFrameDrop.test_drop_multiindex_other_level_nanE  s     555&#sC0---%	   Y''Z\\ 	 C((FE("UBFC$89  
 
 
 	fh/////r6   c                     t          g dg dg dgg dg d          }d}t          j        t          |          5  |                    d	d
           d d d            d S # 1 swxY w Y   d S )Nr   rM   rN   r8   rO   rS   z0Cannot specify both 'axis' and 'index'/'columns'r   r   r   )r   rW   )r   r"   r#   r   r%   )ra   r*   r&   s      r+   (test_drop_raise_with_both_axis_and_indexz:TestDataFrameDrop.test_drop_raise_with_both_axis_and_index]  s    YY			999-!//#OO
 
 
 A]:S111 	' 	'GG#AG&&&	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	's   A""A&)A&c                    t          g dg dg dg dg dg dg dg dg d	g	g d
          }|                    d                                          }|j        |dk             }|                    g d          }|                    |j        d                                          }||j        	                    |j                            }|j        |_        t          j        ||           d S )N)x-axr
   g      ?)r   r   r
   333333?)zz-czr9   g@)r   r   r
   ffffff@)x-br   r   gffffff@)r   r   r   r   )r   r   r   g@)zy-ayr
   r   )zz-br   r   g @)var1var2var3var4rt   r   r   )r   r   r   r   r   )r   groupbysizerx   r{   r%   r   r   r   isinr0   r2   )ra   r*   grp_sizedrop_idxidfr   ri   s          r+   test_drop_nonuniquez%TestDataFrameDrop.test_drop_nonuniquei  s"   &&&&&&&&&&&&&&&&&&&&&&&&&&&
 544
 
 
 ::f%%**,,<A.ll33344 (.22>>@@rw||HN3334~
fh/////r6   c                    |}|                     ddgd          }|j        g d         }t          j        ||           |                     dgd          }|j        g d         }t          j        ||           |j                             ddgd	d
          }|j        g d         j        }t          j        ||           |j                             dgd	d
          }|j        g d         j        }t          j        ||           d S )NbarquxrT   r   )r   r   r   r   r   r   rU   )r   r   r   r   r   	   r   )rW   r    )r%   rG   r0   r2   r   )ra    multiindex_dataframe_random_datar   r   ri   s        r+   test_drop_levelz!TestDataFrameDrop.test_drop_level  s   0UEN':::ooo.
fh///UG844:0001
fh///uen1GDD:ooo.0
fh///ugAX>>:00013
fh/////r6   c                    t          g dd          }t          j        g d          }t          t	          j        d                              dd          t          d          |	          }||d
<   |                    d
d          }t          d          }|j
        j        du sJ |                    |d
          }|j        |dk             }t          j        ||           d S )N)r   r   r	   r	   r   idr\   )z2016-03-23 14:00z2016-03-23 15:002016-03-23 16:00r   z2016-03-23 17:00r   r   r   abr>   tstampT)append201603231600Fr   r	   )r   rE   to_datetimer   rA   arangereshaper_   r{   r   r   	is_uniquer%   rx   r0   r2   )ra   idxidxdtr*   tsr   ri   s          r+   "test_drop_level_nonunique_datetimez4TestDataFrameDrop.test_drop_level_nonunique_datetime  s    OOO$///  
 
 ry}},,Q22DJJcRRR8\\(4\00~&&x!U****8,,6#(#
fh/////r6   c                    t          dd          }t          dd          }t          j        ||d          } |dgt          |          z  |          }|                    |          }t          d	d          }t          j        ||d          } |dgt          |          z  |          }	t          j        ||	           d S )
Nz
2017-10-29zEurope/Berlin)tzz2017-10-29 04:00:0015min)r<   r   )r   r   z2017-10-29 00:15:00)r   rE   rF   lenr%   r0   assert_equal)
ra   frame_or_seriesstartendr   r   r   expected_startexpected_idxri   s
             r+   'test_drop_tz_aware_timestamp_across_dstz9TestDataFrameDrop.test_drop_tz_aware_timestamp_across_dst  s    ,?;;;-/BBBeSw777QC#e**$4EBBB5!!"#8_MMM}^SwGGG"?c,.?.?(?|TTT
)))))r6   c                    t          j        g dg dgddg          }t          t          j                            d                              d          |          }|                    d	g          }|j        j	        d
k    sJ d S )N)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   rA   rB   rC   rD   r%   r   r   )ra   r   r*   r   s       r+   test_drop_preserve_namesz*TestDataFrameDrop.test_drop_preserve_names  s    &!3!3!34UEN
 
 
 ry,,Q//??GGuUUU&""|!^333333r6   	operation)__iadd____isub____imul____ipow__rX   FTc                    t          dt          d          i          }|                                }t          d          |d<   |d         }t          j        d           5  |r|                    dd|           n|                    dd|          } t          ||          d           t          j        ||           d d d            d S # 1 swxY w Y   d S )Nr   r   r   r   rY   )r   rz   rH   r0   r   r%   getattrr2   )ra   r  rX   r*   ri   r   s         r+   test_inplace_drop_and_operationz1TestDataFrameDrop.test_inplace_drop_and_operation  s    U1XX''7799((3sG'-- 	0 	0 ;!W5555WWSq'W:: "GAy!!!$$$!"h///	0 	0 	0 	0 	0 	0 	0 	0 	0 	0 	0 	0 	0 	0 	0 	0 	0 	0s   !A"CCCc                    t          j        g dg dg          }t          g d|          }|                    d          }t          dgt          j        dgdgg                    }t	          j        ||           d S )	N)r   r   r   )r   jr   r   r   r   r   r   r  )r   r!   r   r%   r0   r2   )ra   r(   r*   r   ri   s        r+   $test_drop_with_non_unique_multiindexz6TestDataFrameDrop.test_drop_with_non_unique_multiindex  s    #___ooo$FGGyyy+++s##aS
(>u~(N(NOOO
fh/////r6   indexerr
   r
   c                 >   t          j        ddgddgg          }t          dt          t	          |                    i|          }|                    dg          }t          dddgit          j        ddg                    }t          j        ||           d S )	Nr
   r   r   r   r  r   r   )r   r
   )	r   from_productr   rz   r   r%   r   r0   r2   )ra   r  r   r*   r   ri   s         r+   *test_drop_tuple_with_non_unique_multiindexz<TestDataFrameDrop.test_drop_tuple_with_non_unique_multiindex  s     %SzC:&>??U3s88__-S999
|,,1a&M!7Z8P!Q!Q
 
 
 	fh/////r6   c                     t          g dg dg dgg d          }|                    dgd          }t          dgdgdggdg          }t          j        ||           |                    dd          }t          j        ||           d S )N)r   r   g      @)r   r
   r
   rt   r
   r   rV   r   r   r   s       r+    test_drop_with_duplicate_columnsz2TestDataFrameDrop.test_drop_with_duplicate_columns  s    [[+++{{{3=N=N=N
 
 
 #Q''qcA3_ug>>>
fh///1%%
fh/////r6   c                    t          t          j                            d                              d          t          j                            d                              d          t          j                            d                              d          g dd          }|                    g dd          }|                    g dd          }|                    d	d          }t          j        ||           d S )
Nr   r   )r
   r   r9   rP   rQ   r   )r   r   r   r   rV   )r   r   r   r   r   ro   )	r   rA   rB   rC   rD   taker%   r0   r2   )ra   r*   ri   df2r   s        r+   !test_drop_with_duplicate_columns2z3TestDataFrameDrop.test_drop_with_duplicate_columns2  s    Y**1--==a@@Y**1--==a@@Y**1--==a@@...	 
 
 7799917--ggoooAg..#A&&
fh/////r6   c                 b   t          dg dit          dgd                    }|j        }|                    dgdd           t	          j        |j        t          g d                     ||                                z  }t	          j        |j        t          g d                     d S )	Nr
   r   objectr   rt   r   TrY   )r   r   r
   r%   r0   r`   r?   mean)ra   r*   r
   s      r+   .test_drop_inplace_no_leftover_column_referencez@TestDataFrameDrop.test_drop_inplace_no_leftover_column_reference  s    YYY'uH1M1M1MNNND
At,,,
bj%(*C*C*CDDD	QVVXX
bj%(*C*C*CDDDDDr6   c                    t          t          j        t          d          t          d          g                    }t	          j        t          d          5  |                    dd           d d d            d S # 1 swxY w Y   d S )Nr   r   zlabels \[5\] not found in levelr   r   r   r   )r   r   r  rz   r"   r#   r$   r%   )ra   r*   s     r+   (test_drop_level_missing_label_multiindexz:TestDataFrameDrop.test_drop_level_missing_label_multiindex  s    Z4eAhha5IJJKKK]8+NOOO 	  	 GGAQG	  	  	  	  	  	  	  	  	  	  	  	  	  	  	  	  	  	 s   A??BBz
idx, levelNc                 J   t          dddt          j        gdd|                              |          }|                    t          dt          j        g          |          }t          dgdd|                              |          }t          j        ||           d S )Nr   r   d   r
   r   r   r   )r   rE   NAr{   r%   r   r0   r2   )ra   any_numeric_ea_dtyper   r    r*   r   ri   s          r+   test_drop_index_ea_dtypez*TestDataFrameDrop.test_drop_index_ea_dtype  s     aBE"--5I
 
 

)C.. 	 25z**%88#C  (<
 
 

)C.. 	 	fh/////r6   c                    t          ddgddgdt          d          t          d          g          }|                    dd          }t          dgdgdt          d          g          }t          j        ||           d S )	Nr   r   r!  z
2000-01-03z
2000-01-04r   r   rV   )r   r   r%   r0   r2   r   s       r+   &test_drop_parse_strings_datetime_indexz8TestDataFrameDrop.test_drop_parse_strings_datetime_index"  s    a&1v&&\**Il,C,CD
 
 
 A..A3aS11)L:Q:Q9RSSS
fh/////r6   )&__name__
__module____qualname__rj   r~   r   r   r"   markparametrizer   r{   r   r   r   rA   arrayr   r   r   r   r   r   r   r   r   r   r   r  r
  r  r  r  r  r  r  r$  r&  r   r6   r+   rK   rK   C   s(       -7 -7 -7^3, 3, 3,j0 0 04% % %N [Iyyy))),S#J???Iyyy))),S#J???Iyyy))),S#J???II1a&QQIyyy))),S#J???II1a&QQ		
 9 9 9 [Wyyy)))&<==[]R!qcN;;F F <; >=F [Wyyy)))&<==[]aVaV,<==5 5 >= >=5
 [BHRLLF2-...E"IIM"	

 
	0 	0
 
	00 0 0B0 0 00
' 
' 
'0 0 0:0 0 0&0 0 0,
* 
* 
*4 4 4 [EEE  [Y660 0 76 0"0 0 0 [Yj\(BCC0 0 DC00 0 00 0 0 E E E      [\c3ZOseT]+KLL	0 	0 ML	00 0 0 0 0r6   rK   )r   numpyrA   r"   pandasrE   r   r   r   r   r   r   pandas._testing_testingr0   r*  r+  r,   r5   rI   rK   r   r6   r+   <module>r1     sO   				                               44 	% 	% 	% (H)=>>
+ 
+ ?>
+   g0 g0 g0 g0 g0 g0 g0 g0 g0 g0r6   