o i@sddlmZddlmZmZddlZddlZddlZddl m m Z ddl mZmZmZddlmZddlmZddlmZddlmZ egdgd d ejd gd Zegd gdgdgddZdZddZejddgdddZ ejej!de "dddgdddZ#dd Z$d!d"Z%d#d$Z&d%d&Z'd'd(Z(d)d*Z)ej*+d+e,e-d,d-d.d/e-d,d-Dgd0d1Z.d2Z/d3d4Z0d5d6Z1d7d8Z2d9d:Z3d;d<Z4d=d>Z5d?d@Z6dAdBZ7dCdDZ8dEdFZ9dGdHZ:dIdJZ;dKdLZdQdRZ?dSdTZ@dUdVZAdWdXZBdYZCdZd[ZDe "dd\d]ZEe "dej*+d^gd_d`daZFdbdcZGe "ddddeZHdfdgZIdhdiZJe KddjdkZLdldmZMdnZNe "ddodpZOe "ddqdrZPe "ddsdtZQe "ddudvZRe "dej*+dwdxdygdzd{ZSe "dd|d}ZTe "dd~dZUe "dddZVddZWe "dddZXe "dddZYe "dddZZd2Z[ddZ\ddZ]ddZ^ddZ_ej*j`e "de "dddZadS)) annotations)BytesIOStringION)NA DataFrameIndex)_compression_to_extension) get_handle)read_xml)squarecircletriangle)hrshapedegreessides)MercuryVenusEarthMarsJupiterSaturnUranusNeptune) terrestrialrrr gas giantr ice giantr )innerr!r!r!outerr"r"r")g`|x ?gc AJx@g2r@gl?gh|@g @gd]FU@g hY@)planettypelocationmassac 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|dur |dd}|S)Nz%|dddd}t||d}|j|d}t|}|tksJdS)Nr=r>r?r@rA)r rGr+rLrNr:rOrPrSr)r)r*test_str_outputs   rXcCsFd}tjtddtj||dWddS1swYdS)Nz/my/fake/path/output.xmlz6Cannot save file into a non-existent directory: .*pathmatchrA)pytestraisesOSErrorgeom_dfrGr:rQr)r)r*test_wrong_file_paths"r`c Csd}|dddd}t||d}td9}|j|d|d t|d }|d }Wdn1s6wYt|}||ksEJWddS1sPwYdS) 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@rArBFindexr:r,rC r rErFrGrHrIrJrKr+rNr:expectedrOrPrQrRrSr)r)r*test_index_falses   "rfc Csd}|dddd}t||d}td;}|j|dd d |d t|d }|d }Wdn1s8wYt|}||ksGJWddS1sRwYdS)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@rArBFbooksbook)rb root_namerow_namer:r,rCrcrdr)r)r* test_index_false_rename_row_roots    "rk offset_index cCsg|]}t|qSr))str).0ir)r)r* .srrcCs<d}t}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 FraN)r^copyrrbrGr+)r:rlreoffset_geom_dfrSr)r)r*(test_index_false_with_offset_input_index-s   rua 0 square 360 4.0 1 circle 360 2 triangle 180 3.0 cCs$tj|d}t|}|tksJdS)NrAr^rGr+ na_expectedr:rSr)r)r*test_na_elem_outputns rycCs&tjd|d}t|}|tksJdS)Nna_repr:rvrxr)r)r*test_na_empty_str_elem_optionusr}cCs*d}tjd|d}t|}||ksJdS)Na 0 square 360 4.0 1 circle 360 0.0 2 triangle 180 3.0 z0.0r{r^rGr+r:rerSr)r)r*test_na_empty_elem_option|srcCs.d}tjgd|d}t|}||ksJdS)Nz r attr_colsr:r~rr)r)r*test_attrs_cols_nan_outputsrcCs6d}tjgdddid|d}t|}||ksJdS)Na> )rbrrrr(zhttp://example.xom)r namespacesprefixr:r~rr)r)r*test_attrs_cols_prefixs rcCFtjtddtjgd|dWddS1swYdS)Nno valid columnrYrdegreerrr[r\KeyErrorr^rGrAr)r)r*test_attrs_unknown_column"rcCBtjtddtjd|dWddS1swYdS)Nz!is not a valid type for attr_colsrY"shape", "degree", "sides"rr[r\ TypeErrorr^rGrAr)r)r*test_attrs_wrong_type"rcCs0d}tjdgd|d}t|}||ksJdS)NaD 360 4.0 square 360 circle 180 3.0 triangle F)rrr)rb elem_colsr:r~r:elems_cols_expectedrSr)r)r*test_elems_cols_nan_outputs  rcCr)NrrYrrr:rrAr)r)r*test_elems_unknown_columnrrcCr)Nz!is not a valid type for elem_colsrYrrrrAr)r)r*test_elems_wrong_typerrcCs4d}tjdddgdg|d}t|}||ksJdS)Na# 360 4.0 360 180 3.0 Frrr)rbrrr:r~rr)r)r*test_elems_and_attrs_colssrcCsHd}tjddgdgdddd}|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&countsummeanTrbvaluesaggfuncmarginsrA) planet_df pivot_tableroundrGr+r:repvtrSr)r)r*test_hierarchical_columnss! rcCsVd}tjddgdgdddd}|jt|jj|d }t|}||ks)JdS) Na r%r$r&rTrrr) rrrrGlist reset_indexcolumnsrr+rr)r)r*test_hierarchical_attrs_columnsFs rcCsJd}tddgdgdd}|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)rgroupbyaggrrGr+r:rerrSr)r)r*test_multi_indexds  rcCsXd}tddgdgdd}|jt|jj|d}t |}||ks*JdS)Na$ r%r$r&rrr) rrrrrGrrrrr+rr)r)r*test_multi_index_attrs_colss rcCs.d}tjddi|d}t|}||ksJdS)Na 0 square 360 4.0 1 circle 360 2 triangle 180 3.0 rzhttp://example.com)rr:r~rr)r)r*test_default_namespacesrcCs0d}tjddid|d}t|}||ksJdS)Na 0 square 360 4.0 1 circle 360 2 triangle 180 3.0 r(rrrr:r~rr)r)r*test_namespace_prefixs  rcCsHtjtddtjddid|dWddS1swYdS)Nz!doc is not included in namespacesrYrzrr(rrrAr)r)r*test_missing_prefix_in_nmsps  "rcCsFd}tjdddd|d}t|}|dur|dd}||ks!JdS) Na7 0 square 360 4.0 1 circle 360 2 triangle 180 3.0 rzhttp://other.org)rzr(r(rz7xmlns:doc="http://other.org" xmlns="http://example.com"z7xmlns="http://example.com" xmlns:doc="http://other.org"r^rGr+r'rr)r)r*!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}|dur"|d d }|tks(JdS) Nr=r>r?baby_names.xml ISO-8859-1)r:encodingrr:z*r?rrr7rrBF)rbrr:r rErFrG)rNrOrPrQr)r)r*test_correct_encoding_fileMs  "rr)zUTF-8zUTF-16rcCs^|dddd}t|ddd}td}|j|d ||d WddS1s(wYdSrr)rNr:rrOrPrQr)r)r*test_wrong_encoding_option_lxmlVs  "rcCr)Nzunknown encodingrYzuft-8r)r[r\ LookupErrorr^rGrAr)r)r*test_misspelled_encoding`rrcCs d}tjdd}||ksJdS)Na\ 0 square 360 4.0 1 circle 360 2 triangle 180 3.0 F)xml_declaration)r^rG)rerSr)r)r*!test_xml_declaration_pretty_prinths rcCs>d}tjd|d}t|}|dur|dd}||ksJdS)Na4 0square3604.01circle3602triangle1803.0F) pretty_printr: />/>rrr)r)r*test_no_pretty_print_with_decls  rcCs8d}tjdd|d}|dur|dd}||ksJdS)Na 0square3604.01circle3602triangle1803.0F)rrr:rr)r^rGr'rr)r)r*test_no_pretty_print_no_decls   rcCs<tjtdd tWddS1swYdS)Nz7lxml not found, please install or use the etree parser.rY)r[r\ ImportErrorr^rGr)r)r)r*test_default_parser_no_lxmls  "rcCs@tjtddtjddWddS1swYdS)Nz,Values for parser can only be lxml or etree.rYbs4rAr[r\ ValueErrorr^rGr)r)r)r*test_unknown_parsers "ra% 0 square 360 4.0 1 circle 360 2 triangle 180 3.0 cCsR|dddd}t||}tj|dtksJWddS1s"wYdSNr=r>r?row_field_output.xsl stylesheet)rHr^rG xsl_expected)rNr5xslrRr)r)r*test_stylesheet_file_likes "rcCsv|dddd}t||}|dkrt|}nt|}Wdn1s(wYtj|d}|tks9JdS)Nr=r>r?rr,r)rHrrIrr^rGr)rNr5xsl_pathrRxsl_objrSr)r)r*test_stylesheet_ios   rcCs\|dddd}t|| }|}Wdn1swYtj|d}|tks,JdSr)rHrIr^rGr)rNr5rrRrrSr)r)r*test_stylesheet_buffered_readers    rcCs\ddlm}tjddd}tj|ddtj|dWddS1s'wYdS) NrXMLSyntaxErrorr>r?zrow_field_output.xslt!Start tag expected, '<' not foundrYr) lxml.etreerosrQjoinr[r\r^rGrrr)r)r*test_stylesheet_wrong_paths "rvalrzcCsZddlm}dgd}tj||dtj|dWddS1s&wYdS)Nrr|)zDocument is emptyrzNone \(line 0\)rYr)rrrr[r\r^rG)rrmsgr)r)r*test_empty_string_stylesheet s  "rcCPddlm}d}tj|ddtj|dWddS1s!wYdS)NrraR zOpening and ending tag mismatchrYr)rrr[r\r^rGrr)r)r*test_incorrect_xsl_syntax "rcCr)Nr)XSLTParseErroraT zfailed to compilerYr)rrr[r\r^rG)rrr)r)r*test_incorrect_xsl_eval:rrc Csddlm}d}tj|dd-td}tj||dWdn1s'wYWddSWddS1s?wYdS)Nr)XSLTApplyErrorag zCannot resolve URIrYrBr)rrr[r\rErFr^rG)rrrQr)r)r*test_incorrect_xsl_applyWs  "rcCsFd}tjtddtjd|dWddS1swYdS)NaI z*To use stylesheet, you need lxml installedrYr9)r:rr)rr)r)r*test_stylesheet_with_etreels  "rcCs<d}tjdd}|dur|}tj|d}||ksJdS)NaQ , ,shape,degrees,sides  )lineterminatorr)r^to_csvrKrG)rout_csvout_xmlr)r)r*test_style_to_csv~s   rcCs*d}t}tjd|d}||ksJdS)Na shape degrees sides NaN)r|r)r^ to_stringrG)rout_strrr)r)r*test_style_to_stringsrcCs(d}t}tj|d}||ksJdS)Na] " {"shape":{ },"degrees":{ },"sides":{ }} , r)r^to_jsonrG)rout_jsonrr)r)r*test_style_to_jsons% rc Cst,}tj|||dt|d|d }|j}Wdn1s$wYWdn1s3wYt|}t| ksDJdS)Nr: compressionr-r) rErFr^rGr handlerIr+geom_xmlrK)r:compression_onlyrQ handle_objrSr)r)r*test_compression_outputs   r c Csdt|}tj|d,}tj|||dt|d|d }|j}Wdn1s,wYWdn1s;wYt|}t | ksLJdS)Nzxml.)rOrr-r) rrErFr^rGr rrIr+rrK)r:r compfilerQr rSr)r)r*test_filename_and_suffix_comps   r cCs<d}tdtgi|}|j|d}t||ksJdS)Nzd 0 arA)rrastyperGr+rK)any_numeric_ea_dtyper:redfresultr)r)r*test_ea_dtypes&s rc Cs|tjtdd-t}tj||ddWdn1swYWddSWddS1s7wYdS)NzUnrecognized compression typerY7zr)r[r\rrErFr^rGr_r)r)r*test_unsuported_compression4s "rs3fscCsbddl}tjtdd|jdd}|dtjdd|d WddS1s*wYdS) Nrz Access DeniedrYT)anonz pandas-testzs3://pandas-test/geom.xmlzip)rr:)rr[r\PermissionError S3FileSystemlsr^rG)r: s3_resourcerfsr)r)r*test_s3_permission_output=s   "r)b __future__rr=rrrnumpynpr[pandas.util._test_decoratorsutil_test_decoratorstdpandasrrrpandas._testing_testingrE pandas.tests.io.test_compressionrpandas.io.commonr pandas.io.xmlr nanr^rrLr+fixturer5r2 skip_if_nor:rTrVrXr`rfrkmark parametrizerrangerurwryr}rrrrrrrrrrrrrrrrrrrrrrrrrskip_if_installedrrrrrrrrrrrrrrrrr r rr single_cpurr)r)r)r*s     $    -    (*" &! /'! +$                ! /