U f/e; @sHddlmZddlmZmZddlZddlZddlZddl m Z ddl m m ZddlmZddlmZddlmZddlmZedd d gd d d gd ejdgdZeddddddddgddddddddgddddddddgdddd d!d"d#d$gd%Zd&Zd'd(Zejd)d*gd+d,d-Zejejd.e d.d/d0gd+d1d2Z!d3d4Z"d5d6Z#d7d8Z$d9d:Z%d;d<Z&d=d>Z'd?Z(d@dAZ)dBdCZ*dDdEZ+ej,j-e dFdGdHdIZ.ej,j-e dJdGdKdLZ/dMdNZ0dOdPZ1dQdRZ2dSdTZ3dUdVZ4dWdXZ5dYdZZ6ej,j-e dJdGd[d\Z7d]d^Z8ej,j-e dJdGd_d`Z9dadbZ:dcddZ;dedfZdjdkZ?e d.dldmZ@e d.ej,AdndodpdqgdrdsZBdtduZCe d.dvdwZDdxdyZEdzd{ZFeGd.d|d}ZHd~dZIdZJe d.ddZKe d.ddZLe d.ddZMe d.ddZNe d.ej,AdddgddZOe d.ddZPe d.ddZQe d.ddZRddZSe d.ddZTe d.ddZUe d.ddZVd?ZWej,AdddddgddZXej,Adddddgej,AdddddgddZYddZZej[e de d.ddZ\dS)) annotations)BytesIOStringION)PY38) DataFrame) 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%c Cs~|dddd}t||d}tdP}|j||dt|d}|d}W5QRXt|}|t kspt W5QRXdSr%r-r7rrrtest_file_output_bytes_reads   r?cCs>|dddd}t||d}|j|d}t|}|tks:tdS)Nr&r'r(r)r*)rr0rr5r6r8r$r9r:r=rrrtest_str_outputs   rAc Cs,tjtddtjd|dW5QRXdS)Nu5No such file or directory|没有那个文件或目录matchz/my/fake/path/output.xmlr*)pytestraisesFileNotFoundErrorgeom_dfr0r*rrrtest_wrong_file_paths rHc 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+F)indexr$rr, rr.r/r0r1r2r3r4rr6r8r$expectedr9r:r;r<r=rrrtest_index_falses   rMc 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)rI root_nameZrow_namer$rr,rJrKrrr test_index_false_rename_row_roots   rOa 0 square 360 4.0 1 circle 360 2 triangle 180 3.0 cCs$tj|d}t|}|tks tdS)Nr*rGr0r na_expectedr6r8r$r=rrrtest_na_elem_output@s rScCs&tjd|d}t|}|tks"tdS)Nna_repr$rPrRrrrtest_na_empty_str_elem_optionGsrWcCs*d}tjd|d}t|}||ks&tdS)Na 0 square 360 4.0 1 circle 360 0.0 2 triangle 180 3.0 z0.0rUrGr0rr6r8r$rLr=rrrtest_na_empty_elem_optionNsrZz'etree alpha ordered attributes < py 3.8)reasoncCs0d}tjdddg|d}t|}||ks,tdS)Nz r r r attr_colsr$rXrYrrrtest_attrs_cols_nan_outputosr^z&etree alpha ordered attributes < py3.8cCs:d}tjddddgddid|d}t|}||ks6tdS) Na> rIr r rrzhttp://example.xom)r] namespacesprefixr$rXrYrrrtest_attrs_cols_prefixs  rac Cs2tjtddtjdddg|dW5QRXdS)Nno valid columnrBr degreerr\rDrEKeyErrorrGr0r*rrrtest_attrs_unknown_columnsrfc Cs,tjtddtjd|dW5QRXdS)Nz!is not a valid type for attr_colsrB"shape", "degree", "sides"r\rDrE TypeErrorrGr0r*rrrtest_attrs_wrong_typesrjcCs2d}tjddddg|d}t|}||ks.tdS)NaD 360 4.0 square 360 circle 180 3.0 triangle Fr rr )rI elem_colsr$rXr8r$Zelems_cols_expectedr=rrrtest_elems_cols_nan_outputsrmc Cs2tjtddtjdddg|dW5QRXdS)NrbrBr rcrrkr$rdr*rrrtest_elems_unknown_columnsroc Cs,tjtddtjd|dW5QRXdS)Nz!is not a valid type for elem_colsrBrgrnrhr*rrrtest_elems_wrong_typesrpcCs4d}tjdddgdg|d}t|}||ks0tdS)Na# 360 4.0 360 180 3.0 Fr rr )rIrkr]r$rXrlrrrtest_elems_and_attrs_colssrqcCsJd}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 rrrcountsummeanTrIvaluesZaggfuncZmarginsr*) planet_df pivot_tableroundr0rr6r8r$rLZpvtr=rrrtest_hierarchical_columnss! r|cCsXd}tjddgddddgdd d }|jt|jj|d }t|}||ksTt dS) Na rrrrrrsrtTrurwr\) rxryrzr0list reset_indexcolumnsrvrr6r{rrrtest_hierarchical_attrs_columns s 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 rrrrrrsrtrwr*)rxgroupbyaggrzr0rr6r8r$rLrr=rrrtest_multi_indexBs rcCsZd}tddgddddgd}|jt|jj|d }t |}||ksVt dS) Na$ rrrrrrsrtrwr\) rxrrrzr0r}r~rrvrr6rrrrtest_multi_index_attrs_colsis rcCs.d}tjddi|d}t|}||ks*tdS)Na 0 square 360 4.0 1 circle 360 2 triangle 180 3.0 rThttp://example.com)r_r$rXr$rLr=rrrtest_default_namespacesrcCs0d}tjddid|d}t|}||ks,tdS)Na 0 square 360 4.0 1 circle 360 2 triangle 180 3.0 rrr_r`r$rXrrrrtest_namespace_prefixsrc Cs2tjtddtjddid|dW5QRXdS)Nz!doc is not included in namespacesrBrTrrrrdr*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)rTrrrz7xmlns:doc="http://other.org" xmlns="http://example.com"z7xmlns="http://example.com" xmlns:doc="http://other.org"rGr0rrr6rrrr!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)rGr0r6)rLr=rrr!test_xml_declaration_pretty_printJs rcCs>d}tjd|d}t|}|dk r.|dd}||ks:tdS)Na4 0square3604.01circle3602triangle1803.0F) pretty_printr$ />/>rrrrrtest_no_pretty_print_with_declgs  rcCs8d}tjdd|d}|dk r(|dd}||ks4tdS)Na 0square3604.01circle3602triangle1803.0F)rrr$rr)rGr0rr6rrrrtest_no_pretty_print_no_decl|s   rc Cs&tjtddtW5QRXdS)Nz7lxml not found, please install or use the etree parser.rB)rDrE ImportErrorrGr0rrrrtest_default_parser_no_lxmls rc Cs*tjtddtjddW5QRXdS)Nz,Values for parser can only be lxml or etree.rBZbs4r*rDrE ValueErrorrGr0rrrrtest_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)r1rGr0 xsl_expectedr6)r8r!xslr<rrrtest_stylesheet_file_likes rc Csb|dddd}t||(}|dkr0t|}n t|}W5QRXtj|d}|tks^tdS)Nr&r'r(rrr)r1rr2rrGr0rr6)r8r!Zxsl_pathr<xsl_objr=rrrtest_stylesheet_ios  rc CsH|dddd}t||}|}W5QRXtj|d}|tksDtdSr)r1r2rGr0rr6)r8r!rr<rr=rrrtest_stylesheet_buffered_readers   rc CsFddlm}tjddd}tj|ddtj|dW5QRXdS) NrXMLSyntaxErrorr'r(zrow_field_output.xsltz!Start tag expected, '<' not foundrBr) lxml.etreerosr;joinrDrErGr0)r8rrrrrtest_stylesheet_wrong_paths rvalrTc Cs6ddlm}tj|ddtj|dW5QRXdS)Nrrz3Document is empty|Start tag expected, '<' not foundrBrrrrDrErGr0)rrrrrtest_empty_string_stylesheets  rc Cs:ddlm}d}tj|ddtj|dW5QRXdS)NrraR zOpening and ending tag mismatchrBrr)rrrrrtest_incorrect_xsl_syntaxs rc Cs:ddlm}d}tj|ddtj|dW5QRXdS)Nr)XSLTParseErroraT zfailed to compilerBr)rrrDrErGr0)rrrrrtest_incorrect_xsl_evals rc CsRddlm}d}tj|dd*td}tj||dW5QRXW5QRXdS)Nr)XSLTApplyErrorag zCannot resolve URIrBr+r)rrrDrEr.r/rGr0)r$rrr;rrrtest_incorrect_xsl_apply2s    rc Cs0d}tjtddtjd|dW5QRXdS)NaI z*To use stylesheet, you need lxml installedrBr#)r$rr)r8rrrrtest_stylesheet_with_etreeGs  rcCs<d}tjdd}|dk r |}tj|d}||ks8tdS)NaQ , ,shape,degrees,sides  )Zline_terminatorr)rGZto_csvr4r0r6)rZout_csvout_xmlrrrtest_style_to_csvYs   rcCs*d}t}tjd|d}||ks&tdS)Na shape degrees sides NaN)rVr)rGZ to_stringr0r6)rZout_strrrrrtest_style_to_stringvsrcCs(d}t}tj|d}||ks$tdS)Na] " {"shape":{ },"degrees":{ },"sides":{ }} , r)rGZto_jsonr0r6)rZout_jsonrrrrtest_style_to_jsons% rcompbz2gzipxzzipc Csdt:}tj|||dt|d|d}|j}W5QRXW5QRXt|}t| ks`t dS)Nr$ compressionrr r.r/rGr0rhandler2rgeom_xmlr4r6)r$rr; handle_objr=rrrtest_compression_outputs rcompfilezxml.bz2zxml.gzzxml.xzzxml.zipc Cshtj|d:}tj|||dt|d|d}|j}W5QRXW5QRXt|}t| ksdt dS)N)r9rrrr)r$rrr;rr=rrrtest_filename_and_suffix_compsrc CsBtjtdd*t}tj||ddW5QRXW5QRXdS)NzUnrecognized compression typerBZ7zr)rDrErr.r/rGr0)r8r$r;rrrtest_unsuported_compressions rs3fsc CsLddl}tjtdd,|jdd}|dtjdd|d W5QRXdS) Nrz Access DeniedrBT)Zanonz pandas-testzs3://pandas-test/geom.xmlr)rr$)rrDrEPermissionErrorZ S3FileSystemZlsrGr0)r$rfsrrrtest_s3_permission_output s   r)] __future__rr&rrrZnumpynprDZ pandas.compatrZpandas.util._test_decoratorsutilZ_test_decoratorstdZpandasrZpandas._testingZ_testingr.Zpandas.io.commonrZ pandas.io.xmlrnanrGrxr5rZfixturer!rZ skip_if_nor$r>r?rArHrMrOrQrSrWrZmarkZskipifr^rarfrjrmrorprqr|rrrrrrrrrrZ parametrizerrrrrZskip_if_installedrrrrrrrrrrrrrrrrrrrnetworkrrrrrs4      &    -     (,!   / ' ! +$               ! /