E g TddlmZddlmZmZddlZddlZddlZddl m cm Z ddl mZmZmZddlmZddlmZddlmZejdZejdZejd Zd Zejd d g dZejejde jddg dZ dZ!dZ"dZ#dZ$dZ%dZ&ej'(de)e*ddde*ddDgdZ+dZ,dZ-d Z.d!Z/d"Z0d#Z1d$Z2d%Z3d&Z4d'Z5d(Z6d)Z7d*Z8d+Z9d,Z:d-Z;d.Zd1Z?d2Z@d3ZAd4ZBd5ZCej'(d6gd7d8ZDd9ZEd:ZFd;ZGd<ZHe jIdd=ZJd>ZKd?ZLd@ZMdAZNdBZOdCZPej'(dDdEdFgdGZQdHZRdIZSdJZTdKZUdLZVdMZWdNZXdZYdOZZdPZ[dQZ\dRZ]ej'j^dSZ_dS)T) annotations)BytesIOStringION)NA DataFrameIndex) get_handle)read_xmlcJtgdgddtjdgdS)N)squarecircletriangle)hrshapedegreessides)rnpnanT/var/www/sysmax/venv/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 rHrIFindexrEr6rJrKrTrEexpectedrUrVrWrXs rtest_index_falserosP H4y000G  $ $"t5888 $   6VVXX__W--3355F 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6v&&!!!!!""""""""""""""""""s5)C:B CB CB CC C cd}t||}tjd5}||ddd|t |d5}|d }dddn #1swxYwYt|}||ksJ ddddS#1swxYwYdS) 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 rHrIFbooksbook)rl root_namerow_namerEr6rJrKrms r test_index_false_rename_row_rootrus` H4y000G  $ $ " 6&    $   6VVXX__W--3355F 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6v&&!!!!! " " " " " " " " " " " " " " " " " "s5+C :B C B C  B !C  C C  offset_index c,g|]}t|Sr)str).0is r r}.s*I*I*Ia3q66*I*I*Ircd}|}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 FrkN)copyrrlrNr5)rErvrrnoffset_geom_dfrXs r(test_index_false_with_offset_input_indexr-sd H(\\^^N ..N  " "v " > >F 6 " "F X      ra 0 square 360 4.0 1 circle 360 2 triangle 180 3.0 cj||}t|}|tksJdSr_rNr5 na_expectedrErrXs rtest_na_elem_outputrns9 ^^6^ * *F 6 " "F [ rcl|d|}t|}|tksJdS)Nna_reprErrs rtest_na_empty_str_elem_optionrus; ^^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.0rrNr5rErrnrXs rtest_na_empty_elem_optionr|sD 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> )rlrrrr4zhttp://example.xom)r namespacesprefixrErrs rtest_attrs_cols_prefixrsa H^^888/0 F 6 " "F X      rctjtd5|gd|ddddS#1swxYwYdS)Nno valid columnrbrdegreerrrdreKeyErrorrNrErs rtest_attrs_unknown_columnr x(9 ; ; ;NN!=!=!=fMMMNNNNNNNNNNNNNNNNNNAA Actjtd5|d|ddddS#1swxYwYdS)Nz!is not a valid type for attr_colsrb"shape", "degree", "sides"rrdre TypeErrorrNrs 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)rl elem_colsrErrErelems_cols_expectedrXs rtest_elems_cols_nan_outputrsY (^^<< 360 4.0 360 180 3.0 Frrr)rlrrrErrs 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.countsummeanTrlvaluesaggfuncmarginsrH) pivot_tableroundrNr5rEr/rnpvtrXs rtest_hierarchical_columnsrs HB   6"(((     eAhh ZZvZ & &F 6 " "F X      rcd}|ddgdgddd}|t|jj| }t|}||ksJdS) Na r-r,r.rTrrr)rrrNlist reset_indexcolumnsrr5rs rtest_hierarchical_attrs_columnsrFs  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)groupbyaggrrNr5rEr/rnrrXs rtest_multi_indexrds H6 :v.//7  % % % & & q ZZvZ & &F 6 " "F X      rcBd}|ddgdgdd}|t |jj|}t|}||ksJdS)Na$ r-r,r.rrr) rrrrNrrrrr5rs 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 namespacesrbrrr4rrrs rtest_missing_prefix_in_nmsprs x(K M M M  01%                      sAA 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)rEencodingrrEz* 0 square 360 4.0 1 circle 360 2 triangle 180 3.0 F)xml_declaration)rdrrN)rrnrXs r!test_xml_declaration_pretty_printr~sF  H,^^E^ 2 2F X      rcd}|d|}t|}||dd}||ksJdS)Na4 0square3604.01circle3602triangle1803.0F) pretty_printrE />/>)rNr5r3rs rtest_no_pretty_print_with_declrsb  ^^v^ > >F 6 " "Ft,, X      rczd}|dd|}||dd}||ksJdS)Na 0square3604.01circle3602triangle1803.0F)rrrErr)rNr3rs rtest_no_pretty_print_no_declrsW  ^^Ef^ U UFt,, X      rctjtd5|ddddS#1swxYwYdS)Nz7lxml not found, please install or use the etree parser.rb)rdre ImportErrorrNrs rtest_default_parser_no_lxmlrs U    s>AActjtd5|dddddS#1swxYwYdS)Nz,Values for parser can only be lxml or etree.rbbs4rHrdre ValueErrorrNrs rtest_unknown_parserrs I   %% e$$$%%%%%%%%%%%%%%%%%%sAAAa% 0 square 360 4.0 1 circle 360 2 triangle 180 3.0 ctjdt|||dkrdnd5}||tksJ ddddS#1swxYwYdSNrAr7rJr stylesheet)rdrrOrN xsl_expected)xsl_row_field_outputr?rrWs rtest_stylesheet_file_likers  d WW   < ~~~++|;;;;;<<<<<<<<<<<<<<<<< >C !2   '' #&&& ''''''''''''''''''sA55A9<A9valrrctjd}dgd}tj|j|5||ddddS#1swxYwYdS)Nr|)zDocument is emptyrzNone \(line 0\)rbr )rdrrrerrN)r!rrmsgs rtest_empty_string_stylesheetr%"s$\22J ((     C z0 < < <''#&&&''''''''''''''''''sA++A/2A/ctjd}d}tj|jd5||ddddS#1swxYwYdS)NraR zOpening and ending tag mismatchrbr )rdrrerrNrs rtest_incorrect_xsl_syntaxr'3s$\22J C* !*K   '' #&&&''''''''''''''''''AAActjd}d}tj|jd5||ddddS#1swxYwYdS)NraT zfailed to compilerbr )rdrreXSLTParseErrorrNrs rtest_incorrect_xsl_evalr+Qs$\22J C* z09L N N N''#&&&''''''''''''''''''r(c tjd}d}tj|jd5t jd5}|||dddn #1swxYwYddddS#1swxYwYdS)Nrag zCannot resolve URIrbrIr )rdrreXSLTApplyErrorrLrMrN)rrrrVs rtest_incorrect_xsl_applyr.ms$\22J C z09M O O O11 _Z ( ( 1D NN4CN 0 0 0 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1111111111111111111s5BA+ B+A/ /B2A/ 3BB Bcd}tjtd5|d|ddddS#1swxYwYdS)NaI z*To use stylesheet, you need lxml installedrbrC)rEr r)rrs rtest_stylesheet_with_etreer0s C G   77 g#666777777777777777777rictjdd}|d}||}||}||ksJdS)NrAaQ , ,shape,degrees,sides  )lineterminatorr )rdrto_csvrRrN)rrout_csvout_xmls rtest_style_to_csvr7sl  C$nnDn11G--//nnn,,G g      rctjdd}|}|d|}||ksJdS)NrAa shape degrees sides NaN)rr )rdr to_stringrN)rrout_strr6s rtest_style_to_stringr<sY  C4!!GnnEcn::G g      rctjdd}|}||}||ksJdS)NrAa] " {"shape":{ },"degrees":{ },"sides":{ }} , r )rdrto_jsonrN)rrout_jsonr6s rtest_style_to_jsonr@sV # CJ  Hnnn,,G w      rcntj5}||||t|d|5}|j}dddn #1swxYwYdddn #1swxYwYt |}t|ksJdS)NrE compressionr7rC rLrMrNr handlerPr5geom_xmlrR)rEcompression_onlyrrV handle_objrXs rtest_compression_outputrJs0   .dtF8HIII   (    .&++--F  . . . . . . . . . . . . . . ................6 " "F v||~~ % % % % % %s4+A<A% A<%A) )A<,A) -A<<BBcd||z}tj|5}||||t|d|5}|j}dddn #1swxYwYdddn #1swxYwYt |}t|ksJdS)Nzxml.)filenamerBr7rDrE)rErHrcompression_to_extensioncompfilerVrIrXs rtest_filename_and_suffix_comprO*sH01ABBH ( + + +.ttF8HIII   (    .&++--F  . . . . . . . . . . . . . . ................6 " "F v||~~ % % % % % %s5+B  A2& B 2A6 6B 9A6 :B  B B cd}tdtgi|}||}t ||ksJdS)Nzd 0 arH)rrastyperNr5rR)any_numeric_ea_dtyperErndfresults rtest_ea_dtypesrV=sk H C";   & &'; < rzs""""""  ))))))))) ''''''""""""J,,,^   >c{###$#  V=2=3H3HIII7STTTUT , , , , , ,(((,,,$"$"$"N&"&"&"RTT%%B--((*I*I55R==*I*I*IJ!!!L 0!!!!!!B   .NNN NNN))):NNN NNN )))@,,,^<$$$N4<H@   H! H ' ' 'PPP%F%F%FGGLLHGL888:*,f%%% 0<<<"""& " " " ' ' 'S ** ' '+* ' '''<'''8111(777$:D***` 0 & & & &&&& 5 5 5BBB       r