U c @sddlmZddlmZmZddlZddlZddlZddl m m Z ddl mZmZmZddlmZddlmZddlmZddlmZedd d gd d d gd ejdgdZeddddddddgddddddddgddddddddgdddd d!d"d#d$gd%Zd&Zd'd(Zejd)d*gd+d,d-Z ejej!d.e "d.d/d0gd+d1d2Z#d3d4Z$d5d6Z%d7d8Z&d9d:Z'd;d<Z(d=d>Z)ej*+d?e,e-d@dAdBdCe-d@dADgdDdEZ.dFZ/dGdHZ0dIdJZ1dKdLZ2dMdNZ3dOdPZ4dQdRZ5dSdTZ6dUdVZ7dWdXZ8dYdZZ9d[d\Z:d]d^Z;d_d`ZdedfZ?dgdhZ@didjZAdkdlZBdmZCdndoZDe "d.dpdqZEe "d.ej*+drdsdtdugdvdwZFdxdyZGe "d.dzd{ZHd|d}ZId~dZJe Kd.ddZLddZMdZNe "d.ddZOe "d.ddZPe "d.ddZQe "d.ddZRe "d.ej*+dddgddZSe "d.ddZTe "d.ddZUe "d.ddZVddZWe "d.ddZXe "d.ddZYe "d.ddZZdFZ[ddZ\ddZ]ddZ^ddZ_ej*j`e "de "d.ddZadS)) annotations)BytesIOStringION)NA DataFrameIndex)_compression_to_extension) get_handle)read_xmlZsquareZcircleZtriangleih)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 r2rr7r8r:r&r;r<r?rrrtest_str_outputs   rCc Cs0d}tjtddtj||dW5QRXdS)Nz/my/fake/path/output.xmlz6Cannot save file into a non-existent directory: .*pathmatchr,)pytestraisesOSErrorgeom_dfr2r&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 r0r1r2r3r4r5r6rr8r: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-FZbooksZbook)rM root_nameZrow_namer&rr.rNrOrrr test_index_false_rename_row_roots   rS offset_index cCsg|] }t|qSr)str).0irrr .srZcCs<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)rIcopyrrMr2rr8)r&rTrPZoffset_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,rIr2r na_expectedr8r&r?rrrtest_na_elem_outputns r`cCs&tjd|d}t|}|tks"tdS)Nna_repr&r]r_rrrtest_na_empty_str_elem_optionusrdcCs*d}tjd|d}t|}||ks&tdS)Na 0 square 360 4.0 1 circle 360 0.0 2 triangle 180 3.0 z0.0rbrIr2rr8r&rPr?rrrtest_na_empty_elem_option|srgcCs0d}tjdddg|d}t|}||ks,tdS)Nz rrr attr_colsr&rerfrrrtest_attrs_cols_nan_outputsrjcCs:d}tjddddgddid|d}t|}||ks6tdS) Na> rMrrrrzhttp://example.xom)ri namespacesprefixr&rerfrrrtest_attrs_cols_prefixs  rmc Cs2tjtddtjdddg|dW5QRXdS)Nno valid columnrDrdegreerrhrFrGKeyErrorrIr2r,rrrtest_attrs_unknown_columnsrrc Cs,tjtddtjd|dW5QRXdS)Nz!is not a valid type for attr_colsrD"shape", "degree", "sides"rhrFrG TypeErrorrIr2r,rrrtest_attrs_wrong_typesrvcCs2d}tjddddg|d}t|}||ks.tdS)NaD 360 4.0 square 360 circle 180 3.0 triangle Frrr)rM elem_colsr&rer&Zelems_cols_expectedr?rrrtest_elems_cols_nan_outputsryc Cs2tjtddtjdddg|dW5QRXdS)NrnrDrrorrwr&rpr,rrrtest_elems_unknown_columnsr{c Cs,tjtddtjd|dW5QRXdS)Nz!is not a valid type for elem_colsrDrsrzrtr,rrrtest_elems_wrong_typesr|cCs4d}tjdddgdg|d}t|}||ks0tdS)Na# 360 4.0 360 180 3.0 Frrr)rMrwrir&rerxrrrtest_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_tableroundr2rr8r&rPZpvtr?rrrtest_hierarchical_columnss! rcCsXd}tjddgddddgdd d }|jt|jj|d }t|}||ksTt dS) Na rrrr~rrTrrrh) rrrr2list reset_indexcolumnsrrr8rrrrtest_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 rrrr~rrrr,)rgroupbyaggrr2rr8r&rPrr?rrrtest_multi_indexds rcCsZd}tddgddddgd}|jt|jj|d }t |}||ksVt dS) Na$ rrrr~rrrrh) rrrrr2rrrrrr8rrrrtest_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 rahttp://example.com)rkr&rerfrrrtest_default_namespacesrcCs0d}tjddid|d}t|}||ks,tdS)Na 0 square 360 4.0 1 circle 360 2 triangle 180 3.0 rrrkrlr&rerfrrrtest_namespace_prefixsrc Cs2tjtddtjddid|dW5QRXdS)Nz!doc is not included in namespacesrDrarrrrpr,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)rarrrz7xmlns:doc="http://other.org" xmlns="http://example.com"z7xmlns="http://example.com" xmlns:doc="http://other.org"rIr2rrr8rfrrr!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* 0 square 360 4.0 1 circle 360 2 triangle 180 3.0 F)xml_declaration)rIr2r8)rPr?rrr!test_xml_declaration_pretty_prinths rcCs>d}tjd|d}t|}|dk r.|dd}||ks:tdS)Na4 0square3604.01circle3602triangle1803.0F) pretty_printr& />/>rrfrrrtest_no_pretty_print_with_decls  rcCs8d}tjdd|d}|dk r(|dd}||ks4tdS)Na 0square3604.01circle3602triangle1803.0F)rrr&rr)rIr2rr8rfrrrtest_no_pretty_print_no_decls   rc Cs&tjtddtW5QRXdS)Nz7lxml not found, please install or use the etree parser.rD)rFrG ImportErrorrIr2rrrrtest_default_parser_no_lxmls rc Cs*tjtddtjddW5QRXdS)Nz,Values for parser can only be lxml or etree.rDZbs4r,rFrG ValueErrorrIr2rrrrtest_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)r3rIr2 xsl_expectedr8)r:r#xslr>rrrtest_stylesheet_file_likes rc Csb|dddd}t||(}|dkr0t|}n t|}W5QRXtj|d}|tks^tdS)Nr(r)r*rrr)r3rr4rrIr2rr8)r:r#Zxsl_pathr>xsl_objr?rrrtest_stylesheet_ios  rc CsH|dddd}t||}|}W5QRXtj|d}|tksDtdSr)r3r4rIr2rr8)r:r#rr>rr?rrrtest_stylesheet_buffered_readers   rc CsFddlm}tjddd}tj|ddtj|dW5QRXdS) NrXMLSyntaxErrorr)r*zrow_field_output.xslt!Start tag expected, '<' not foundrDr) lxml.etreerosr=joinrFrGrIr2rrrrrtest_stylesheet_wrong_paths rvalrac CsFddlm}ddddg}tj||dtj|dW5QRXdS) Nrr|zDocument is emptyrzNone \(line 0\)rDr)rrrrFrGrIr2)rrmsgrrrtest_empty_string_stylesheet s  rc Cs:ddlm}d}tj|ddtj|dW5QRXdS)NrraR zOpening and ending tag mismatchrDr)rrrFrGrIr2rrrrtest_incorrect_xsl_syntaxs rc Cs:ddlm}d}tj|ddtj|dW5QRXdS)Nr)XSLTParseErroraT zfailed to compilerDr)rrrFrGrIr2)rrrrrtest_incorrect_xsl_eval:s rc CsRddlm}d}tj|dd*td}tj||dW5QRXW5QRXdS)Nr)XSLTApplyErrorag zCannot resolve URIrDr-r)rrrFrGr0r1rIr2)rrr=rrrtest_incorrect_xsl_applyWs    rc Cs0d}tjtddtjd|dW5QRXdS)NaI z*To use stylesheet, you need lxml installedrDr%)r&rr)rrrrtest_stylesheet_with_etreels  rcCs<d}tjdd}|dk r |}tj|d}||ks8tdS)NaQ , ,shape,degrees,sides  )lineterminatorr)rIZto_csvr6r2r8)rZout_csvout_xmlrrrtest_style_to_csv~s   rcCs*d}t}tjd|d}||ks&tdS)Na shape degrees sides NaN)rcr)rIZ to_stringr2r8)rZout_strrrrrtest_style_to_stringsrcCs(d}t}tj|d}||ks$tdS)Na] " {"shape":{ },"degrees":{ },"sides":{ }} , r)rIZto_jsonr2r8)rZout_jsonrrrrtest_style_to_jsons% rc Csdt:}tj|||dt|d|d}|j}W5QRXW5QRXt|}t| ks`t dS)Nr& compressionrr) r0r1rIr2r handler4rgeom_xmlr6r8)r&compression_onlyr= handle_objr?rrrtest_compression_outputs rc Cstdt|}tj|d:}tj|||dt|d|d}|j}W5QRXW5QRXt|}t | kspt dS)Nzxml.)r;rrr) rr0r1rIr2r rr4rrr6r8)r&rZcompfiler=rr?rrrtest_filename_and_suffix_comps rcCs<d}tdtgi|}|j|d}t||ks8tdS)Nzd 0 ar,)rrZastyper2rr6r8)Zany_numeric_ea_dtyper&rPZdfresultrrrtest_ea_dtypes&s rc CsBtjtdd*t}tj||ddW5QRXW5QRXdS)NzUnrecognized compression typerDZ7zr)rFrGrr0r1rIr2rJrrrtest_unsuported_compression4s rs3fsc CsLddl}tjtdd,|jdd}|dtjdd|d W5QRXdS) Nrz Access DeniedrDT)Zanonz pandas-testzs3://pandas-test/geom.xmlzip)rr&)rrFrGPermissionErrorZ S3FileSystemZlsrIr2)r&Z s3_resourcerfsrrrtest_s3_permission_output=s   r)b __future__rr(rrrZnumpynprFZpandas.util._test_decoratorsutilZ_test_decoratorstdZpandasrrrZpandas._testingZ_testingr0Z pandas.tests.io.test_compressionrZpandas.io.commonr Z pandas.io.xmlr nanrIrr7rZfixturer#r Z skip_if_nor&r@rArCrKrQrSmarkZ parametrizerranger\r^r`rdrgrjrmrrrvryr{r|r}rrrrrrrrrrrrrrrrZskip_if_installedrrrrrrrrrrrrrrrrrrrrZ single_cpurrrrrs     &    -     (*  &! /'! +$               ! /