
    Pi                         d dl Z d dl mZ d dlZd dlZd dlZd dlmZ d dlm	Z	 d dl
Zd dl
mZmZmZmZmZmZmZ ej        j        gZ ej        d          Zej                            d          d             Zd	 Zd
 Zd Zd Zd Zd Z d Z!d Z"d Z#d Z$d Z%ej        &                    dg d          d             Z'd Z(d Z)d Z*d Z+d Z,d Z-d Z.d Z/dS )    N	timedelta)	Timestamp)PY312)	DataFrameIndexSeries_testingconcat
date_rangeread_hdftablesz!ignore::tables.NaturalNameWarningc                    t          t          j                            d                              d          t          t          d                    t          ddd                    }|                     d	|d d
                    |                     d	|d
d                     t          j
        | d	         |           |                     d|d d
         d           |                     d|d
d                     t          j
        | d         |           |                     d|d d
                    |                     d|d
d                     t          j
        | d         |           |                     d|d d
                    |                     d|d
d                     t          j
        | d         |           t          ddgddgddgddgg          }d|d<   t          j        |j        d<   |                     d|           t          j
        | d         |           t          t          t          j                            d                              ddd          t          j                  t          t          j                            d                              ddd          t          j                  t          t          j                            d                              ddd          t          j                  t          g dt          j                  d t          j        d          !          }|                     d"|           t          j
        | d"         |d#$           |                     d"           |                     d"|g d%&           t          j
        | d"         |d#$           d S )'N   )      ABCD
2000-01-01r   Bperiodsfreqcolumnsindexdf1
   df2table)formatz/df3df3z/df3 foozdf3 foo   r   )datatestingmixed_column)r   r%   df      )highsizedtypei  i   @)l           l          @l            l            l            )u08u16u32u64r   uintsTcheck_index_type)r-   r.   r/   data_columns)r   nprandomdefault_rngstandard_normalr   listr   appendtmassert_frame_equalputnanlocr	   integersuint8uint16uint32uint64arangeremove)temp_hdfstorer&   	uint_datas      x/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/pandas/tests/io/pytables/test_append.pytest_appendrL      s    

	a  0099d6ll##r<<<
 
 
B
 3B3(((233(((-.333eRWW555233(((-.333CRC)))BCC)))-.333 RW---RW----	2B777 
!Q!Q!Q!Q8	9	9	9B"B~ "BFr"""--r222 	%%a((11!#A1FFh   	%%a((11!%a1HHi   	%%a((11!%a1HHi   333i  	
 	
$ ill'  I* ),,,-0)dSSSS !!!):O:O:OPPP-0)dSSSSSS    c                    t          t          d          t          j        d t          d          D                       }t          t          j        dt          j                  t          dd                    }t          t          j        d	                    }|                     d
|           | d
         }t          j        ||           |j	        J |                     d|           | d         }t          j        ||           |j	        J d|_	        |                     d|           | d         }t          j        ||           |j	        |j	        k    sJ ||dk             }| 
                    dd          }t          j        ||           ||dk    |j        dk     z           }t          |j        j                  |_        | 
                    dd          }t          j        ||d           t          t          j                            d                              d          dg          }t          j        t%          |                    |d<   d|d<   d|j        dddf<   |                    ddgd           |                                }|j                            d          |_        |                     d|           t          j        | d         |d           d S ) Nr   c                     g | ]}d | S )i_ .0is     rK   
<listcomp>z&test_append_series.<locals>.<listcomp>_   s    3P3P3PHHH3P3P3PrM   )r,   r   r   r+   z
2020-01-01r   r1   d   sstsfoons<   zfoo>60F   Z   zfoo>70 and index<90Tr3   r   )r(   r"   Ar   r   Cbar   r(   inplacemi)r	   ranger7   float64rG   r   r<   r=   assert_series_equalnameselectr   r   _datar   r8   r9   r:   lenrA   	set_indexstack	droplevel)rI   rX   rY   r[   resultexpectedrf   ss           rK   test_append_seriesrt   ]   s   	b		3P3PeBii3P3P3P	Q	Q	QB	
	"BJ'''z,PR/S/S/S
 
 
B 
	#		Br"""4 F62&&&;r"""4 F62&&&;BGr"""4 F62&&&;"'!!!! "r'{H!!$11F68,,, 27rx"}-.H8>/00HN!!$(=>>F68dCCCC 
29((++;;FCCcU	S	S	SBiB  BsGBsGBF1Q38LL#sTL***


Ag""AGq!!!=.DIIIIIIrM   c                 4   t          t          t          j                            d                              d                                        d          t          j                            d                              d          t          j                            d                              d          ddt          d                              d          t          d                              d          d	t          j	        d          
          }t          j
        |j        ddg df<   |                     d|d d                    |                     d|dd                     t          j        | d         |d           |                                }t          j
        |d<   |                     d           |                     d|d d                    |                     d|dd                     t          j        | d         |d           |                                }t          j
        |d<   |                     d|d d                    |                     d|dd                     t          j        | d         |d           |                                }t          j
        |d<   |                     d|d d                    |                     d|dd                     t          j        | d         |d           d S )Nr   r   int32rZ   rb   
