a Pf@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)Zsquarecircletriangle)hr shapedegreessides)rnpnanrrX/var/www/ai-form-bot/venv/lib/python3.9/site-packages/pandas/tests/io/xml/test_to_xml.pygeom_df;s  rcCs"tgdgdgdgddS)N)ZMercuryZVenusZEarthZMarsZJupiterZSaturnZUranusZNeptune) terrestrialrrr gas giantr ice giantr)innerrrrouterrrr)g`|x ?gc AJx@g2r@gl?gh|@g @gd]FU@g hY@)Zplanettypelocationmass)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%rrrtest_file_output_bytes_reads   0rEcCs0t||d}|j|d}t|}||ks,JdSNr3)r r9r%)r?r1r"r@rCrrrtest_str_outputs  rGcCsDd}tjtdd|j||dWdn1s60YdS)Nz/my/fake/path/output.xmlz6Cannot save file into a non-existent directory: .*pathmatchr3)pytestraisesOSErrorr9r1rrArrrtest_wrong_file_paths rNc Csd}t||d}tdp}|j|d|dt|d"}|d}Wdn1s^0Yt|}||ks|JWdn1s0YdS)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 r3r4Findexr1r&r5r6r?r1expectedr@rArBrCrrrtest_index_falses   0rSc Csd}t||d}tdt}|j|ddd|dt|d"}|d }Wdn1sb0Yt|}||ksJWdn1s0YdS) 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 r3r4Fbooksbook)rPZ root_nameZrow_namer1r&r5r6rQrrr test_index_false_rename_row_roots    0rV offset_index cCsg|] }t|qSr)str).0irrr .r]cCs<d}|}t||_|jd|d}t|}||ks8JdS)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 FrON)copyrrPr9r%)r1rWrrRZoffset_geom_dfrCrrr(test_index_false_with_offset_input_index-s  r`a 0 square 360 4.0 1 circle 360 2 triangle 180 3.0 cCs$|j|d}t|}|tks JdSrFr9r% na_expectedr1rrCrrrtest_na_elem_outputns rdcCs&|jd|d}t|}|tks"JdS)Nna_repr1rarcrrrtest_na_empty_str_elem_optionusrhcCs*d}|jd|d}t|}||ks&JdS)Na 0 square 360 4.0 1 circle 360 0.0 2 triangle 180 3.0 z0.0rfr9r%r1rrRrCrrrtest_na_empty_elem_option|srkcCs.d}|jgd|d}t|}||ks*JdS)Nz r attr_colsr1rirjrrrtest_attrs_cols_nan_outputsrncCs6d}|jgdddid|d}t|}||ks2JdS)Na> )rPrrrr$zhttp://example.xom)rm namespacesprefixr1rirjrrrtest_attrs_cols_prefixs rqcCsDtjtdd"|jgd|dWdn1s60YdS)Nno valid columnrHrZdegreerrlrJrKKeyErrorr9r1rrrrtest_attrs_unknown_columnsrwcCs@tjtdd|jd|dWdn1s20YdS)Nz!is not a valid type for attr_colsrH"shape", "degree", "sides"rlrJrK TypeErrorr9rvrrrtest_attrs_wrong_typesr{cCs0d}|jdgd|d}t|}||ks,JdS)NaD 360 4.0 square 360 circle 180 3.0 triangle F)rrr)rP elem_colsr1rir1rZelems_cols_expectedrCrrrtest_elems_cols_nan_outputs  r~cCsDtjtdd"|jgd|dWdn1s60YdS)NrrrHrsr|r1rtrvrrrtest_elems_unknown_columnsrcCs@tjtdd|jd|dWdn1s20YdS)Nz!is not a valid type for elem_colsrHrxrryrvrrrtest_elems_wrong_typesrcCs4d}|jdddgdg|d}t|}||ks0JdS)Na# 360 4.0 360 180 3.0 Frrr)rPr|rmr1rir}rrrtest_elems_and_attrs_colssrcCsHd}|jddgdgdddd}|j|d }t|}||ksDJdS) 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 rrr countsummeanTrPvaluesZaggfuncZmarginsr3) pivot_tableroundr9r%r1r!rRZpvtrCrrrtest_hierarchical_columnss! rcCsVd}|jddgdgdddd}|jt|jj|d }t|}||ksRJdS) Na rrr rTrrrl)rrr9list reset_indexcolumnsrr%rrrrtest_hierarchical_attrs_columnsFs rcCsJd}|ddgdgdd}|j|d}t|}||ksFJdS)Na inner terrestrial 4 11.81 2.95 outer gas giant 2 2466.5 1233.25 outer ice giant 2 189.23 94.61 rrr rrr3)groupbyaggrr9r%r1r!rRrrCrrrtest_multi_indexds rcCsXd}|ddgdgdd}|jt|jj|d}t|}||ksTJdS)Na$ rrr rrrl) rrrr9rrrrr%rrrrtest_multi_index_attrs_colss rcCs.d}|jddi|d}t|}||ks*JdS)Na 0 square 360 4.0 1 circle 360 2 triangle 180 3.0 rehttp://example.comror1rirjrrrtest_default_namespacesrcCs0d}|jddd|d}t|}||ks,JdS)Na 0 square 360 4.0 1 circle 360 2 triangle 180 3.0 http://other.orgr)Zothexrrirjrrrtest_unused_namespacessrcCs0d}|jddid|d}t|}||ks,JdS)Na 0 square 360 4.0 1 circle 360 2 triangle 180 3.0 r$rrorpr1rirjrrrtest_namespace_prefixs  rcCsFtjtdd$|jddid|dWdn1s80YdS)Nz!doc is not included in namespacesrHrerr$rrtrvrrrtest_missing_prefix_in_nmsps rcCs2d}|jdddd|d}t|}||ks.JdS)Na7 0 square 360 4.0 1 circle 360 2 triangle 180 3.0 rr)rer$r$rrirjrrr!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}|dur6|dd}|tksBJdS)N ISO-8859-1)r1encodingrr1z* 0 square 360 4.0 1 circle 360 2 triangle 180 3.0 F)xml_declaration)rJrr9)rrRrCrrr!test_xml_declaration_pretty_print~s  rcCs>d}|jd|d}t|}|dur.|dd}||ks:JdS)Na4 0square3604.01circle3602triangle1803.0F) pretty_printr1 />/>)r9r%r#rjrrrtest_no_pretty_print_with_decls  rcCs8d}|jdd|d}|dur(|dd}||ks4JdS)Na 0square3604.01circle3602triangle1803.0F)rrr1rr)r9r#rjrrrtest_no_pretty_print_no_decls   rcCs:tjtdd|Wdn1s,0YdS)Nz7lxml not found, please install or use the etree parser.rH)rJrK ImportErrorr9rrrrtest_default_parser_no_lxmlsrcCs>tjtdd|jddWdn1s00YdS)Nz,Values for parser can only be lxml or etree.rHZbs4r3rJrK ValueErrorr9rrrrtest_unknown_parsersra% 0 square 360 4.0 1 circle 360 2 triangle 180 3.0 cCs\tdt|||dkrdndd$}|j|dtks:JWdn1sN0YdSNr/r'r5r stylesheet)rJrr:r9 xsl_expected)xsl_row_field_outputr.rrBrrrtest_stylesheet_file_likes  rcCstdt|||dkrdndd2}|dkr zOpening and ending tag mismatchrHr)rJrrKrr9rrrrtest_incorrect_xsl_syntax3s  rcCsNtd}d}tj|jdd|j|dWdn1s@0YdS)NraT zfailed to compilerHr)rJrrKZXSLTParseErrorr9rrrrtest_incorrect_xsl_evalQs rc Csztd}d}tj|jddHtd}|j||dWdn1sN0YWdn1sl0YdS)Nrag zCannot resolve URIrHr4r)rJrrKZXSLTApplyErrorr7r8r9)rrrrArrrtest_incorrect_xsl_applyms    rcCsDd}tjtdd|jd|dWdn1s60YdS)NaI z*To use stylesheet, you need lxml installedrHr0)r1rr)rrrrrtest_stylesheet_with_etrees  rcCsFtdd}|jdd}|dur*|}|j|d}||ksBJdS)Nr/aQ , ,shape,degrees,sides  )lineterminatorr)rJrZto_csvr=r9)rrZout_csvout_xmlrrrtest_style_to_csvs   rcCs4tdd}|}|jd|d}||ks0JdS)Nr/a shape degrees sides NaN)rgr)rJrZ to_stringr9)rrZout_strrrrrtest_style_to_strings  rcCs2tdd}|}|j|d}||ks.JdS)Nr/a] " {"shape":{ },"degrees":{ },"sides":{ }} , r)rJrto_jsonr9)rrZout_jsonrrrrtest_style_to_jsons  % rc CstX}|j|||dt|d|d}|j}Wdn1sH0YWdn1sf0Yt|}t|ksJdS)Nr1 compressionr'r r7r8r9rhandler;r%geom_xmlr=)r1compression_onlyrrA handle_objrCrrrtest_compression_outputs Frc Csd||}tj|dX}|j|||dt|d|d}|j}Wdn1sX0YWdn1sv0Yt|}t|ksJdS)Nzxml.)filenamerr'rr)r1rrZcompression_to_extensionZcompfilerArrCrrrtest_filename_and_suffix_comp*s FrcCs<d}tdtgi|}|j|d}t||ks8JdS)Nzd 0 ar3)rrZastyper9r%r=)Zany_numeric_ea_dtyper1rRZdfresultrrrtest_ea_dtypes=s rc CsjtjtddHt }|j||ddWdn1s>0YWdn1s\0YdS)NzUnrecognized compression typerHZ7zr)rJrKrr7r8r9rMrrrtest_unsuported_compressionKs rcCsxtd}tdtttfB|jdd}||j|j d|jdd|dWdn1sj0YdS) Ns3fsr/T)Zanonzs3://z /geom.xmlzip)rr1) rJrr7Zexternal_error_raisedPermissionErrorFileNotFoundErrorZ S3FileSystemZlsnamer9)r1Zs3_public_bucketrrfsrrrtest_s3_permission_outputTs    r)` __future__riorrrnumpyrrJZpandas.util._test_decoratorsutilZ_test_decoratorstdZpandasrrrZpandas._testingZ_testingr7Zpandas.io.commonrZ pandas.io.xmlr Zfixturerr!r"r%r.r+Z skip_if_nor1rDrErGrNrSrVmarkZ parametrizerranger`rbrdrhrkrnrqrwr{r~rrrrrrrrrrrrrrrrrrrrZskip_if_installedrrrrrrrrrrrrrrrrrrrrZ single_cpurrrrrs    %  /        ')" &! /'$ $$      "0