Pi7*ddlmZddlmZmZddlZddlZddlZddl m cm Z ddl mZmZmZddlmZddlmZddlmZejdZejdZejd Zd Zejd d g dZejejde jddg dZ dZ!dZ"dZ#dZ$dZ%dZ&ej'(de)e*gdZ+dZ,dZ-dZ.dZ/dZ0d Z1d!Z2d"Z3d#Z4d$Z5d%Z6d&Z7d'Z8d(Z9d)Z:d*Z;d+Zd.Z?d/Z@d0ZAd1ZBd2ZCej'(d3gd4d5ZDd6ZEd7ZFd8ZGd9ZHe jIdd:ZJd;ZKdZNd?ZOd@ZPej'(dAedBedCgdDZQdEZRdFZSdGZTdHZUdIZVdJZWdKZXdZYdLZZdMZ[dNZ\dOZ]ej'j^dPZ_dS)Q) annotations)BytesIOStringION)NA DataFrameIndex) get_handle)read_xmlcJtgdgddtjdgdS)N)squarecircletriangle)hrshapedegreessides)rnpnans/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/pandas/tests/io/xml/test_to_xml.pygeom_dfr;s9 555&^    rc:tgdgdgdgddS)N)MercuryVenusEarthMarsJupiterSaturnUranusNeptune) terrestrialr&r&r& gas giantr' ice giantr()innerr)r)r)outerr*r*r*)g`|x ?gc AJx@g2r@gl?gh|@g @gd]FU@g hY@)planettypelocationmass)rrrr planet_dfr/FsZ             ?) ) + + +rcdS)Nac 0 cooking Everyday Italian Giada De Laurentiis 2005 30.0 1 children Harry Potter J K. Rowling 2005 29.99 2 web Learning XML Erik T. Ray 2003 39.95 rrrrfrom_file_expectedr1vs   rc6||dd}|S)Nz% cooking Everyday Italian Giada De Laurentiis 2005 30.0 children Harry Potter J K. Rowling 2005 29.99 web Learning XML Erik T. Ray 2003 39.95 rHFindexrErIrJrLrQrErRexpectedrSrTs rtest_index_falseris{ H4y000G NN9E&N999  ' 2 2 8 8 : :F 6 " "F X      rcd}t||}||ddd||d}t |}||ksJdS) Na, cooking Everyday Italian Giada De Laurentiis 2005 30.0 children Harry Potter J K. Rowling 2005 29.99 web Learning XML Erik T. Ray 2003 39.95 rHFbooksbook)rf root_namerow_namerErIrJrLrgs r test_index_false_rename_row_rootros H4y000G NN'F6 ' 2 2 8 8 : :F 6 " "F X      rtypcd}fdtddD}|}t||_|d|}t |}||ksJdS)z Tests that the output does not contain the `` field when the index of the input Dataframe has an offset. This is a regression test for issue #42458. aD square 360 4.0 circle 360 triangle 180 3.0 c&g|] }|Srr).0irps r z.As!222qCCFF222r FreN)rangecopyrrfrMr5)rErprrh offset_indexoffset_geom_dfrTs ` r(test_index_false_with_offset_input_indexr|%s H&3222E"bMM222L\\^^N ..N  " "v " > >F 6 " "F X      ra 0 square 360 4.0 1 circle 360 2 triangle 180 3.0 cj||}t|}|tksJdSrZrMr5 na_expectedrErrTs rtest_na_elem_outputrds9 ^^6^ * *F 6 " "F [ rcl|d|}t|}|tksJdS)Nna_reprEr~rs rtest_na_empty_str_elem_optionrks; ^^2f^ 5 5F 6 " "F [ rcfd}|d|}t|}||ksJdS)Na 0 square 360 4.0 1 circle 360 0.0 2 triangle 180 3.0 z0.0rrMr5rErrhrTs rtest_na_empty_elem_optionrrsD H.^^5^ 8 8F 6 " "F X      rcjd}|gd|}t|}||ksJdS)Nz r attr_colsrErrs rtest_attrs_cols_nan_outputrsK H^^&C&C&CF^ S SF 6 " "F X      rcrd}|gdddid|}t|}||ksJdS)Na> )rfrrrr4zhttp://example.xom)r namespacesprefixrErrs rtest_attrs_cols_prefixrsa H^^888/0 F 6 " "F X      rctjtd5|gd|ddddS#1swxYwYdS)Nno valid columnr]rdegreerrr_r`KeyErrorrMrErs rtest_attrs_unknown_columnr x(9 ; ; ;NN!=!=!=fMMMNNNNNNNNNNNNNNNNNNAA Actjtd5|d|ddddS#1swxYwYdS)Nz!is not a valid type for attr_colsr]"shape", "degree", "sides"rr_r` TypeErrorrMrs rtest_attrs_wrong_typer y)L N N NNN!=fMMMNNNNNNNNNNNNNNNNNNAAAcld}|dgd|}t|}||ksJdS)NaD 360 4.0 square 360 circle 180 3.0 triangle F)rrr)rf elem_colsrErrErelems_cols_expectedrTs rtest_elems_cols_nan_outputrsY (^^<< 360 4.0 360 180 3.0 Frrr)rfrrrErrs rtest_elems_and_attrs_colsrs^ "^^g&) F 6 " "F ( ( ( ( ( ( (rcd}|ddgdgddd}|| }t|}||ksJdS) Na inner terrestrial 4 11.81 2.95 outer gas giant 2 2466.5 1233.25 outer ice giant 2 189.23 94.61 All 8 2667.54 333.44 r-r,r.countsummeanTrfvaluesaggfuncmarginsrH) pivot_tableroundrMr5rEr/rhpvtrTs rtest_hierarchical_columnsr s HB   6"(((     eAhh ZZvZ & &F 6 " "F X      rcd}|ddgdgddd}|t|jj| }t|}||ksJdS) Na r-r,r.rTrrr)rrrMlist reset_indexcolumnsrr5rs rtest_hierarchical_attrs_columnsr<s  H   6"(((     eAhh ZZ$s'8'8'@'G"H"HQWZ X XF 6 " "F X      rcd}|ddgdgdd}||}t |}||ksJdS)Na inner terrestrial 4 11.81 2.95 outer gas giant 2 2466.5 1233.25 outer ice giant 2 189.23 94.61 r-r,r.rrrH)groupbyaggrrMr5rEr/rhrrTs rtest_multi_indexrZs H6 :v.//7  % % % & & q ZZvZ & &F 6 " "F X      rcBd}|ddgdgdd}|t |jj|}t|}||ksJdS)Na$ r-r,r.rrr) rrrrMrrrrr5rs rtest_multi_index_attrs_colsrs  H :v.//7  % % % & & q ZZ$s'8'8'@'G"H"HQWZ X XF 6 " "F X      rcjd}|ddi|}t|}||ksJdS)Na 0 square 360 4.0 1 circle 360 2 triangle 180 3.0 rhttp://example.comrrErrs rtest_default_namespacersJ H.^^,@'A&^ Q QF 6 " "F X      rcld}|ddd|}t|}||ksJdS)Na 0 square 360 4.0 1 circle 360 2 triangle 180 3.0 http://other.orgr)othexrrrs rtest_unused_namespacesrsX H.^^-5IJJF6 " "F X      rcld}|ddid|}t|}||ksJdS)Na 0 square 360 4.0 1 circle 360 2 triangle 180 3.0 r4rrrrErrs rtest_namespace_prefixrsVH.^^/0vF6 " "F X      rctjtd5|ddid|ddddS#1swxYwYdS)Nz!doc is not included in namespacesr]rrr4rrrs rtest_missing_prefix_in_nmsprs x(K M M M  01%                      AA Acnd}|dddd|}t|}||ksJdS)Na7 0 square 360 4.0 1 circle 360 2 triangle 180 3.0 rr)rr4r4rrrs r!test_namespace_prefix_and_defaultrs[H.^^,5GHHF 6 " "F X      ru 0 1 José Sofía 1 2 Luis Valentina 2 3 Carlos Isabella 3 4 Juan Camila 4 5 Jorge Valeria ct||dd}|d|}||dd}|tksJdS)N ISO-8859-1)rErKrKrEz* 0 square 360 4.0 1 circle 360 2 triangle 180 3.0 F)xml_declaration)r_rrM)rrhrTs r!test_xml_declaration_pretty_printrrsF  H,^^E^ 2 2F X      rcd}|d|}t|}||dd}||ksJdS)Na4 0square3604.01circle3602triangle1803.0F) pretty_printrE />/>)rMr5r3rs rtest_no_pretty_print_with_declrsb  ^^v^ > >F 6 " "Ft,, X      rczd}|dd|}||dd}||ksJdS)Na 0square3604.01circle3602triangle1803.0F)rrrErr)rMr3rs rtest_no_pretty_print_no_declrsW  ^^Ef^ U UFt,, X      rctjtd5|ddddS#1swxYwYdS)Nz7lxml not found, please install or use the etree parser.r])r_r` ImportErrorrMrs rtest_default_parser_no_lxmlrs U    s>AActjtd5|dddddS#1swxYwYdS)Nz,Values for parser can only be lxml or etree.r]bs4rHr_r` ValueErrorrMrs rtest_unknown_parserrs I   %% e$$$%%%%%%%%%%%%%%%%%%sAAAa% 0 square 360 4.0 1 circle 360 2 triangle 180 3.0 ctjdt|||dkrdnd5}||tksJ ddddS#1swxYwYdSNrAr7rIrJ stylesheetr_ropenrM xsl_expected)xsl_row_field_outputr?rfs rtest_stylesheet_file_liker s  d WW   < ~~~++|;;;;;<<<<<<<<<<<<<<<<< zOpening and ending tag mismatchr]r)r_rr`rrMrrr rs rtest_incorrect_xsl_syntaxr%$s$\22J C* !)J   11 (3--000111111111111111111$A##A'*A'ctjd}d}tj|jd5|t |ddddS#1swxYwYdS)NraT zfailed to compiler]r)r_rr`XSLTParseErrorrMrr$s rtest_incorrect_xsl_evalr)Bs$\22J C* z08K L L L11(3--000111111111111111111r&ctjd}d}tj|jd5||t |ddddS#1swxYwYdS)Nrag zCannot resolve URIr]r)r_rr`XSLTApplyErrorrMr)rrRr rs rtest_incorrect_xsl_applyr,^s$\22J C z08L M M M<<yXc]];;;<<<<<<<<<<<<<<<<< z*To use stylesheet, you need lxml installedr]rC)rEr)r_r`rrMrrs rtest_stylesheet_with_etreer.qs C z)U V V VAAg(3--@@@AAAAAAAAAAAAAAAAAAs%AAActjdd}|d}||}|t |}||ksJdS)NrAaQ , ,shape,degrees,sides  )lineterminatorr)r_rto_csvrOrMr)rrout_csvout_xmls rtest_style_to_csvr5sr  C$nnDn11G--//nn n66G g      rctjdd}|}|dt |}||ksJdS)NrAa shape degrees sides NaN)rr)r_r to_stringrMr)rrout_strr4s rtest_style_to_stringr:s_  C4!!GnnEhsmmnDDG g      rctjdd}|}|t |}||ksJdS)NrAa] " {"shape":{ },"degrees":{ },"sides":{ }} , r)r_rto_jsonrMr)rrout_jsonr4s rtest_style_to_jsonr>s\ # CJ  Hnn n66G w      rc|}||||t|d|5}|j}dddn #1swxYwYt |}t |ksJdS)NrE compressionr7rArMr handler r5geom_xmlrO)rEcompression_onlyrrRrb handle_objrTs rtest_compression_outputrHs D NN44DNEEE   $   * "'')) ***************6 " "F v||~~ % % % % % %sAAAc8d||z}||z }||||t|d|5}|j}dddn #1swxYwYt |}t |ksJdS)Nzxml.r@r7rBrC) rErFrcompression_to_extensiontmp_pathcompfilerbrGrTs rtest_filename_and_suffix_comprMs01ABBH h D NN44DNEEE   $   * "'')) ***************6 " "F v||~~ % % % % % %sA!!A%(A%cd}tdtgi|}||}t ||ksJdS)Nzd 0 arH)rrastyperMr5rO)any_numeric_ea_dtyperErhdfresults rtest_ea_dtypesrT+sk H C";   & &'; < >tF===>>>>>>>>>>>>>>>>>>rcjtjd}tjdtjtt f5|d}||j| d|jdd|ddddS#1swxYwYdS) Ns3fsrAT)anonzs3://z /geom.xmlzip)rArE) r_rtmexternal_error_raisedPermissionErrorr S3FileSystemlsnamerM)rEs3_bucket_publicrrYfss rtest_s3_permission_outputrdBs  v & &D  !?4E"F G G    D  ) ) #$$$ 4$) 4 4 4%PV                      s AB((B,/B,)` __future__riorrrnumpyrr_pandas.util._test_decoratorsutil_test_decoratorstdpandasrrrpandas._testing_testingr\pandas.io.commonr pandas.io.xmlr fixturerr/r1r5r?r< skip_if_norErUrXr[rcriromark parametrizeintstrr|rrrrrrrrrrrrrrrrrrrrrrrrrrrrrskip_if_installedrrrr rrrr"r%r)r,r.r5r:r>rErHrMrTrW single_cpurdrrrrys""""""  ))))))))) ''''''""""""J,,,^   >c{###$#  V=2=3H3HIII7STTTUT(((((((((,,,"""J$$$Nc ++!!,+!L 0!!!!!!B   .NNN NNN))):NNN NNN )))@,,,^<$$$N4<H@   H! H ' ' 'QQQ%F%F%FGGMMHGM888:*,f%%% 0<<<"""&"""'''"wws|| <== ' '>= ' 111<1118<<<& A A A :D***` 0 & & & &&&& 5 5 5>>>       r