2001-01-01r[   
2001-01-02)r_   A1A2r   ra   DEr1   r      )ry   r   r{   r|   r   r   Tr3   ry   rz   r   r|   r!   )r   r	   r7   r8   r9   r:   astyper   as_unitrG   r@   rA   r<   r=   r>   copyrH   )rI   r&   r   r   r!   s        rK   test_append_some_nansr      s   		--a00@@DDEELLWUU)''**::2>>)''**::2>><((0066<((0066	
 	
 imm
 
 
B +-&BF1R4&&&&'3B3(((233(((-.TJJJJ ''))CCICRC)))BCC)))-.dKKKK ''))CCICRC)))BCC)))-.dKKKK ''))CvCHCRC)))BCC)))-.dKKKKKKrM   c                    t          t          j                            d                              d          t          j                            d                              d          dt          j        d                    }t          j        |j        ddd d f<   |                     d|d d         d	
           |                     d|dd          d	
           t          j
        | d         |dd          d	           |                     d|d d         d
           |                     d|dd          d
           t          j
        | d         |d	           t          j        dd          5  |                     d|d d                    |                     d|dd                     t          j
        | d         |           d d d            n# 1 swxY w Y   t          j        dd	          5  |                     d|d d                    |                     d|dd                     t          j
        | d         |dd                     t          t          j                            d                              d          t          j                            d                              d          dddt          j        d                    }t          j        |j        ddd d f<   |                     d           |                     d|d d         d	
           |                     d|dd          d	
           | d         }|}|r
|dd          }t          j
        ||d	           |                     d           |                     d|d d         d
           |                     d|dd          d
           t          j
        | d         |d	           t          t          j                            d                              d          t          j                            d                              d          ddt          d                              d          t          d                              d          dt          j        d                    }t          j        |j        ddd d f<   |                     d           |                     d|d d         d	
           |                     d|dd          d	
           t          j
        | d         |d	           |                     d           |                     d|d d         d
           |                     d|dd          d
           t          j
        | d         |d	           d d d            d S # 1 swxY w Y   d S )Nr   r   )ry   rz   r1   r   r}   r&   r   T)dropnar3   r   Fio.hdf.dropna_tabler!   df4rZ   rb   )ry   rz   r   ra   rw   r[   rx   )ry   rz   r   ra   r{   r|   )r   r7   r8   r9   r:   rG   r@   rA   r<   r=   r>   pdoption_contextrH   r   r   )rI   using_infer_stringr&   rq   rr   s        rK   test_append_all_nansr      sU   	)''**::2>>)''**::2>>	
 	
 imm
 
 
B fBF1R47O r#2#wt444r"##wt444--r"##wNNNN 3B3666233666-.TJJJJ 
	0%	8	8 8 8UBssG,,,UBrssG,,,
mE2B7778 8 8 8 8 8 8 8 8 8 8 8 8 8 8
 
	0$	7	7 :O :OUBssG,,,UBrssG,,,
mE2BrssG<<< i++A..>>rBBi++A..>>rBB	  )B--
 
 
 &qtQQQwT"""T2crc74888T2bcc74888t$ 	%  }H
fhFFFFU###UBssGE:::UBrssGE:::
mE2BNNNN i++A..>>rBBi++A..>>rBB|,,44T::|,,44T::  )B--

 

 

 &qtQQQwT"""T2crc74888T2bcc74888
mD12MMMMU###UBssGE:::UBrssGE:::
mE2BNNNNu:O :O :O :O :O :O :O :O :O :O :O :O :O :O :O :O :O :Os&   8AGG #G <N>WWWc           	      R   t          t          j                            d                              d          t          t          d                    t          ddd                    }|j        	                    d           |_        | 
                    d	|j        d d d df         d
g           | 
                    d	|j        d d dd f                    t          j        | d	         |           |                     d	d          }|                    dg          }t          j        ||           |                    t"          j                            t(          dt*                               |                     d	d          }|                    dg|j        dd                   }t          j        ||           t-          j        d          }t#          j        t2          |          5  |                     d	d           d d d            d S # 1 swxY w Y   d S )Nr   r   r   r   r   r   r   r   r   r   r   )axes	columns=Ar_   r`   zAST change in PY312)reasonraises)r   zindex=df.index[0:4]r   r   zOpassing a filterable condition to a non-table indexer [Filter: Not Initialized]matchzcolumns=A and index>df.index[4])r   r7   r8   r9   r:   r   r;   r   r   
