o h@sddlmZddlmZmZddlZddlZddlZddl m m Z ddl mZmZmZddlmZddlmZddlmZejddZejd d Zejd d Zd dZejddgdddZejejde dddgdddZ ddZ!ddZ"ddZ#dd Z$d!d"Z%d#d$Z&ej'(d%e)e*d&d'd(d)e*d&d'Dgd*d+Z+d,Z,d-d.Z-d/d0Z.d1d2Z/d3d4Z0d5d6Z1d7d8Z2d9d:Z3d;d<Z4d=d>Z5d?d@Z6dAdBZ7dCdDZ8dEdFZ9dGdHZ:dIdJZ;dKdLZdQdRZ?dSdTZ@dUZAdVdWZBdXdYZCej'(dZgd[d\d]ZDd^d_ZEd`daZFdbdcZGdddeZHe IddfdgZJdhdiZKdjZLdkdlZMdmdnZNdodpZOdqdrZPej'(dsdtdugdvdwZQdxdyZRdzd{ZSd|d}ZTd~dZUddZVddZWddZXd,ZYddZZddZ[ddZ\ddZ]ej'j^ddZ_dS)) annotations)BytesIOStringION)NA DataFrameIndex) get_handle)read_xmlcCs tgdgddtjdgdS)N)squarecircletriangle)hr shapedegreessides)rnpnanrrQ/var/www/html/env/lib/python3.10/site-packages/pandas/tests/io/xml/test_to_xml.pygeom_df;s  rcCs"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)rrrrr planet_dfFs   r+cCsdS)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 rrrrrfrom_file_expectedvsr,cCs|dur |dd}|S)Nz%c Ct||d}td8}|j||dt|d}|d}Wdn1s,wYt|}||ks;JWddS1sFwYdSNr>test.xmlr0utf-8 r tm ensure_cleanto_xmlopenreaddecodestripr/ xml_booksr>r,df_filepathfoutputrrrtest_file_output_str_read   "rRc Cr?r@rDrLrrrtest_file_output_bytes_readrSrTcCs0t||d}|j|d}t|}||ksJdSNrA)r rGr/)rMr>r,rNrQrrrtest_str_outputs  rVcCsFd}tjtdd|j||dWddS1swYdS)Nz/my/fake/path/output.xmlz6Cannot save file into a non-existent directory: .*pathmatchrA)pytestraisesOSErrorrGr>rrOrrrtest_wrong_file_paths"r]c Csd}t||d}td9}|j|d|dt|d}|d}Wdn1s/wYt|}||ks>JWddS1sIwYdS)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 rArBFindexr>r0rCrDrMr>expectedrNrOrPrQrrrtest_index_falses   "rbc Csd}t||d}td;}|j|ddd|dt|d}|d }Wdn1s1wYt|}||ks@JWddS1sKwYdS) 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 rArBFbooksbook)r_ root_namerow_namer>r0rCrDr`rrr test_index_false_rename_row_roots    "rg offset_index cCsg|]}t|qSr)str).0irrr .srncCs<d}|}t||_|jd|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 Fr^N)copyrr_rGr/)r>rhrraoffset_geom_dfrQrrr(test_index_false_with_offset_input_index-s   rqa 0 square 360 4.0 1 circle 360 2 triangle 180 3.0 cCs$|j|d}t|}|tksJdSrUrGr/ na_expectedr>rrQrrrtest_na_elem_outputns rucCs&|jd|d}t|}|tksJdS)Nna_repr>rrrtrrrtest_na_empty_str_elem_optionusrycCs*d}|jd|d}t|}||ksJdS)Na 0 square 360 4.0 1 circle 360 0.0 2 triangle 180 3.0 z0.0rwrGr/r>rrarQrrrtest_na_empty_elem_option|sr|cCs.d}|jgd|d}t|}||ksJdS)Nz r attr_colsr>rzr{rrrtest_attrs_cols_nan_outputsrcCs6d}|jgdddid|d}t|}||ksJdS)Na> )r_rrrr.zhttp://example.xom)r~ namespacesprefixr>rzr{rrrtest_attrs_cols_prefixs rcCFtjtdd|jgd|dWddS1swYdS)Nno valid columnrWrdegreerr}rYrZKeyErrorrGr>rrrrtest_attrs_unknown_column"rcCBtjtdd|jd|dWddS1swYdS)Nz!is not a valid type for attr_colsrW"shape", "degree", "sides"r}rYrZ TypeErrorrGrrrrtest_attrs_wrong_type"rcCs0d}|jdgd|d}t|}||ksJdS)NaD 360 4.0 square 360 circle 180 3.0 triangle F)rrr)r_ elem_colsr>rzr>relems_cols_expectedrQrrrtest_elems_cols_nan_outputs  rcCr)NrrWrrr>rrrrrtest_elems_unknown_columnrrcCr)Nz!is not a valid type for elem_colsrWrrrrrrrtest_elems_wrong_typerrcCs4d}|jdddgdg|d}t|}||ksJdS)Na# 360 4.0 360 180 3.0 Frrr)r_rr~r>rzrrrrtest_elems_and_attrs_colssrcCsHd}|jddgdgdddd}|j|d }t|}||ks"JdS) 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*countsummeanTr_valuesaggfuncmarginsrA) pivot_tableroundrGr/r>r+rapvtrQrrrtest_hierarchical_columnss! rcCsVd}|jddgdgdddd}|jt|jj|d }t|}||ks)JdS) Na r)r(r*rTrrr})rrrGlist reset_indexcolumnsrr/rrrrtest_hierarchical_attrs_columnsFs rcCsJd}|ddgdgdd}|j|d}t|}||ks#JdS)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*rrrA)groupbyaggrrGr/r>r+rarrQrrrtest_multi_indexds  rcCsXd}|ddgdgdd}|jt|jj|d}t|}||ks*JdS)Na$ r)r(r*rrr}) rrrrGrrrrr/rrrrtest_multi_index_attrs_colss rcCs.d}|jddi|d}t|}||ksJdS)Na 0 square 360 4.0 1 circle 360 2 triangle 180 3.0 rvhttp://example.comrr>rzr{rrrtest_default_namespacesrcCs0d}|jddd|d}t|}||ksJdS)Na 0 square 360 4.0 1 circle 360 2 triangle 180 3.0 http://other.orgr)othexrrzr{rrrtest_unused_namespacessrcCs0d}|jddid|d}t|}||ksJdS)Na 0 square 360 4.0 1 circle 360 2 triangle 180 3.0 r.rrrr>rzr{rrrtest_namespace_prefixs  rcCsHtjtdd|jddid|dWddS1swYdS)Nz!doc is not included in namespacesrWrvrr.rrrrrrtest_missing_prefix_in_nmsps  "rcCs2d}|jdddd|d}t|}||ksJdS)Na7 0 square 360 4.0 1 circle 360 2 triangle 180 3.0 rr)rvr.r.rrzr{rrr!test_namespace_prefix_and_defaultsru 0 1 José Sofía 1 2 Luis Valentina 2 3 Carlos Isabella 3 4 Juan Camila 4 5 Jorge Valeria cCsFt||ddd}|jd|d}|dur|dd}|tks!JdS)N ISO-8859-1)r>encodingrr>z*rNrQrrrtest_encoding_option_strVsrcCsZtdt|ddd}td}|j|ddddWddS1s&wYdSNr;rrrBF)r_rr>rY importorskipr rErFrG)rrNrOrrrtest_correct_encoding_filees  "rr)zUTF-8zUTF-16rcCsZtdt|ddd}td}|j|d||dWddS1s&wYdSrr)rr>rrNrOrrrtest_wrong_encoding_option_lxmlms  "rcCr)Nzunknown encodingrWzuft-8r)rYrZ LookupErrorrGrrrrtest_misspelled_encodingvrrcCs*tdd}|jdd}||ksJdS)Nr;a\ 0 square 360 4.0 1 circle 360 2 triangle 180 3.0 F)xml_declaration)rYrrG)rrarQrrr!test_xml_declaration_pretty_print~s  rcCs>d}|jd|d}t|}|dur|dd}||ksJdS)Na4 0square3604.01circle3602triangle1803.0F) pretty_printr> />/>)rGr/r-r{rrrtest_no_pretty_print_with_decls  rcCs8d}|jdd|d}|dur|dd}||ksJdS)Na 0square3604.01circle3602triangle1803.0F)rrr>rr)rGr-r{rrrtest_no_pretty_print_no_decls   rcCs<tjtdd |WddS1swYdS)Nz7lxml not found, please install or use the etree parser.rW)rYrZ ImportErrorrGrrrrtest_default_parser_no_lxmls  "rcCs@tjtdd|jddWddS1swYdS)Nz,Values for parser can only be lxml or etree.rWbs4rArYrZ ValueErrorrGrrrrtest_unknown_parsers "ra% 0 square 360 4.0 1 circle 360 2 triangle 180 3.0 cCs^tdt|||dkrdndd}|j|dtksJWddS1s(wYdSNr;r1rCr stylesheet)rYrrHrG xsl_expected)xsl_row_field_outputr9rrPrrrtest_stylesheet_file_likes "rcCstdt|||dkrdndd}|dkrt|}nt|}Wdn1s.wY|j|d}|tks?JdS)Nr;r1rCrr0r)rYrrHrrIrrGrrr9rrPxsl_objrQrrrtest_stylesheet_ios   rcCshtdt|||dkrdndd }|}Wdn1s!wY|j|d}|tks2JdSr)rYrrHrIrGrrrrrtest_stylesheet_buffered_reader s   rcCs\td}tjddd}tj|jdd|j|dWddS1s'wYdS)N lxml.etreedataxmlzrow_field_output.xslt!Start tag expected, '<' not foundrWr)rYrosrOjoinrZXMLSyntaxErrorrGr lxml_etreexslrrrtest_stylesheet_wrong_paths "rvalrvcCsZtd}dgd}tj|j|d|j|dWddS1s&wYdS)Nr|)zDocument is emptyrzNone \(line 0\)rWr)rYrrrZrrG)rrrmsgrrrtest_empty_string_stylesheet"s  "rcCPtd}d}tj|jdd|j|dWddS1s!wYdS)NraR zOpening and ending tag mismatchrWr)rYrrZrrGrrrrtest_incorrect_xsl_syntax3s "rcCr)NraT zfailed to compilerWr)rYrrZXSLTParseErrorrGrrrrtest_incorrect_xsl_evalQs "rc Cstd}d}tj|jdd-td}|j||dWdn1s'wYWddSWddS1s?wYdS)Nrag zCannot resolve URIrWrBr)rYrrZXSLTApplyErrorrErFrG)rrrrOrrrtest_incorrect_xsl_applyms  "rcCsFd}tjtdd|jd|dWddS1swYdS)NaI z*To use stylesheet, you need lxml installedrWr=)r>rr)rrrrrtest_stylesheet_with_etrees  "rcCsFtdd}|jdd}|dur|}|j|d}||ks!JdS)Nr;aQ , ,shape,degrees,sides  )lineterminatorr)rYrto_csvrKrG)rrout_csvout_xmlrrrtest_style_to_csvs   rcCs4tdd}|}|jd|d}||ksJdS)Nr;a shape degrees sides NaN)rxr)rYr to_stringrG)rrout_strrrrrtest_style_to_strings rcCs2tdd}|}|j|d}||ksJdS)Nr;a] " {"shape":{ },"degrees":{ },"sides":{ }} , r)rYrto_jsonrG)rrout_jsonrrrrtest_style_to_jsons % rc Cst,}|j|||dt|d|d }|j}Wdn1s$wYWdn1s3wYt|}t|ksDJdS)Nr> compressionr1r  rErFrGrhandlerIr/geom_xmlrK)r>compression_onlyrrO handle_objrQrrrtest_compression_outputs   rc Csd||}tj|d,}|j|||dt|d|d }|j}Wdn1s,wYWdn1s;wYt|}t|ksLJdS)Nzxml.)filenamerr1r r )r>rrcompression_to_extensioncompfilerOrrQrrrtest_filename_and_suffix_comp*s   rcCs<d}tdtgi|}|j|d}t||ksJdS)Nzd 0 arA)rrastyperGr/rK)any_numeric_ea_dtyper>radfresultrrrtest_ea_dtypes=s rc Cs|tjtdd-t}|j||ddWdn1swYWddSWddS1s7wYdS)NzUnrecognized compression typerW7zr)rYrZrrErFrGr\rrrtest_unsuported_compressionKs "rcCsztd}tdtttf"|jdd}||j|j d|jdd|dWddS1s6wYdS) Ns3fsr;T)anonzs3://z /geom.xmlzip)r r>) rYrrEexternal_error_raisedPermissionErrorFileNotFoundError S3FileSystemlsnamerG)r>s3_public_bucketrrfsrrrtest_s3_permission_outputTs    "r()` __future__riorrrnumpyrrYpandas.util._test_decoratorsutil_test_decoratorstdpandasrrrpandas._testing_testingrEpandas.io.commonr pandas.io.xmlr fixturerr+r,r/r9r6 skip_if_nor>rRrTrVr]rbrgmark parametrizerrangerqrsruryr|rrrrrrrrrrrrrrrrrrrrrrrrrskip_if_installedrrrrrrrrrrrrrrrr rrrr single_cpur(rrrrs    %  /       ')" &! /'$ $$     "0