U +ifۆ @s ddlmZddlmZmZddlZddlZddlZddl m m Z ddl mZmZmZddlmZddlmmZddlmZddlmZeddd gd d d gd ejd gdZeddddddddgddddddddgddddddddgddddd d!d"d#gd$Zd%Zd&d'Zejd(d)gd*d+d,Z ejej!d-e "d-d.d/gd*d0d1Z#d2d3Z$d4d5Z%d6d7Z&d8d9Z'd:d;Z(de,e-d?d@dAdBe-d?d@DgdCdDZ.dEZ/dFdGZ0dHdIZ1dJdKZ2dLdMZ3dNdOZ4dPdQZ5dRdSZ6dTdUZ7dVdWZ8dXdYZ9dZd[Z:d\d]Z;d^d_ZdddeZ?dfdgZ@dhdiZAdjdkZBdlZCdmdnZDe "d-dodpZEe "d-ej*+dqdrdsdtgdudvZFdwdxZGe "d-dydzZHd{d|ZId}d~ZJe Kd-ddZLddZMdZNe "d-ddZOe "d-ddZPe "d-ddZQe "d-ddZRe "d-ej*+dddgddZSe "d-ddZTe "d-ddZUe "d-ddZVddZWe "d-ddZXe "d-ddZYe "d-ddZZdEZ[ddZ\ddZ]ddZ^ddZ_ej*j`e "de "d-ddZadS)) annotations)BytesIOStringION)NA DataFrameIndex) get_handle)read_xmlZsquarecircletriangleih)shapedegreessidesZMercuryZVenusZEarthZMarsZJupiterZSaturnZUranusZNeptuneZ terrestrialz gas giantz ice giantinnerouterg`|x ?gc AJx@g2r@gl?gh|@g @gd]FU@g hY@)Zplanettypelocationmassac 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 cCs|dk r|dd}|S)Nz%|dddd}t||d}|j|d}t|}|tks:tdS)Nr)r*r+r,r-)r r3rr8r9r;r'r<r=r@rrrtest_str_outputs   rDc Cs0d}tjtddtj||dW5QRXdS)Nz/my/fake/path/output.xmlz6Cannot save file into a non-existent directory: .*pathmatchr-)pytestraisesOSErrorgeom_dfr3)r'r>rrrtest_wrong_file_paths rKc Csd}|dddd}t||d}tdR}|j|d|d t|d }|d }W5QRXt|}||ksvt W5QRXdS) 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 r)r*r+r,r-r.Findexr'rr/ r r1r2r3r4r5r6r7rr9r;r'expectedr<r=r>r?r@rrrtest_index_falses   rQc Csd}|dddd}t||d}tdV}|j|dd d |d t|d }|d }W5QRXt|}||kszt W5QRXdS)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 r)r*r+r,r-r.Fbooksbook)rMZ root_nameZrow_namer'rr/rNrOrrr test_index_false_rename_row_roots   rT offset_index cCsg|] }t|qSr)str).0irrr .sr[cCs<d}t}t||_|jd|d}t|}||ks8tdS)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 FrLN)rJcopyrrMr3rr9)r'rUrPZoffset_geom_dfr@rrr(test_index_false_with_offset_input_index-s  r]a 0 square 360 4.0 1 circle 360 2 triangle 180 3.0 cCs$tj|d}t|}|tks tdS)Nr-rJr3r na_expectedr9r;r'r@rrrtest_na_elem_outputns racCs&tjd|d}t|}|tks"tdS)Nna_repr'r^r`rrrtest_na_empty_str_elem_optionusrecCs*d}tjd|d}t|}||ks&tdS)Na 0 square 360 4.0 1 circle 360 0.0 2 triangle 180 3.0 z0.0rcrJr3rr9r;r'rPr@rrrtest_na_empty_elem_option|srhcCs0d}tjdddg|d}t|}||ks,tdS)Nz rrr attr_colsr'rfrgrrrtest_attrs_cols_nan_outputsrkcCs:d}tjddddgddid|d}t|}||ks6tdS) Na> rMrrrrzhttp://example.xom)rj namespacesprefixr'rfrgrrrtest_attrs_cols_prefixs  rnc Cs2tjtddtjdddg|dW5QRXdS)Nno valid columnrErdegreerrirGrHKeyErrorrJr3r-rrrtest_attrs_unknown_columnsrsc Cs,tjtddtjd|dW5QRXdS)Nz!is not a valid type for attr_colsrE"shape", "degree", "sides"rirGrH TypeErrorrJr3r-rrrtest_attrs_wrong_typesrwcCs2d}tjddddg|d}t|}||ks.tdS)NaD 360 4.0 square 360 circle 180 3.0 triangle Frrr)rM elem_colsr'rfr;r'Zelems_cols_expectedr@rrrtest_elems_cols_nan_outputsrzc Cs2tjtddtjdddg|dW5QRXdS)NrorErrprrxr'rqr-rrrtest_elems_unknown_columnsr|c Cs,tjtddtjd|dW5QRXdS)Nz!is not a valid type for elem_colsrErtr{rur-rrrtest_elems_wrong_typesr}cCs4d}tjdddgdg|d}t|}||ks0tdS)Na# 360 4.0 360 180 3.0 Frrr)rMrxrjr'rfryrrrtest_elems_and_attrs_colssr~cCsJd}tjddgddddgdd d }|j|d }t|}||ksFtdS) 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 rrrcountsummeanTrMvaluesZaggfuncZmarginsr-) planet_df pivot_tableroundr3rr9r;r'rPZpvtr@rrrtest_hierarchical_columnss! rcCsXd}tjddgddddgdd d }|jt|jj|d }t|}||ksTt dS) Na rrrrrrTrrri) rrrr3list reset_indexcolumnsrrr9rrrrtest_hierarchical_attrs_columnsFs rcCsLd}tddgddddgd}|j|d }t|}||ksHtdS) Na inner terrestrial 4 11.81 2.95 outer gas giant 2 2466.5 1233.25 outer ice giant 2 189.23 94.61 rrrrrrrr-)rgroupbyaggrr3rr9r;r'rPrr@rrrtest_multi_indexds rcCsZd}tddgddddgd}|jt|jj|d }t |}||ksVt dS) Na$ rrrrrrrri) rrrrr3rrrrrr9rrrrtest_multi_index_attrs_colss rcCs.d}tjddi|d}t|}||ks*tdS)Na 0 square 360 4.0 1 circle 360 2 triangle 180 3.0 rbhttp://example.com)rlr'rfr'rPr@rrrtest_default_namespacesrcCs0d}tjddid|d}t|}||ks,tdS)Na 0 square 360 4.0 1 circle 360 2 triangle 180 3.0 rrrlrmr'rfrrrrtest_namespace_prefixsrc Cs2tjtddtjddid|dW5QRXdS)Nz!doc is not included in namespacesrErbrrrrqr-rrrtest_missing_prefix_in_nmsps rcCsFd}tjdddd|d}t|}|dk r6|dd}||ksBtdS) Na7 0 square 360 4.0 1 circle 360 2 triangle 180 3.0 rzhttp://other.org)rbrrrz7xmlns:doc="http://other.org" xmlns="http://example.com"z7xmlns="http://example.com" xmlns:doc="http://other.org"rJr3rrr9rrrr!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 cCsT|dddd}t||ddd}|jd|d}|dk rD|d d }|tksPtdS) Nr)r*r+baby_names.xml ISO-8859-1)r'encodingrr'z*rrrtest_correct_encoding_fileMs rrzUTF-8zUTF-16rc CsH|dddd}t|ddd}td}|j|d ||d W5QRXdSrr)r;r'rr<r=r>rrrtest_wrong_encoding_option_lxmlVs rc Cs,tjtddtjd|dW5QRXdS)Nzunknown encodingrEzuft-8r)rGrH LookupErrorrJr3r-rrrtest_misspelled_encoding`srcCs d}tjdd}||kstdS)Na\ 0 square 360 4.0 1 circle 360 2 triangle 180 3.0 F)xml_declaration)rJr3r9)rPr@rrr!test_xml_declaration_pretty_prinths rcCs>d}tjd|d}t|}|dk r.|dd}||ks:tdS)Na4 0square3604.01circle3602triangle1803.0F) pretty_printr' />/>rrrrrtest_no_pretty_print_with_decls  rcCs8d}tjdd|d}|dk r(|dd}||ks4tdS)Na 0square3604.01circle3602triangle1803.0F)rrr'rr)rJr3rr9rrrrtest_no_pretty_print_no_decls   rc Cs&tjtddtW5QRXdS)Nz7lxml not found, please install or use the etree parser.rE)rGrH ImportErrorrJr3rrrrtest_default_parser_no_lxmls rc Cs*tjtddtjddW5QRXdS)Nz,Values for parser can only be lxml or etree.rEZbs4r-rGrH ValueErrorrJr3rrrrtest_unknown_parsers ra% 0 square 360 4.0 1 circle 360 2 triangle 180 3.0 c Cs<|dddd}t||}tj|dtks.tW5QRXdSNr)r*r+row_field_output.xsl stylesheet)r4rJr3 xsl_expectedr9)r;r$xslr?rrrtest_stylesheet_file_likes rc Csb|dddd}t||(}|dkr0t|}n t|}W5QRXtj|d}|tks^tdS)Nr)r*r+rrr)r4rr5rrJr3rr9)r;r$Zxsl_pathr?xsl_objr@rrrtest_stylesheet_ios  rc CsH|dddd}t||}|}W5QRXtj|d}|tksDtdSr)r4r5rJr3rr9)r;r$rr?rr@rrrtest_stylesheet_buffered_readers   rc CsFddlm}tjddd}tj|ddtj|dW5QRXdS) NrXMLSyntaxErrorr*r+zrow_field_output.xsltz!Start tag expected, '<' not foundrEr) lxml.etreerosr>joinrGrHrJr3)r;rrrrrtest_stylesheet_wrong_paths rvalrbc Cs6ddlm}tj|ddtj|dW5QRXdS)Nrrz3Document is empty|Start tag expected, '<' not foundrErrrrGrHrJr3)rrrrrtest_empty_string_stylesheet s  rc Cs:ddlm}d}tj|ddtj|dW5QRXdS)NrraR zOpening and ending tag mismatchrErr)rrrrrtest_incorrect_xsl_syntaxs rc Cs:ddlm}d}tj|ddtj|dW5QRXdS)Nr)XSLTParseErroraT zfailed to compilerEr)rrrGrHrJr3)rrrrrtest_incorrect_xsl_eval3s rc CsRddlm}d}tj|dd*td}tj||dW5QRXW5QRXdS)Nr)XSLTApplyErrorag zCannot resolve URIrEr.r)rrrGrHr1r2rJr3)r'rrr>rrrtest_incorrect_xsl_applyPs    rc Cs0d}tjtddtjd|dW5QRXdS)NaI z*To use stylesheet, you need lxml installedrEr&)r'rr)r;rrrrtest_stylesheet_with_etreees  rcCs<d}tjdd}|dk r |}tj|d}||ks8tdS)NaQ , ,shape,degrees,sides  )Zline_terminatorr)rJZto_csvr7r3r9)rZout_csvout_xmlrrrtest_style_to_csvws   rcCs*d}t}tjd|d}||ks&tdS)Na shape degrees sides NaN)rdr)rJZ to_stringr3r9)rZout_strrrrrtest_style_to_stringsrcCs(d}t}tj|d}||ks$tdS)Na] " {"shape":{ },"degrees":{ },"sides":{ }} , r)rJto_jsonr3r9)rZout_jsonrrrrtest_style_to_jsons% rc Csdt:}tj|||dt|d|d}|j}W5QRXW5QRXt|}t| ks`t dS)Nr' compressionrr) r1r2rJr3rhandler5rgeom_xmlr7r9)r'compression_onlyr> handle_objr@rrrtest_compression_outputs rc Csvdtj|}tj|d:}tj|||dt|d|d}|j}W5QRXW5QRXt |}t | ksrt dS)Nzxml.)r<rrr) icomZ_compression_to_extensionr1r2rJr3rrr5rrr7r9)r'rZcompfiler>rr@rrrtest_filename_and_suffix_compsrcCs<d}tdtgi|}|j|d}t||ks8tdS)Nzd 0 ar-)rrZastyper3rr7r9)Zany_numeric_ea_dtyper'rPZdfresultrrrtest_ea_dtypess rc CsBtjtdd*t}tj||ddW5QRXW5QRXdS)NzUnrecognized compression typerEZ7zr)rGrHrr1r2rJr3)r;r'r>rrrtest_unsuported_compression-s rs3fsc CsLddl}tjtdd,|jdd}|dtjdd|d W5QRXdS) Nrz Access DeniedrET)Zanonz pandas-testzs3://pandas-test/geom.xmlzip)rr')rrGrHPermissionErrorZ S3FileSystemZlsrJr3)r'Z s3_resourcerfsrrrtest_s3_permission_output6s   r)b __future__rr)rrrnumpynprGZpandas.util._test_decoratorsutilZ_test_decoratorstdZpandasrrrZpandas._testingZ_testingr1Zpandas.io.commoncommonrrZ pandas.io.xmlr nanrJrr8rZfixturer$r!Z skip_if_nor'rArBrDrKrQrTmarkZ parametrizerranger]r_rarerhrkrnrsrwrzr|r}r~rrrrrrrrrrrrrrrrZskip_if_installedrrrrrrrrrrrrrrrrrrrrZ single_cpurrrrrs    &    -     (*  &! /'! +$               ! /