_with_freqr<   ilocr=   r>   rk   reindexapplymarkerpytestmarkxfailr   
ValueErrorreescaper   	TypeError)rI   requestr&   rq   rr   msgs         rK   !test_append_frame_column_orientedr     sf   	
	a  0099d6ll##r<<<
 
 
B
 x""4((BH2A2i[AAA122///-.333!!%55Fzz3%z((H(F+++ ( 	 	
 	
   !!%)MNNFzz3%rx!}z==H(F+++ )	$ C 
y	,	,	, G GU$EFFFG G G G G G G G G G G G G G G G G Gs   8HH #H c                 <   t          d          D ]A}t          t          j                            d                              d          t          d                    }t          d          |d<   |dxx         |dz  z  cc<   t          dgt          |          z  d	          |d<   t          dgt          |          z  d
	          |d
<   |dz  dk    r(|d= t          dgt          |          z  d	          |d<   |dz  dk    r|	                    d          }||d<   |
                    dd           |                     d|           Ct          t          j                            d                              d          t          d          d          }t          dgt          |          z  d	          |d<   t          dgt          |          z  d
	          |d
<   |                     d           |                     d|           t          dgt          |          z  d
	          |d<   t          j        d          }t          j        t"          |          5  |                     d|           d d d            n# 1 swxY w Y   t          dgt          |          z  d	          |d<   t          j        d          }t          j        t"          |          5  |                     d|           d d d            d S # 1 swxY w Y   d S )Nr   r   )r   r   ABr`   r   r"   int64r+   int16r   rc   r_   Trd   r&   rh   )r   r,   int16_2zCcannot match existing table structure for [int16] on appending datar         ?float_3zAcannot match existing table structure for [A,B] on appending data)rg   r   r7   r8   r9   r:   r;   r	   rm   poprn   r<   rH   r   r   r   r   r   )rI   rT   r&   ar   s        rK   )test_append_with_different_block_orderingr   4  s}   2YY ' 'I!!!$$44W==tDzz
 
 
 Bii7
7q2vaS3r77]':::7aS3r77]':::7q5A::7 !s2wwg>>>BwKq5A::sABsG
Wd+++T2&&&& 

	a  0099T


 
 
B
 !s2wwg666BwK!s2wwg666BwKr""" A3R=888ByM
)M C 
z	-	-	- ' 'T2&&&' ' ' ' ' ' ' ' ' ' ' ' ' ' ' C53r77?)<<<ByM
)W
X
XC	z	-	-	- ' 'T2&&&' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' 's$   +JJJ-LLLc           	      	     fd}t          ddgddgg          }                     d|           t          j                             d          |            |ddd           t          d	d
gddgg          }                     d|           t          ||g          }t          j                             d          |            |ddd           t          ddgddgg          }                     d|ddi           t          j                             d          |            |ddd                                d|           t          d	dgddgg          }d}t          j        t          |          5                       d|           d d d            n# 1 swxY w Y   t          g dg dt          g d          t          dd          d                              d          }                     d|d         d d!i           t          j                             d          |d                                         d"|d         d#d d!i$           t          j                             d"          |d                                         d%|d&d d'i(           |                                                                                    d)*                              d          }                     d%|           t          j                             d%          t          ||g                                          d+|d         d&d d'i(                                d+|d                    t          j                             d+          t          |d         |d         g                     t          t"          j                            d,                              d-          t          t+          d.                    t          d/d0d1          2          }d3|d4<   t"          j        |j        |j        d5d!         d4f<   d6|d7<   t"          j        |j        |j        d!d8         d7f<   d9|d:<   t"          j        |j        |j        d5d          d:f<                        d;|                                d;          }t          j        ||           d S )<Nc                 x    t                              |           j        j        |          j        |k    sJ d S Ngetattr
get_storerr   descriptionitemsizekeyrj   r*   rI   s      rK   	check_colz+test_append_with_strings.<locals>.check_colf  H    M,,S117CTJJS     rM   {   	asdqwertyiY  dggnhebbsdfbdfbdf_bigvalues_block_1r}   |   asdqyiZ  dggnhefbdfbdf_big2values2   min_itemsizedf_new
abcdefqhijabcdefghijklmnopqrtsuvwxyzzTrying to store a string with len \[26\] in \[values_block_1\] column but\nthis column has a limit of \[15\]!\nConsider using min_itemsize to preset the sizes on these columnsr   )        r   g       @g      @g      @)r   r   r   r   r   )foo1foo2foo3foo4foo520130101r(   rV   )r_   r   ra   r{   ra   rX   r   r   r   ss2Tr6   r   ss3r      )r    r   longer)ra   ss4r   r   r   r   r   r   r   rZ   stringr"   rb   string2   bahstring3r&   )r   r<   r=   r>   rk   r   r   r   r   r   r   rn   ri   r?   r   reset_indexassignr7   r8   r9   r:   r;   r@   rA   r   )rI   r   r&   r   rr   r   r   rq   s   `       rK   test_append_with_stringsr   e  sQ   
 
 
 
 
 
S+&.?(@A	B	BB2&&&-..x88"===Ih("--- c7^c=%9:
;
;C3'''r3i  H-..x88(CCCIh("--- 
S+&.?(@A	B	BBBh^DDD-..y992>>>Ii)2... 2&&&l+c3O-PQRRF	  
z	-	-	- / /Xv.../ / / / / / / / / / / / / / / 
******???@@J222		
 	

 
 inn  r#wgq\BBB=//55r#w??? 3d'STVVV=//663@@@ eRwlKKK
''))


!
!
(
(8
(
4
4
>
>s
C
CC$$$-..u55vr3i7H7HIII eRWWGQ<PPPC)))=//663S?R8S8STTT 

	a  0099d6ll##r<<<
 
 
B
 BxL&(fBF28AaC=("#ByM')vBF28AaC=)#$ByM&(fBF28ABB<"#r"""!!$''F&"%%%%%s   F''F+.F+c                 t     fd}t          dddt          d                    }                     d|dd	i
            |ddd	                                d          j        dgk    sJ                      d                                d|dgdd	i            |ddd	                                d          j        ddgk    sJ                      d                                d|dgdd	i            |ddd	            |ddd	                                d          j        dgk    sJ                      d                                d|d d         d	
                                d|dd          d	
           t          j         d         |           t          g ddg          }                     d           t          j	        d          }t          j        t          |          5                       d|ddd
           d d d            d S # 1 swxY w Y   d S )Nc                 x    t                              |           j        j        |          j        |k    sJ d S r   r   r   s      rK   r   z,test_append_with_strings2.<locals>.check_col  r   rM   rZ   rb   r_   r   r   r1   r&   r_      r   r   r   r   values_block_0r(   )rZ   rZ   rZ   barhr   r   r`   zBmin_itemsize has the key [foo] which is not an axis or data_columnr   r   )rZ   foobar)r   rg   r<   r   r6   rH   r=   r>   r   r   r   r   r   )rI   r   r&   r   s   `   rK   test_append_with_strings2r     s   
 
 
 
 
 
