j@4 ddlZddlZddlZddlmZddlZddlmZm Z m Z m Z m Z m ZmZmZddlmZej(j*gZdZdZdZdZd Zd Zd Zej<d Zd Z dZ!dZ"dZ#dZ$ej(jKdddgej(jKde e&dDcgc] }e'| c}e ejPde)e ejPdeddejTddgdZ+dZ,dZ-dZ.dZ/ej(jKdddgdZ0ycc}w) N) Timestamp) DataFrameHDFStoreIndex MultiIndexSeries_testingconcat date_range)_test_decoratorsctdddgi}|jd|d|jd|d|jdjdk(sJ|jdjdk(sJy) NAafixedformatbtable)rput get_storer format_type) temp_hdfstoredfs b/root/aria/tools/markitdown-venv/lib/python3.12/site-packages/pandas/tests/io/pytables/test_put.pytest_format_typers| C!Q= !Bc2g.c2g.  # #C ( 4 4 ?? ?  # #C ( 4 4 ?? ?cd}tjt|5t|ddddy#1swYyxYw)Nz-format is not a defined argument for HDFStorematchrr)pytestraises ValueErrorr) temp_h5_pathmsgs r test_format_kwarg_in_constructorr'!s6 :C z -/g.///s5>c Ptdtjdjdzt t dt t dDcgc]}d| c}}tjdd 5|jd ||jd jrJd }tjt| 5|jd |ddddddtjdd5|j!d |jd ||jd jsJ|jd ||jd jsJ dddycc}w#1swYxYw#1swYxYw#1swYyxYw)N皙?xABCDr,i-columnsindexio.hdf.default_formatrrCan only append to Tablesr df2r)rnparangereshaperlistrangepdoption_contextrris_tabler"r#r$appendremove)rirr&s rtest_api_default_formatrA*sn  biin$$W--d6l#uRy1!r!X12 B  2G <,$# ++D1::::) ]]:S 1 ,   + , ,  2G <7T"$#''-6666UB'''-6666 772 , , ,,77s8 E??A F FFA0FF FFF%c tdtjdjdzt t dt t dDcgc]}d| c}}tjdd 5|j|d t|5}|jd jrJ dddd }tjt| 5|j|ddddddddtjdd5|j|d t|5}|jdjsJ ddd|j|ddt|5}|jdjsJ ddddddycc}w#1swYxYw#1swYxYw#1swYxYw#1swYxYw#1swYGxYw#1swYyxYw)Nr)r*r+r.r,r/r0r3rr)keyr4r r5T)rCr>rdf3df4)rr6r7r8rr9r:r;r<to_hdfrrr=r"r#r$)r%r@rstorer&s rtest_api_default_format_pathrHCs  biin$$W--d6l#uRy1!r!X12 B  2G << ,D ) l # 7u''-66 666 7) ]]:S 1 < IIldI ; < <  2G <4 ,E * l # 4u##E*33 33 4 ,E$ 7 l # 4u##E*33 33 4 442  7 7 < < << 4 4 4 4 44s~ F2?GF7=%G"G7GG3<G'G3G'!G37G <GG GGG$ G3'G0 ,G33G<c |}ttjdtjt dd}t tj jdjdttdt d d d  }||d<|dd|d<|dd|d<|dd|d<|dd|d<|jd|dddd}tjt|5|jd|ddddddtjt|5|jd|ddddddtjt|5|jd|dddddd|jd|ddddtj |dd|dy#1swYxYw#1swYxYw#1swYVxYw)N dtype 2020-01-01periodsr2r)r-r. 2000-01-01rQBrOfreqr0rrz foo/bar/bahfooz/foocrrr4r T)r>fF)rr>)rr6r7float64r rrandom default_rngstandard_normalrr9rr"r#r$tmassert_frame_equal)rrGtsrr&s rtest_putr`[s E  "BJJ'z,PR/S B  a 009d6l#r< B E#JCRE#Jcr7E-cr7E%LsGE&M IIc2cr77I+ &C z -- #r"#wt ,- z -- #r"#wt ,- z -- #r"#wt ,- IIc2cr775I9"Sb'5:.-- ----s$/F:*G%G:GGGc`|}ttdDcgc]}d| c}}ttjd|}t ||d}||d<t j|d|||d<t j|d|tdgtdDcgc]}d| c}z}ttjd|}t ||d}||d<t j|d|||d<t j|d|ycc}wcc}w) NrQzI am a very long string index: rP)rrSrr$abcdefghijklmnopqrstuvwxyz1234567890) rr:rr6r7rr]assert_series_equalr^)rrGr@r2srs rtest_put_string_indexrfs" E %)LQ4QC8L MEryy}E*A # $BE#J5:q)E#J%*b)  /0:?) DQ ,QC 0 D E E ryy}E*A # $BE#J5:q)E#J%*b)+M Es D&& D+ c ttjjdj dt t dtddd}|jd |d d tj|d |d }tjt|5|jd|dd dddy#1swYyxYw)NrrJr-r.rRrJrSrTr0rWrzlibrcomplib0Compression not supported on Fixed format storesr rr)rr6rZr[r\rr9r rr]r^r"r#r$rrr&s rtest_put_compressionrns  a 009d6l#r< B c2gv>-,b1 =C z -C#r'6BCCCs $CC c ttjjdj dt t dtddd}d }tjt| 5|jd |d d ddd|jd|dd tj|d|y#1swY8xYw)Nrrhr.rRrJrSrTr0rlr rrbloscrjrWr)rr6rZr[r\rr9r r"r#r$rr]r^rms rtest_put_compression_bloscrqs  a 009d6l#r< B =C z -D#r'7CDc2gw?-,b1 DDs 6CC ctdtdjdgz}|jd||j }|j d}t j||y)N20010102nsser)rras_unitrcopygetr]rd)rperformance_warningusing_infer_stringrvexpectedresults rtest_put_datetime_serr~s_ i +33D9:: ;CeS!xxzH   u %F68,rc Fttjjdj dt t dtddd}d |d <d |d <|d dkD|d<|ddkD|d<d|d<d|d<d|d<tdjd|d<tdjd|d<tdjd|d<tdjd|d<tj|j|jddd gf<|j}|rdn|}tj|5|j!d|ddd|j#d}tj$||y#1swY1xYw) Nrrhr.rRrJrSrTr0rVobj1barobj2rrbool1bool2Tbool3rint1int2rtru timestamp120010103 timestamp2 datetime1 datetime2rsr)rr6rZr[r\rr9r rrwnanlocr2 _consolidater]assert_produces_warningrryr^)rrzr{rwarningr|s rtest_put_mixed_typers  a 009d6l#r< B BvJBvJS'A+BwKS'A+BwKBwKBvJBvJ ,44T:B| ,44T:B| +33D9B{O +33D9B{O&(ffBFF288Aa=6( "#  B(d.AG # #G ,$$#$  &H(B' $$s FF cbtj|}tdtjdtjdg|i}|j d||j tjurdnd}|j|}|jd}tj||y)NrxyrKrstrstring) r; StringDtyperarrayNArna_valuer6rastyperyr]r^)rrzstring_dtype_argumentsrLrexpected_dtyper|r}s rtest_put_str_framers NN2 3E C3s"35AB CBdB#nn6UHNyy(H   t $F&(+rc8tj|}tdtjdg|}|j d||j t jurdnd}|j|}|jd}tj||y)NrrrKrvrr) r;rrrrrr6rrryr]rd)rrzrrLrvrr|r}s rtest_put_str_seriesrs| NN2 3E #ruuc"% 0CeS!#nn6UHNzz.)H   u %F68,rctj|}ttjdtjdg|gdd}|j d|d|j tjurd nd }|j}|d j||d <|jd}tj||y) NrrrK)rrrs)rrrr) complevelrrr)r;rrrrrrr6rrxrryr]r^)rrrLrrr|r}s rtest_put_str_frame_complevelrs NN2 3E 3s"35A R SBdB!,#nn6UHNwwyHSM((8HSM   t $F&(+rrrrr2rJrKrMrNcttjjdj dt d|}|j d||tj||dy)Nr)rJrABr0rr) rr6rZr[r\r9rr]r^)rrr2rs rtest_store_index_typesrsa  a 009T  B dBv."mD12rctjgdddg}ttjdj dd|}|j |jj}|jd |tj|d |d d |jd |d tj|d |d d tjd}tjt |5|jd|d dgdddtjd}tjt |5|jd|d d dddy#1swYXxYw#1swYyxYw)N)rr)rr)rSr)rSrfirstsecondnames rsr-r1rTcheck_index_typecheck_column_typedf1rrzB{{288,,./HdBdXPTeR0ehQU ))R SC z -I%G3%HI ))Q RC z -H%G$GHHIIHHsE1E=1E:=Fc*tjgdddg}ttjdj dd|}|j d ||j d |tj|d t||fy) Nrrrrrrsr-rr5) rrrr6r7r8r>r]r^r )rr{r2rs rtest_column_multiindex_existingr5s  " "8(@S E 299R=((A. >B##-.Bx0@Arc>ttjdjddt t dd}|j |jj}|jd|d tj|d|d d y) Nrrsr-r.rV)namerrrrTr) rr6r7r8rr9rr2rrr]r^)rr{rr|s r%test_column_multiindex_non_index_axesrBs| 299R=((A.d6lQV8W XB{{288,,./HeR0ehQUrcdd}ttjdddg|}|jd|t j |j d||jdttjdddg|gd}|jd|t j |j d|ttjd|gd }|jd |ttjd|gd  }t j|j d ||jdttjdddg|gd }d }tjt|5|jd|ddd|jdttjdddg|gd}tjt|5|jd|ddd|jdttjdddg|gd}|jd|t j |j d|y#1swYxYw#1swYxYw)Ncvtdd}tj|tdtdg|}|S)Nz 2013-12-01z 2013-12-02rrsr)r r from_productr:)rdtimis r make_indexz)test_store_multiindex..make_indexPs4|4  $ $c58U1X%>e L r)rrrrr0r)dateNNrrPrv)rlevel_1level_2)rrtzBduplicate names/columns in the multi-index when storing as a tabler )rrr)rrer)N) rr6zerosr>r]r^selectr?rrdr"r#r$)rrrrvxpr&s rtest_store_multiindexrLsT 288G$sCj MBr"-..t4b9  c -. B r"-..t4b9 "Z0D%E FC$  J/M$N OB=//6;  c +, B OC z -'T2&'  c 12 B z -'T2&'  c +, B r"-..t4b9+''''s7J8)K8KK c$ttjjdj dt j ddd}|j|dd | t j|d}tj||y) Nr)r20220101Mr)rUrOrPrw)rCmoder) rr6rZr[r\r; period_rangerFread_hdfr]r^)r%rrr|s rtest_store_periodindexrsq  a 008oojsA> B IIl3vI>{{<.H"h'r)1rnumpyr6r"pandas._libs.tslibsrpandasr;rrrrrr r]r r pandas.utilr tdmark single_cpu pytestmarkrr'rArHr`rfrnskip_if_windowsrqr~rrrr parametrizer:rr7floatrrrrrrr)r@s0rrs  )   /kk$$ % @/7240!/H*4 C  2 2-(8 , - ,GW#56  uRy)!s1v)* ibii%() ibiim<, b1   3 7 3H8 B;:|GW#56 (7 (K*s=E