U++599	=	=	=B rc
;;;IdC##D))63%???? rS#JOOOIdC##D))63*DDDD rXsOTTTIdCId$c***##D))63%???? r"1"vC888r!""vC888--r222 
@@@3%	P	P	PB
)L C 
z	-	-	- O OT2B"4M4MNNNO O O O O O O O O O O O O O O O O Os   H--H14H1c                    t          dg di          }|                     d|d d         ddi           |                     d|dd          ddi           t          j        |                     d          |           d S )Nx)r   bcdef r&   r"   r   )r   r<   r=   r>   rk   )rI   r&   s     rK   test_append_with_empty_stringr     s    	C;;;<	=	=Br#2#wc1X>>>r"##wc1X>>>-..t44b99999rM   c           
          t          t          j                            d                              d          t          t          d                    t          dddd          	          }d
|j        d|j	        
                    d          f<                        d|d d         dg                                d|dd                     t          j         d         |            j        j        j        j        j        j        j        du sJ  j        j        j        j        j        j        j        du sJ                      dd          }||j        dk             }t          j        ||                                dd          }|                    |j        dd                    }||j        dk             }t          j        ||           |                                }d|d<   t          j        |j        |j        dd         df<   d|j        |j        dd         df<                        d                                d|dg                                dd          }||j        dk             }t          j        ||            fd}                     d                                d|dgddi            |ddd                                d                                d|dgd            |ddd                                d                                d|dgddi            |ddd           d|d<   d |d!<   d"|d#<                        d                                d|ddgdd$d%d&            |ddd            |ddd$            |dd'd%           |                                }d
|j        d|j	        
                    d(          f<   d)|j        d|j	        
                    d          f<   d|d<   |j	        
                    d          }t          j        |j        dd|f<   d|j        dd|f<   d|d<   |j	        
                    d          }t          j        |j        dd|f<   d|j        d*d+|f<                        d                                d|g d,                                dd-          }||j        dk    |j        dk    z  |j        dk    z  |j        dk     z           }t          j        ||d./                                dd0          }||j        dk    |j        d1k    z           }t          j        ||           |                                }d|d<   t          j        |j        |j        dd         df<   d|j        |j        d*d2         df<   d1|d<   t?          d3                               d          |d4<   t          j        |j        |j        d5d         g d6f<                        d7|g d8                                d7d          }||j        dk             }t          j        ||                                d7g d9          }||j        dk    |j!        dk    z  |j        dk    z           }t          j        ||d./           t          d:d+;          }t          t          j                            d                              d<          |g d=>          }d|d<   t          j        |j        |j        dd         df<   d|j        |j        d*d2         df<   |dd?g         "                                |dd?g<   d1|d<                        d7                                d7|g d@                                d7d          }||j        dk             }t          j        ||                                d7g dA          }||j        dk    |j!        dk    z  |j        dk    z           }t          j        ||           d S )BNr   r   r   r   r   r   r[   )r   r   unitr   r   r   r&   r5   TB>0zB>0 and index>df.index[3]r   r1   rZ   r   r"   rb   r(   r   zstring='foo'c                 x    t                              |           j        j        |          j        |k    sJ d S r   r   r   s      rK   r   z0test_append_with_data_columns.<locals>.check_col  r   rM      r   r   	foobarbahr   
foobarbah1string_block1
foobarbah2string_block2(   r   )r   r   r   r   r_   g         r   )r_   r   r   r   z.string='foo' and string2='foo' and A>0 and B<0F)
check_freqzstring='foo' and string2='cool'cool	   20010102datetimerc   )r_   r   r   df_dc)r   ra   r   r   r   )B > 0C > 0zstring == fooz1/1/2000rV   )r   rc   )r_   r   ra   )r   r   ra   )r   ra   r   r   )r   r  zstring == "foo")#r   r7   r8   r9   r:   r   r;   r   r   r   get_locr<   r=   r>   _handlerootr&   r   colsr   
is_indexedr   rk   r   r   r@   rA   rH   r   r   r_   r   r   ra   abs)	rI   r&   rq   rr   r   r   slr   r   s	   `        rK   test_append_with_data_columnsr	    s   	
	a  0099d6ll##r$GGG
 
 
B
 +.BGArz!!#&&&'r"1"vSE:::r!""v&&&--r222  %(.39DLLLL %(.35@DHHHH !!$..F"$(|H&(+++ !!$(CDDFZZbhqrrlZ++Ffhl#H&(+++ WWYYFF8.0fFJv|AaC (*+.3FJv|AaC (*+vXJ???!!$77Ffmu,-H&(+++
 
 
 
 
 fH:XrN     IdHb!!!vXJRPPPIdHb!!!fH:XrN     IdHb!!!#F9*F?*F?	* "rR@@	     IdHb!!!IdIr"""Id$b))) WWYYF25FK6>))#.../26FK6>))#.../F8				)	)B6FK!R FK!RF9					*	*B6FK!R FK!Rv4S4S4STTT!!> F 	%	>U"	$8a<	 8a<	H &(u==== !!$(IJJFv}-&.F2JKLH&(+++ GGIIEE(O,.FEIek!A#(),1EIek!A#()E)!*--55d;;E*:<&EIek!A# 6 6 667%P%P%P     !!'511FUWq[!H&(+++!!'+N+N+NOOFegkegk2ele6KLMH&(u==== z1---E
	a  0088  E
 E(O,.FEIek!A#(),1EIek!A#()sCj)--//E3*E) !!!%6U6U6UVVV!!'511FUWq[!H&(+++!!'+P+P+PQQFegkegk2ele6KLMH&(+++++rM   c                    |}d |j         _        |                     d|           |                     d          }t	          j        ||           |                     dddg          }|                    ddg          }t	          j        ||           |                    | dd           t          | dddg          }|                    ddg          }t	          j        ||           d S )Nrf   r_   r   r`   r&   r   )r   r    )	r   rj   r<   rk   r=   r>   r   to_hdfr   )rI    multiindex_dataframe_random_datar&   rq   rr   s        rK   test_append_hierarchicalr    s    	)BBJOr"""!!$''F&"%%% !!$c
!;;Fzz3*z--H&(+++IImgI666mTC:>>>Fzz3*z--H&(+++++rM   c           
         t          dt          j        d                              d          z  t	          t          d                    t	          d t          d          D                                 }|                     d|d	
           |                     d          }t          j
        ||           |                     d|d           |                     d          }t          j
        ||           d S )N皙?x   r   r   r   c                     g | ]}d | S zi-rQ   rR   s     rK   rU   z$test_append_misc.<locals>.<listcomp>      111!X!XX111rM   r   r   r&   r"   	chunksizer   r   )expectedrows)r   r7   rG   reshaper   r;   rg   r<   rk   r=   r>   )rI   r&   rq   s      rK   test_append_miscr    s    	binn$$W---d6ll##11uRyy11122
 
 
B
 rQ///!!$''F&"%%%444!!%((F&"%%%%%rM   r  )r   r   i  c           
      |   t          dt          j        d                              d          z  t	          t          d                    t	          d t          d          D                                 }d|d	<   d
|d<   |d                             d          |d<   |d         dk    |d<   t          d          	                    d          |d<   t          d          	                    d          |d<   | 
                    d||           |                     d          }t          j        ||           d S )Nr  r  r  r   c                     g | ]}d | S r  rQ   rR   s     rK   rU   z.test_append_misc_chunksize.<locals>.<listcomp>  r  rM   r   r   rZ   r   r   float322float32r   boolr   r[   time120130102time2objr  )r   r7   rG   r  r   r;   rg   r~   r   r   r<   rk   r=   r>   )rI   r  r&   rq   s       rK   test_append_misc_chunksizer#    s-    
binn$$W---d6ll##11uRyy11122
 
 
B
 BxLBzN
^**955BzNJ!#BvJJ''//55BwKJ''//55BwKi888!!%((F&"%%%%%rM   c                 F   t          t          d                    }|                     d|           t          j        t
          d          5  |                     d           d d d            n# 1 swxY w Y   t          t          j        	                    d                              d          t          d                    }|                     d|           t          j        |                     d          |           |                     d|           t          j        |                     d          |           t          t          d                    }|                     d|           t          j        |                     d          |           d S )	NABCr`   r&   z 'No object named df in the file'r   r   )r   rc   r   )r   r;   r<   r   r   KeyErrorrk   r7   r8   r9   r=   r>   r?   )rI   df_emptyr&   s      rK   test_append_misc_empty_framer(    s    e---Hx(((	x'I	J	J	J # #T"""# # # # # # # # # # # # # # # 
29((++227;;T%[[	Q	Q	QBr"""-..t44b999x(((-..t44b999 
4;;	'	'	'BeR   -..u55r:::::s   A11A58A5c           
      L
   t          dt          j        d                              d          z  t	          t          d                    t	          d t          d          D                                 }dggt          |          z  |d	<   |j        d	         t          j	        k    sJ t          j        d
          }t          j        t          |          5  |                     d|           d d d            n# 1 swxY w Y   dggt          |          z  |d<   dggt          |          z  |d<   t          j        t          |          5  |                     d|           d d d            n# 1 swxY w Y   t          dt          j        d                              d          z  t	          t          d                    t	          d t          d          D                                 }t!          t#          j        ddd          |j                  }|                    t(                    }t          j        |dd<   ||d	<   |j        d	         t          j	        k    sJ d}t          j        t          |          5  |                     d|           d d d            n# 1 swxY w Y   d}t          j        t          |          5  |                     dt          j        d                     d d d            n# 1 swxY w Y   t          j        d          }t          j        t          |          5  |                     dt!          t          j        d                               d d d            n# 1 swxY w Y   t          dt          j        d                              d          z  t	          t          d                    t	          d t          d          D                                 }|                     d|           d|d<   t          j        d          }t          j        t,          |          5  |                     d|           d d d            n# 1 swxY w Y   t/          d          |d<   |                     d           |                     d|           d|d<   t          j        d          }t          j        t,          |          5  |                     d|           d d d            d S # 1 swxY w Y   d S ) Nr  r  r  r   c                     g | ]}d | S r  rQ   rR   s     rK   rU   z%test_append_raise.<locals>.<listcomp>  r  rM   r   r   r   invalidziCannot serialize the column [invalid]
because its data contents are not [string] but [mixed] object dtyper   r&   invalid2invalid3c                     g | ]}d | S r  rQ   rR   s     rK   rU   z%test_append_raise.<locals>.<listcomp>  r  rM   i  r"   r   r1   r   r(   z>too many timezones in this block, create separate data columnsz(value must be None, Series, or DataFramer   zQcannot properly create the storer for: [group->df,value-><class 'pandas.Series'>]c                     g | ]}d | S r  rQ   rR   s     rK   rU   z%test_append_raise.<locals>.<listcomp>	  r  rM   rZ   zinvalid combination of [non_index_axes] on appending data [(1, ['A', 'B', 'C', 'D', 'foo'])] vs current table [(1, ['A', 'B', 'C', 'D'])]r   rb   znCannot serialize the column [foo] because its data contents are not [string] but [datetime64[us]] object dtype)r   r7   rG   r  r   r;   rg   rm   dtypesobject_r   r   r   r   r   r<   r	   r   r   r~   objectr@   r   r   rH   )rI   r   r&   r   rs   s        rK   test_append_raiser3    s|    
binn$$W---d6ll##11uRyy11122
 
 
B
 UGc"gg%ByM9Y2:----
)	G C 
y	,	,	, ' 'T2&&&' ' ' ' ' ' ' ' ' ' ' ' ' ' ' eWs2ww&BzNeWs2ww&BzN	y	,	,	, ' 'T2&&&' ' ' ' ' ' ' ' ' ' ' ' ' ' ' 
binn$$W---d6ll##11uRyy11122
 
 
B
 	x q!,,BH===A	AVAacFByM9Y2:----
JC	y	,	,	, ' 'T2&&&' ' ' ' ' ' ' ' ' ' ' ' ' ' ' 5C	y	,	,	, 2 2T29R==1112 2 2 2 2 2 2 2 2 2 2 2 2 2 2 )	5 C 
y	,	,	, : :T6")B--#8#8999: : : : : : : : : : : : : : : 
binn$$W---d6ll##11uRyy11122
 
 
B
 r"""BuI
)	& C
 
z	-	-	- ' 'T2&&&' ' ' ' ' ' ' ' ' ' ' ' ' ' ' *%%BuIr"""BuI
)	, C
 
z	-	-	- ' 'T2&&&' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' 's~   C??DDE66E:=E:J00J47J4)LLL6N		NNQ99Q= Q=5TT Tc                    t          d                              d          t          fdt          d          D             d          }|d         |d         z
  |d<   |d                             d	| d
          |d<   t
          j        |j        dddf<   |                     d|d           | 	                    d          }t          j        ||           | 	                    dd          }t          j        ||           | 	                    dd          }t          j        ||j        dd                     | 	                    dd          }t          j        ||j        dd                     | 	                    dd          }|                    dg          }t          j        ||j        dd                     | 	                    dd          }|j        dd          }t          j        ||j        dd                     |                     d|           | 	                    d          }t          j        ||           d S )Nr   r[   c                 8    g | ]}t          |d           z   S )r   )dayssecondsr   )rS   rT   rY   s     rK   rU   z.test_append_with_timedelta.<locals>.<listcomp>,  s+    HHH"ya4444HHHrM   r   r   r_   r   ra   zm8[]rc   r(   r&   Tr5   zC<100000)wherezC<pd.Timedelta('-3D')zC<'-3D'zC<'-500000s')subsetr   z	C<'-3.5D'r"   r   r   )r   r   r   rg   r~   r7   r@   rA   r<   rk   r=   r>   r   r   r?   )rI   r   r&   rq   rY   s       @rK   test_append_with_timedeltar;  $  sL    
:			&	&t	,	,B	HHHHeBiiHHH	
 	

 
B g3BsGgnn]4]]]++BsGvBF1Q38 r555!!$''F&"%%%!!$j!99F&"%%%!!$.E!FFF&"'!""+...!!$	22F&"'!""+... !!$77F]]3%]((F&"'!""+...!!$44F[_F&"'!""+... eR   !!%((F&"%%%%%rM   c           	         t          t          j                            d                              d          t          t          d                    t          ddd                    }|                                	                    d	j
        
          }d|d<   t          ||gd          }d}t          j        t          |          5  |                     ddgd d|d           d d d            n# 1 swxY w Y   t          j        t          |          5  |                     d d d|d           d d d            n# 1 swxY w Y   d}t          j        t          |          5  |                     d|d           d d d            n# 1 swxY w Y   |                     ddgd d|d           |                     ddgddgd          }||j        dk    |j        dk    z           }t'          j        ||           d S )Nr   r   r   r   r   r   r   r   {}_2r`   rb   rZ   r"   axisz=append_to_multiple requires a selector that is in passed dictr   r_   r   r   r!   )selectorzQappend_to_multiple must have a dictionary specified as the way to split the valuer   r   zA>0r   )r9  rA  r   )r   r7   r8   r9   r:   r   r;   r   r   renamer    r   r   r   r   append_to_multipleselect_as_multipler_   r   r=   r>   )rI   r   r   r&   r   rq   rr   s          rK   test_append_to_multiplerE  Q  s   

	a  0099d6ll##r<<<  C
 ((**

FM

2
2CCJ	c
	#	#	#B JC	z	-	-	- 
 
((#Jt,,b5 	) 	
 	
 	

 
 
 
 
 
 
 
 
 
 
 
 
 
 

 
z	-	-	- Y Y((d)C)CRRW(XXXY Y Y Y Y Y Y Y Y Y Y Y Y Y Y	  
z	-	-	- ; ;((E:::; ; ; ; ; ; ; ; ; ; ; ; ; ; ; $$c
4(("u %    --	uenu .  F 24!8q)*H&(+++++s6   C99C= C=EEE/FFFc           	         t          t          j                            d                              d          t          t          d                    t          ddd                    }t          t          j                            d                              d          t          t          d                    t          ddd                                        d	j	        
          }t          j
        |j        d|j                            ddg          f<   t          ||gd          }|                     ddgd d|dd           |                     ddg          }|                                }t%          j        ||d           t%          j        |                     d          j        |                     d          j                   d S )Nr   r   r   r   r   r   r   r   r=  r`   r"   r_   r>  r@  r   TrA  r   r   r3   )r   r7   r8   r9   r:   r   r;   r   rB  r    r@   r   r   get_indexerr   rC  rD  r   r=   r>   assert_index_equalrk   r   )rI   r   r   r&   rq   rr   s         rK   test_append_to_multiple_dropnarJ  w  s   

	a  0099d6ll##r<<<  C
 
	a  0099d6ll##r<<<   fV]f##	 
 8:vCHQ''c
3334	c
	#	#	#B $$c
4(("uT %    --uen==Fyy{{H&(TBBBBU##)=+?+?+F+F+L    rM   c           	      ~   t          t          j                            d                              d          t          t          d                    t          ddd                    }|                                	                    d	j
        
          }t          j        |j        d|j                            ddg          f<   t          ||gd          }t!          j        dd          5  |                     ddgd d|dd           d}t'          j        t*          |          5  |                     ddg           d d d            n# 1 swxY w Y   |                     d          j                            |                     d          j                  rJ 	 d d d            d S # 1 swxY w Y   d S )Nr   r   r   r   r   r   r   r   r=  r`   r"   r_   r>  r   T)df1adf2arL  FrG  z,all tables must have exactly the same nrows!r   rM  )r   r7   r8   r9   r:   r   r;   r   r   rB  r    r@   r   r   rH  r   r   r   rC  r   r   r   rD  rk   r   equals)rI   r   r   r&   r   s        rK   $test_append_to_multiple_dropna_falserO    s;   

	a  0099d6ll##r<<<  C
 ((**

FM

2
2C79vCHQ''c
3334	c
	#	#	#B		0$	7	7 
 
((3Z..VE 	) 	
 	
 	
 =]:S111 	? 	?,,ff-=>>>	? 	? 	? 	? 	? 	? 	? 	? 	? 	? 	? 	? 	? 	? 	? !''//5<<  ((.
 
 	
 	
 
 	

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
s7   2<F2.EF2E	F2E	A
F22F69F6c           
      `   t          t          j        dd          t          j        dd          t          j        dd          dz  d t          d          D             d t          d          D             d          }|j        dg         }t          t          t          t          |j                                                |_        | 	                    d	gd
dgddgd|j        dg         ddddd           | 
                    g d          }t          j        ||d           d S )Nr"      X   c                     g | ]}d S )r   rQ   rS   _s     rK   rU   z8test_append_to_multiple_min_itemsize.<locals>.<listcomp>  s    +++AC+++rM   r   c                     g | ]}d S )abcderQ   rT  s     rK   rU   z8test_append_to_multiple_min_itemsize.<locals>.<listcomp>  s    333A333rM   )IXNumBigNumStrLongStrr   rX  rY  rZ  r[  r\  )r   numsstrsr   r   rW   r   )r[  r\  rY  r   Tr3   )r   r7   rG   rg   r   r   r;   rm   r   rC  rD  r=   r>   )rI   r&   rr   rq   s       rK   $test_append_to_multiple_min_itemsizer_    sF   	)Ar""9Q##i2&&+++r+++33r333	
 	

 
B ws|H4c(.&9&9 : :;;<<HN$$VH%I&	
 	

 	C:: % 	 	 	 --.G.G.GHHF&(TBBBBBBrM   c                 J   t          dddt          j        d                    }|                                }t          j        |j        ddd d f<   d}|                     d	|d
                    t          j        t          |          5  |                     d	|d
                    d d d            n# 1 swxY w Y   |                     d|d         d           t          j        t          |          5  |                     d|d                    d d d            n# 1 swxY w Y   |                     d|d
         d           |                     d|d
                    | d         }t          |d
         |d
         g          }t          j        ||           d S )Nr   rZ   r   r   r1   r   r   z8NaN representation is too large for existing column sizesar_   r   sbr   bars)nan_repscn)r   r7   rG   r   r@   rA   r<   r   r   r   r   r=   ri   )rI   r&   df_nanr   rq   rr   s         rK   test_append_string_nan_reprh    s   	5))2	?	?	?BWWYYFFJqsAAAv
DC r#w'''	z	-	-	- 0 0T6#;///0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 r#w777	z	-	-	- 0 0T6#;///0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 r#w444vc{+++4 Fr#ws,--H68,,,,,s$   B77B;>B;;D$$D(+D()0r   r   r   numpyr7   r   pandas._libs.tslibsr   pandas.compatr   pandasr   r   r   r	   r
   r=   r   r   r   r   
single_cpu
pytestmarkimportorskipr   filterwarningsrL   rt   r   r   r   r   r   r   r   r	  r  r  parametrizer#  r(  r3  r;  rE  rJ  rO  r_  rh  rQ   rM   rK   <module>rr     s          				      ) ) ) ) ) )                            k$%
		X	&	& ?@@>T >T A@>TB-J -J -J`'L 'L 'LTTO TO TOn#G #G #GL.' .' .'bQ& Q& Q&h(O (O (OV: : :X, X, X,v, , ,&& & & ooo66& & 76&$; ; ;*P' P' P'f*& *& *&Z#, #, #,L  4
 
 
2C C C:- - - - -rM   