o
    Ti                     @  s  d dl mZ d dlmZmZ d dlmZ d dlZd dlm	Z	 d dl
mZ d dlmZ d dlmZ d dlZd dlZd d	lmZ d d
lmZmZ d dlm  mZ d dlZd dlmZmZm Z  d dl!m"Z# d dl$m%Z%m&Z& d dl'm(Z( d dl)m*Z* eg dg ddej+dgdZ,dZ-dZ.eddddddddddddddddddd d d d d dddddddd d!d"d#d$dd%Z/d&d' Z0ej1d(d)gd*d+d, Z2ej1ej3d-e4d-d.d/gd*d0d1 Z5d2d3 Z6d4d5 Z7d6d7 Z8ej9j:ej9j;d8d9 Z<d:d; Z=d<d= Z>d>d? Z?d@dA Z@dBdC ZAdDdE ZBdFdG ZCej9DdHdIdJgdKdL ZEej9DdHdIdJgdMdN ZFdOdP ZGej9j:ej9j;dQdR ZHej9j:ej9j;dSdT ZIdUdV ZJdWdX ZKdYdZ ZLd[d\ ZMd]d^ ZNd_d` ZOdadb ZPdcdd ZQdedf ZRdgdh ZSdidj ZTej9DdkdIdgdldm ZUdndo ZVdpdq ZWdrds ZXdtdu ZYdvdw ZZdxdy Z[dzd{ Z\d|d} Z]d~d Z^dd Z_dd Z`dd Zadd Zbdd Zcdd Zddd Zedd Zfdd Zgdd Zhdd Ziejd-dd Zkdd Zldd Zmdd Zndd Zodd Zpdd Zqdd Zrdd Zsdd Ztdd Zudd Zvdd Zwdd Zxej9DdHdIdJgdd Zydd Zzdd Z{ej9j:ej9j;dd Z|dd Z}dd Z~dd Zdd ZddÄ Zddń ZddǄ ZddɄ Zdd˄ Zdd̈́ Zddτ Zddф Zddӄ Zej9j:ej9j;ddՄ Zddׄ Zddل ZdS )    )annotations)BytesIOStringIO)	LZMAErrorN)	ReadError)	HTTPError)
ParseError)
BadZipFile)import_optional_dependency)EmptyDataErrorParserError)NA	DataFrameSeries)ArrowStringArrayStringArray)
get_handle)read_xmlsquarecircletriangleh  r            shapedegreessidesa[  <?xml version='1.0' encoding='utf-8'?>
<data xmlns="http://example.com">
  <row>
    <shape>square</shape>
    <degrees>360</degrees>
    <sides>4</sides>
  </row>
  <row>
    <shape>circle</shape>
    <degrees>360</degrees>
    <sides/>
  </row>
  <row>
    <shape>triangle</shape>
    <degrees>180</degrees>
    <sides>3</sides>
  </row>
</data>a  <?xml version='1.0' encoding='utf-8'?>
<doc:data xmlns:doc="http://example.com">
  <doc:row>
    <doc:shape>square</doc:shape>
    <doc:degrees>360</doc:degrees>
    <doc:sides>4.0</doc:sides>
  </doc:row>
  <doc:row>
    <doc:shape>circle</doc:shape>
    <doc:degrees>360</doc:degrees>
    <doc:sides/>
  </doc:row>
  <doc:row>
    <doc:shape>triangle</doc:shape>
    <doc:degrees>180</doc:degrees>
    <doc:sides>3.0</doc:sides>
  </doc:row>
</doc:data>ZID_00001ZID_00002ZID_00003ZID_00004ZID_00005r         r   r   zBlue Line (Forest Park)zRed, Purple Linez#LineStyle01ZclampedToGroundz-87.77678526964958,41.8708863930319,0 -87.77826234150609,41.87097820122218,0 -87.78251583439344,41.87130129991005,0 -87.78418294588424,41.87145055520308,0 -87.7872369165933,41.8717239119163,0 -87.79160214925886,41.87210797280065,0a\  -87.65758750947528,41.96427269188822,0 -87.65802133507393,41.96581929055245,0 -87.65819033925305,41.96621846093642,0 -87.6583189819129,41.96650362897086,0 -87.65835858701473,41.96669002089185,0 -87.65838428411853,41.96688150295095,0 -87.65842208882658,41.96745896091846,0 -87.65846556843937,41.9683761425439,0 -87.65849296214573,41.96913893870342,0a  -87.65492939166126,41.95377494531437,0 -87.65557043199591,41.95376544118533,0 -87.65606302030132,41.95376391658746,0 -87.65623502146268,41.95377379126367,0 -87.65634748981634,41.95380103566435,0 -87.65646537904269,41.95387703994676,0 -87.65656532461145,41.95396622645799,0 -87.65664760856414,41.95404201996044,0 -87.65671750555913,41.95416647054043,0 -87.65673983607117,41.95429949810849,0 -87.65673866475777,41.95441024240925,0 -87.6567690255541,41.95490657227902,0 -87.65683672482363,41.95692259283837,0 -87.6568900886376,41.95861070983142,0 -87.65699865558875,41.96181418669004,0 -87.65756347177603,41.96397045777844,0 -87.65758750947528,41.96427269188822,0ah  -87.65362593118043,41.94742799535678,0 -87.65363554415794,41.94819886386848,0 -87.6536456393239,41.95059994675451,0 -87.65365831235026,41.95108288489359,0 -87.6536604873874,41.9519954657554,0 -87.65362592053201,41.95245597302328,0 -87.65367158496069,41.95311153649393,0 -87.65368468595476,41.9533202828916,0 -87.65369271253692,41.95343095587119,0 -87.65373335834569,41.95351536301472,0 -87.65378605844126,41.95358212680591,0 -87.65385067928185,41.95364452823767,0 -87.6539390793817,41.95370263886964,0 -87.6540786298351,41.95373403675265,0 -87.65430648647626,41.9537535411832,0 -87.65492939166126,41.95377494531437,0a   -87.65345391792157,41.94217681262115,0 -87.65342448305786,41.94237224420864,0 -87.65339745703922,41.94268217746244,0 -87.65337753982941,41.94288140770284,0 -87.65336256753105,41.94317369618263,0 -87.65338799707138,41.94357253961736,0 -87.65340240886648,41.94389158188269,0 -87.65341837392448,41.94406444407721,0 -87.65342275247338,41.94421065714904,0 -87.65347469646018,41.94434829382345,0 -87.65351486483024,41.94447699917548,0 -87.65353483605053,41.9453896864472,0 -87.65361975532807,41.94689193720703,0 -87.65362593118043,41.94742799535678,0idnameZstyleUrlZextrudeZaltitudeModeZcoordinatesc                  C  sJ   t d d} tjt| d tt W d    d S 1 sw   Y  d S )NlxmlPassing literal xml to 'read_xml' is deprecated and will be removed in a future version. To read from a literal string, wrap it in a 'StringIO' object.match)pytestimportorskiptmZassert_produces_warningFutureWarningr   xml_default_nmspmsg r2   \/var/www/html/evchargy.com/venv/lib/python3.10/site-packages/pandas/tests/io/xml/test_xml.pytest_literal_xml_deprecation   s   

"r4   rbr)paramsc                 C     | j S Nparamrequestr2   r2   r3   mode     r>   r'   )Zmarksetreec                 C  r8   r9   r:   r<   r2   r2   r3   parser  r?   rA   c              	   K  st   t  ,}t|ddd}||  W d    n1 sw   Y  t|fi |W  d    S 1 s3w   Y  d S )Nwutf-8encoding)r-   ensure_cleanopenwriter   )datakwargspathfr2   r2   r3   read_xml_iterparse  s   
$rM   c              
   K  s   t | d|dK}t 6}t|ddd}||j  W d    n1 s'w   Y  t|fi |W  d    W  d    S 1 sEw   Y  W d    d S 1 sUw   Y  d S )Nr6   )compressionrB   rC   rD   )r   r-   rF   rG   rH   handlereadr   )	comp_pathcompression_onlyrJ   ZhandlesrK   rL   r2   r2   r3   read_xml_iterparse_comp  s   
"rS   c                 C  sv   t d t| dd}t| dd}t| ddg did}t| ddg did}t|| t|| t|| d S )Nr'   rA   r@   bookcategorytitleyearauthorpricerA   	iterparser+   r,   r   r-   assert_frame_equal)	xml_booksZdf_file_lxmlZdf_file_etreedf_iter_lxmldf_iter_etreer2   r2   r3   test_parser_consistency_file"  s    


rc   c                 C  sJ   |j td ttt| d}ttt | dg did}t|| d S )NcontentrT   rowr   r\   )serve_contentr/   r   r   r   encoder-   r_   )rA   
httpserverdf_xpathdf_iterr2   r2   r3   test_parser_consistency_url7  s   

rl   c                 C  s~   t | ||dkr	dnd d}t||d}W d    n1 sw   Y  tg dg dg dg dg d	d
}t|| d S Nr6   rC   rD   rT   cookingchildrenZwebzEveryday ItalianzHarry PotterzLearning XMLzGiada De LaurentiiszJ K. RowlingzErik T. Ray  rt   i  g      >@g=
ףp=@gC@rW   rX   rZ   rY   r[   )rG   r   r   r-   r_   )r`   rA   r>   rL   df_filedf_expectedr2   r2   r3   test_file_likeF  s   
ry   c                 C  s   t | ||dkr	dnd d}| }W d    n1 sw   Y  tt|tr+t|nt||d}tg dg dg dg dg d	d
}t	|| d S rm   )
rG   rP   r   
isinstancebytesr   r   r   r-   r_   )r`   rA   r>   rL   xml_objZdf_iorx   r2   r2   r3   test_file_ioW  s    

r}   c                 C  s   t | ||dkr	dnd d}| }W d    n1 sw   Y  |dkr,t| }n|dkr4t|}t||d}tg dg dg dg d	g d
d}t|| d S Nr6   rC   rD   r5   rT   rn   rq   rr   rs   ru   rv   )rG   rP   r   decoder   r   r-   r_   r`   rA   r>   rL   r|   df_strrx   r2   r2   r3    test_file_buffered_reader_stringm  s"   

r   c                 C  s   t | ||dkr	dnd d}t| | }W d    n1 s w   Y  |dkr0t| }n|dkr8t|}t||d}tg dg dg dg d	g d
d}t|| d S r~   )	rG   nextrP   r   r   r   r   r-   r_   r   r2   r2   r3   ,test_file_buffered_reader_no_xml_declaration  s$   

r   c                 C  s8   d}t t|| d}tddddgd}t|| d S )N<   <中文標籤><row><c1>1</c1><c2>2</c2></row></中文標籤>rT   r"   r#   )c1c2r   )indexr   r   r   r-   r_   )rA   txtr   rx   r2   r2   r3   test_string_charset  s   r   c                 C  sD   t | |d}tg ddddgd dgg dd}t|| d S )	NrT   )uP   問  若箇是邪而言破邪 何者是正而道(Sorry, this is Big5 only)申正u;   問 既破有得申無得 亦應但破性執申假名以不uO   問 既破性申假 亦應但破有申無 若有無兩洗 亦應性假雙破耶 u<   答  邪既無量 正亦多途  大略為言不出二種 謂u;   有得與無得 有得是邪須破 無得是正須申
		故uL   答  不例  有無皆是性 所以須雙破 既分性假異 故有破不破)NuV   答 性執是有得 假名是無得  今破有得申無得 即是破性執申假名也N)u   問u   答a)r   r   joinr-   r_   )Zxml_doc_ch_utfrA   rw   rx   r2   r2   r3   test_file_charset  s   
r   c                 C  sN   t | d}tt| |d |jrJ W d    d S 1 s w   Y  d S )Nr5   rT   )rG   r   r   rP   closed)r`   rA   rL   r2   r2   r3   test_file_handle_close  s   "r   valr       c                 C  s   t d}dddg}t j|j|d' t| tr#tt| dd ntt	| dd W d    d S W d    d S 1 s>w   Y  d S )N
lxml.etree|zDocument is emptyzNone \(line 0\)r)   r'   rT   )
r+   r,   r   raisesXMLSyntaxErrorrz   strr   r   r   )r   
lxml_etreer1   r2   r2   r3   test_empty_string_lxml  s   

"r   c                 C  sp   t jtdd' t| trtt| dd ntt| dd W d    d S W d    d S 1 s1w   Y  d S )Nzno element foundr)   r@   rT   )r+   r   r   rz   r   r   r   r   )r   r2   r2   r3   test_empty_string_etree  s   
"r   c                 C  sT   d}t jddd}tjt|d t|| d W d    d S 1 s#w   Y  d S )Nr(   rI   htmlz	books.xmlr)   rT   )osrK   r   r+   r   r.   r   )rA   r1   filenamer2   r2   r3   test_wrong_file_path  s   "r   c                 C  s   t d t|dd}| j| d t| jdd}W d    n1 s%w   Y  tg dg dg d	g d
g dd}t	|| d S )Nr'   rC   rD   rd   .//book[count(*)=4]xpathrn   rq   rr   rs   ru   rv   )
r+   r,   rG   rg   rP   r   urlr   r-   r_   )ri   xml_filerL   Zdf_urlrx   r2   r2   r3   test_url  s   

r   c                 C  sR   |j ddd tjtdd t|jd| d W d    d S 1 s"w   Y  d S )Nz	NOT FOUNDi  )codezHTTP Error 404: NOT FOUNDr)   r   r   rA   )rg   r+   r   r   r   r   )rA   ri   r2   r2   r3   test_wrong_url  s   "r   c                 C  sj   d}t t|| dd}t|| dg didd}tg dg dg d	ddd
}t|| t|| d S )Na  
      <data>
        <row sides=" 4 ">
          <shape>
              square
          </shape>
          <degrees>&#009;360&#009;</degrees>
        </row>
        <row sides=" 0 ">
          <shape>
              circle
          </shape>
          <degrees>&#009;360&#009;</degrees>
        </row>
        <row sides=" 3 ">
          <shape>
              triangle
          </shape>
          <degrees>&#009;180&#009;</degrees>
        </row>
      </data>string)rA   dtyperf   )r    r   r   )rA   r]   r   )z 4 z 0 z 3 )z 
              square
          z 
              circle
          z"
              triangle
          )	360	r   z	180	r   r   r   rM   r   r-   r_   rA   xmlrj   rk   rx   r2   r2   r3   test_whitespace  s"   
	r   c                 C  sL   t d t jtdd t| ddd W d    d S 1 sw   Y  d S )Nr'   xpath does not return any nodesr)   z	.//pythonr   r+   r,   r   
ValueErrorr   r`   r2   r2   r3   test_empty_xpath_lxmlL  s   
"r   c                 C  B   t jtdd t| ddd W d    d S 1 sw   Y  d S )Nz/You have used an incorrect or unsupported XPathr)   	.//[book]r@   r   r+   r   SyntaxErrorr   r   r2   r2   r3   test_bad_xpath_etreeR  s
   "r   c                 C  N   t d}t j|jdd t| ddd W d    d S 1 s w   Y  d S )Nr   zInvalid expressionr)   r   r'   r   r+   r,   r   ZXPathEvalErrorr   )r`   r   r2   r2   r3   test_bad_xpath_lxmlY     
"r   c                 C  l   t ttdddi| d}tt| dg did}tg dg d	d
tddgd}t|| t|| d S )N	.//ns:rownshttp://example.comr   
namespacesrA   rf   r   r\   r   r         @nan      @)r   r   r/   rM   r   floatr-   r_   rA   Zdf_nmsprk   rx   r2   r2   r3   test_default_namespacec  s&   
r   c                 C  r   )N
.//doc:rowdocr   r   rf   r   r\   r   r   r   r   r   )r   r   xml_prefix_nmsprM   r   r   r-   r_   r   r2   r2   r3   test_prefix_namespace}  s"   r   c                  C  sJ   t d tttdddidd} tttdddidd}t| | d S )	Nr'   r   r   r   r   r   r   r@   )r+   r,   r   r   r/   r-   r_   df_lxmldf_etreer2   r2   r3   "test_consistency_default_namespace     
r   c                  C  sJ   t d tttdddidd} tttdddidd}t| | d S )Nr'   r   r   r   r   r@   )r+   r,   r   r   r   r-   r_   r   r2   r2   r3   !test_consistency_prefix_namespace  r   r   c                 C  B   t jtdd t| d|d W d    d S 1 sw   Y  d S )Nr   r)   z.//Placemarkr   r+   r   r   r   r`   rA   r2   r2   r3   *test_missing_prefix_with_default_namespace     "r   c                 C  r   )Nz'you used an undeclared namespace prefixr)   .//kml:Placemarkr@   r   r   )kml_cta_rail_linesr2   r2   r3   $test_missing_prefix_definition_etree  r   r   c                 C  r   )Nr   zUndefined namespace prefixr)   r   r'   r   r   )r   r   r2   r2   r3   #test_missing_prefix_definition_lxml  r   r   keyc                 C  sV   t d t jtdd tttd| didd W d    d S 1 s$w   Y  d S )Nr'   z0empty namespace prefix is not supported in XPathr)   r   http://www.opengis.net/kml/2.2r   )r+   r,   r   	TypeErrorr   r   r/   )r   r2   r2   r3   test_none_namespace_prefix  s   
"r   c                 C  sf   t | |d}t | |dg did}tg dg dg dg dg d	d}t|| t|| d S )
NrT   rU   rv   r\   rn   rq   rr   rs   ru   r   r   r-   r_   r`   rA   rw   rk   rx   r2   r2   r3   test_file_elems_and_attrs  s    

r   c                 C  sN   t | d|d}t | |ddgid}tdg di}t|| t|| d S )NT)
attrs_onlyrA   rU   rW   r\   rn   r   r   r2   r2   r3   test_file_only_attrs  s
   r   c                 C  sb   t | d|d}t | |dg did}tg dg dg dg d	d}t|| t|| d S )
NT)
elems_onlyrA   rU   )rX   rZ   rY   r[   r\   rq   rr   rs   ru   r   r   r2   r2   r3   test_file_only_elems  s   
	r   c                 C  sD   t jtdd t| dd|d W d    d S 1 sw   Y  d S )Nz3Either element or attributes can be parsed not bothr)   T)r   r   rA   r   )r   rA   r2   r2   r3   test_elem_and_attrs_only  s   "r   c                 C  L   d}t jtdd tt|dd| d W d    d S 1 sw   Y  d S )NaU  
      <data>
        <row>
          <shape sides="4">square</shape>
          <degrees>360</degrees>
        </row>
        <row>
          <shape sides="0">circle</shape>
          <degrees>360</degrees>
        </row>
        <row>
          <shape sides="3">triangle</shape>
          <degrees>180</degrees>
        </row>
      </data>-xpath does not return any nodes or attributesr)   ./rowT)r   r   rA   r+   r   r   r   r   rA   r   r2   r2   r3   test_empty_attrs_only  s   "r   c                 C  r   )Nz
      <data>
        <row sides="4" shape="square" degrees="360"/>
        <row sides="0" shape="circle" degrees="360"/>
        <row sides="3" shape="triangle" degrees="180"/>
      </data>r   r)   r   T)r   r   rA   r   r   r2   r2   r3   test_empty_elems_only1  s   "r   c                  C  sv   t d d} tt| dd}tt| ddd}t| ddd	gid
}t| dddd	gid}t|| t|| d S )Nr'   a  <?xml version="1.0" encoding="UTF-8"?>
<TrainSchedule>
      <Stations>
         <station Name="Manhattan" coords="31,460,195,498"/>
         <station Name="Laraway Road" coords="63,409,194,455"/>
         <station Name="179th St (Orland Park)" coords="0,364,110,395"/>
         <station Name="153rd St (Orland Park)" coords="7,333,113,362"/>
         <station Name="143rd St (Orland Park)" coords="17,297,115,330"/>
         <station Name="Palos Park" coords="128,281,239,303"/>
         <station Name="Palos Heights" coords="148,257,283,279"/>
         <station Name="Worth" coords="170,230,248,255"/>
         <station Name="Chicago Ridge" coords="70,187,208,214"/>
         <station Name="Oak Lawn" coords="166,159,266,185"/>
         <station Name="Ashburn" coords="197,133,336,157"/>
         <station Name="Wrightwood" coords="219,106,340,133"/>
         <station Name="Chicago Union Sta" coords="220,0,360,43"/>
      </Stations>
</TrainSchedule>z
.//stationr   r@   r   stationNameZcoordsr]   r\   )r+   r,   r   r   rM   r-   r_   )r   r   r   Z
df_iter_lxZ
df_iter_etr2   r2   r3   test_attribute_centric_xml@  s   
r   c                 C  sr   t | g d|d}t | |g ddg did}tg dg dg dg d	g d
d}t|| t|| d S )N)Col1Col2Col3ZCol4ZCol5namesrA   rU   rv   )rA   r   r]   rn   rq   rr   rs   ru   r   r   r2   r2   r3   test_names_option_outpute  s&   


r   c                 C  sp   d}t t|d| g dd}t|| dg dig dd}tdd	gd
dgddgd}t|| t|| d S )Nz<shapes>
  <shape type="2D">
    <name>circle</name>
    <type>curved</type>
  </shape>
  <shape type="3D">
    <name>sphere</name>
    <type>curved</type>
  </shape>
</shapes>.//shape)Ztype_dimr   Z	type_edger   rA   r   r   )typer&   r   rA   r]   r   2D3Dr   sphereZcurvedr   r   r2   r2   r3   test_repeat_names~  s*   
r  c                 C  sj   d}t t|d| ddgd}t|| dddgiddgd}tg d	g d
d}t|| t|| d S )NaB  <shapes>
  <shape>
    <name>rectangle</name>
    <family>rectangle</family>
  </shape>
  <shape>
    <name>square</name>
    <family>rectangle</family>
  </shape>
  <shape>
    <name>ellipse</name>
    <family>ellipse</family>
  </shape>
  <shape>
    <name>circle</name>
    <family>ellipse</family>
  </shape>
</shapes>r   r&   groupr   r   familyr   )	rectangler   ellipser   )r  r  r  r  )r&   r  r   r   r2   r2   r3   test_repeat_values_new_names  s"   
r  c                 C  sv   d}t t|d| g dd}t|| dg dig dd}tg dg d	g d
g dd}t|| t|| d S )Na  <shapes>
  <shape>
    <value item="name">circle</value>
    <value item="family">ellipse</value>
    <value item="degrees">360</value>
    <value item="sides">0</value>
  </shape>
  <shape>
    <value item="name">triangle</value>
    <value item="family">polygon</value>
    <value item="degrees">180</value>
    <value item="sides">3</value>
  </shape>
  <shape>
    <value item="name">square</value>
    <value item="family">polygon</value>
    <value item="degrees">360</value>
    <value item="sides">4</value>
  </shape>
</shapes>r   )r&   r  r   r    r   r   )valuer  r  r  r   )r   r   r   )r  polygonr	  )r   r   r   )r   r   r   r   r   r2   r2   r3   test_repeat_elements  s,   
	r
  c                 C  sF   t jtdd t| g d|d W d    d S 1 sw   Y  d S )Nznames does not match lengthr)   )r   r   r   r   r   r   r2   r2   r3   test_names_option_wrong_length  s   "r  c                 C  r   )Nzis not a valid type for namesr)   zCol1, Col2, Col3r   r+   r   r   r   r   r2   r2   r3   test_names_option_wrong_type  r   r  c                 C  s@   t jtdd t| |d W d    d S 1 sw   Y  d S )Nz'utf-8' codec can't decoder)   rT   r+   r   UnicodeDecodeErrorr   xml_baby_namesrA   r2   r2   r3   test_wrong_encoding  s   "r  c                 C  r   )NzIUTF-16 stream does not start with BOM|'utf-16-le' codec can't decode byter)   zUTF-16rE   rA   )r+   r   UnicodeErrorr   r  r2   r2   r3   test_utf16_encoding  s   "r  c                 C  r   )Nzunknown encoding: UFT-8r)   zUFT-8r  )r+   r   LookupErrorr   r  r2   r2   r3   test_unknown_encoding  r   r  c                 C  r   )Nz'ascii' codec can't decode byter)   asciir  r  r  r2   r2   r3   test_ascii_encoding!  r   r  c                 C  s~   t d t| ddd}t| ddd}t| dddg did}t| dddg did}t|| t|| t|| d S )	Nr'   z
ISO-8859-1rA   rE   r@   z
iso-8859-1rf   )ZrankZmalenameZ
femalename)rA   rE   r]   r^   )r  Zdf_xpath_lxmlZdf_xpath_etreera   rb   r2   r2   r3   %test_parser_consistency_with_encoding&  s$   


r  c                  C  sT   t d d} t jtdd tt| dd d W d    d S 1 s#w   Y  d S )Nr'   -<data>
  <row>
    <a>c</a>
  </row>
</data>
zencoding Noner)   r  )r+   r,   r   r   r   r   )rI   r2   r2   r3   test_wrong_encoding_for_lxml=  s
   
"r  c                  C  s4   d} t t| dd d}tddgi}t|| d S )Nr  r@   r  r   cr   )rI   resultexpectedr2   r2   r3   test_none_encoding_etreeJ  s   r!  c                 C  s<   t jtdd t|  W d    d S 1 sw   Y  d S )Nz7lxml not found, please install or use the etree parser.r)   )r+   r   ImportErrorr   r   r2   r2   r3   test_default_parser_no_lxmlZ  s
   
"r#  c                 C  s@   t jtdd t| dd W d    d S 1 sw   Y  d S )Nz,Values for parser can only be lxml or etree.r)   Zbs4rT   r   r   r2   r2   r3   test_wrong_parserb  s
   "r$  c                 C  sN   t d t| dddi|d}t| dg did}tt| tt| d S )	Nr'   .//k:Placemarkkr   r   r   
stylesheetZ	Placemarkr$   r   )r+   r,   r   r-   r_   df_kml)r   xsl_flatten_docdf_stylerk   r2   r2   r3   test_stylesheet_filel  s   
r,  c                 C  sh   t d t|||dkrdnd d}t| dddi|d}W d    n1 s'w   Y  tt| d S 	Nr'   r6   rC   rD   r%  r&  r   r'  )r+   r,   rG   r   r-   r_   r)  )r   r*  r>   rL   r+  r2   r2   r3   test_stylesheet_file_like  s   
r.  c                 C  s   t d t|||dkrdnd d}|dkrt| }nt| }W d    n1 s.w   Y  t| dddi|d	}tt	| d S )
Nr'   r6   rC   rD   r5   r%  r&  r   r'  )
r+   r,   rG   r   rP   r   r   r-   r_   r)  r   r*  r>   rL   xsl_objr+  r2   r2   r3   test_stylesheet_io  s   
r1  c                 C  sp   t d t|||dkrdnd d}| }W d    n1 s!w   Y  t| dddi|d}tt| d S r-  )r+   r,   rG   rP   r   r-   r_   r)  r/  r2   r2   r3   test_stylesheet_buffered_reader  s   

r2  c                  C  s>   t d d} d}tt| }tt| |d}t|| d S )Nr'   r   u  <xsl:stylesheet version="1.0" xmlns:xsl="http://www.w3.org/1999/XSL/Transform">
 <xsl:output omit-xml-declaration="yes" indent="yes"/>
 <xsl:strip-space elements="*"/>

 <xsl:template match="node()|@*">
     <xsl:copy>
       <xsl:apply-templates select="node()|@*"/>
     </xsl:copy>
 </xsl:template>

 <xsl:template match="中文標籤">
     <根>
       <xsl:apply-templates />
     </根>
 </xsl:template>

</xsl:stylesheet>r(  )r+   r,   r   r   r-   r_   )r   xslZdf_origr+  r2   r2   r3   test_style_charset  s   
r5  c                 C  sL   t d}t j|jdd t| |d W d    d S 1 sw   Y  d S )Nr   zdocument is not a stylesheetr)   r3  r+   r,   r   ZXSLTParseErrorr   )r   r`   r   r2   r2   r3   test_not_stylesheet  s   
"r7  c                 C  P   t d}d}t j|jdd t| |d W d    d S 1 s!w   Y  d S )Nr   a  <xsl:stylesheet version="1.0" xmlns:xsl="http://www.w3.org/1999/XSL/Transform"
                              xmlns:k="http://www.opengis.net/kml/2.2"/>
    <xsl:output method="xml" omit-xml-declaration="yes"
                cdata-section-elements="k:description" indent="yes"/>
    <xsl:strip-space elements="*"/>

    <xsl:template match="node()|@*">
     <xsl:copy>
       <xsl:apply-templates select="node()|@*"/>
     </xsl:copy>
    </xsl:template>

    <xsl:template match="k:MultiGeometry|k:LineString">
        <xsl:apply-templates select='*'/>
    </xsl:template>

    <xsl:template match="k:description|k:Snippet|k:Style"/>
</xsl:stylesheet>z(Extra content at the end of the documentr)   r3  r+   r,   r   r   r   r   r   r4  r2   r2   r3   test_incorrect_xsl_syntax  s   
"r;  c                 C  r8  )Nr   a  <xsl:stylesheet version="1.0" xmlns:xsl="http://www.w3.org/1999/XSL/Transform"
                              xmlns:k="http://www.opengis.net/kml/2.2">
    <xsl:output method="xml" omit-xml-declaration="yes"
                cdata-section-elements="k:description" indent="yes"/>
    <xsl:strip-space elements="*"/>

    <xsl:template match="node(*)|@*">
     <xsl:copy>
       <xsl:apply-templates select="node()|@*"/>
     </xsl:copy>
    </xsl:template>

    <xsl:template match="k:MultiGeometry|k:LineString">
        <xsl:apply-templates select='*'/>
    </xsl:template>

    <xsl:template match="k:description|k:Snippet|k:Style"/>
</xsl:stylesheet>zfailed to compiler)   r3  r6  r:  r2   r2   r3   test_incorrect_xsl_eval  s
   
"r<  c                 C  r8  )Nr   ag  <xsl:stylesheet version="1.0" xmlns:xsl="http://www.w3.org/1999/XSL/Transform">
    <xsl:output method="xml" encoding="utf-8" indent="yes" />
    <xsl:strip-space elements="*"/>

    <xsl:template match="@*|node()">
        <xsl:copy>
            <xsl:copy-of select="document('non_existent.xml')/*"/>
        </xsl:copy>
    </xsl:template>
</xsl:stylesheet>zCannot resolve URIr)   r3  )r+   r,   r   ZXSLTApplyErrorr   r:  r2   r2   r3   test_incorrect_xsl_apply  s
   
"r=  c                 C  sT   t d}|d }t j|jdd t| |d W d    d S 1 s#w   Y  d S )Nr   zflatten.xslz!Start tag expected, '<' not foundr)   r3  r9  )r   Zxml_data_pathZ	xml_etreer4  r2   r2   r3   test_wrong_stylesheet*  s   
"r>  c                 C  s   t d t|||dkrdnd d%}|dkrt| }nt| }t| |d |jr/J W d    d S 1 s:w   Y  d S )Nr'   r6   rC   rD   r5   r3  )r+   r,   rG   r   rP   r   r   r   )r   r*  r>   rL   r0  r2   r2   r3   test_stylesheet_file_close6  s   
"r?  c                 C  sL   t d t jtdd t| d|d W d    d S 1 sw   Y  d S )Nr'   z*To use stylesheet, you need lxml installedr)   r@   )rA   r(  r   )r   r*  r2   r2   r3   test_stylesheet_with_etreeG  s   
"r@  c                 C  s^   t d d}tjddd}t jt|d t|| d W d    d S 1 s(w   Y  d S )Nr'   r(   rI   r   zcta_rail_lines.kmlr)   r3  )r+   r,   r   rK   r   r   r.   r   )r   r1   Zkmlr2   r2   r3   test_empty_stylesheetO  s   
"rA  c              	   C  s   t | ||dkr	dnd dF}|dkrA|dkrAtjtdd t||dg did	 W d    n1 s3w   Y  	 W d    d S t||dg did	}W d    n1 sVw   Y  tg d
g dg dg dg dd}t|| d S Nr6   rC   rD   r'   z.reading file objects must return bytes objectsr)   rU   rV   r\   rn   rq   rr   rs   ru   rv   )rG   r+   r   r   r   r   r-   r_   )r`   rA   r>   rL   Zdf_filelikerx   r2   r2   r3   test_file_like_iterparse^  s:   


rC  c              
   C  s6  |dkrt nt}t| ||dkrdnd db}|| L}|dkrV|dkrVtjtdd t||dg did	 W d    n1 sBw   Y  	 W d    W d    d S t||dg did	}W d    n1 skw   Y  W d    n1 szw   Y  tg d
g dg dg dg dd}t	
|| d S rB  )r   r   rG   rP   r+   r   r   r   r   r-   r_   )r`   rA   r>   ZfuncIOrL   bZ	df_fileiorx   r2   r2   r3   test_file_io_iterparse  sP   

rE  c              	   C  s   t |dd<}|j| d tjtdd t|j| dg did W d    n1 s-w   Y  W d    d S W d    d S 1 sEw   Y  d S )	NrC   rD   rd   )iterparse is designed for large XML filesr)   rf   r   r   r    dater\   )rG   rg   rP   r+   r   r   r   r   )rA   ri   r   rL   r2   r2   r3   test_url_path_error  s   
"rI  c              	   C  s   t jdd<}tj|| |d tjtdd t|| dg di|d W d    n1 s-w   Y  W d    d S W d    d S 1 sEw   Y  d S )	Nzgeom_xml.zipr   rA   rN   rF  r)   rf   rG  rA   r]   rN   )r-   rF   geom_dfto_xmlr+   r   r   r   )rA   rR   rK   r2   r2   r3   test_compression_error  s   
"rO  c                 C  sF   t jtdd t| |g dd W d    d S 1 sw   Y  d S )Nz&list is not a valid type for iterparser)   rV   r\   r  r   r2   r2   r3   test_wrong_dict_type  s   "rP  c                 C  sF   t jtdd t| |ddid W d    d S 1 sw   Y  d S )Nz8<class 'str'> is not a valid type for value in iterparser)   rU   rW   r\   r  r   r2   r2   r3   test_wrong_dict_value  s
   "rQ  c              	   C  s   d}t jddQ}t|ddd}|| W d    n1 s w   Y  tjtdd t|| d	gd
g did W d    n1 sDw   Y  W d    d S W d    d S 1 s\w   Y  d S )Na  <?xml version='1.0' encoding='utf-8'?>
  <row>
    <shape>square</shape>
    <degrees>00360</degrees>
    <sides>4.0</sides>
    <date>2020-01-01</date>
   </row>
  <row>
    <shape>circle</shape>
    <degrees>00360</degrees>
    <sides/>
    <date>2021-01-01</date>
  </row>
  <row>
    <shape>triangle</shape>
    <degrees>00180</degrees>
    <sides>3.0</sides>
    <date>2022-01-01</date>
  </row>
zbad.xmlrJ  rB   rC   rD   zDExtra content at the end of the document|junk after document elementr)   rH  rf   rG  )rA   Zparse_datesr]   )r-   rF   rG   rH   r+   r   r   r   )rA   Zbad_xmlrK   rL   r2   r2   r3   test_bad_xml  s&   
"rR  c                 C  ^   d}t t|d| d}t|| dddgid}tdd	gd
dgd}t|| t|| d S )Na-  <!-- comment before root -->
<shapes>
  <!-- comment within root -->
  <shape>
    <name>circle</name>
    <type>2D</type>
  </shape>
  <shape>
    <name>sphere</name>
    <type>3D</type>
    <!-- comment within child -->
  </shape>
  <!-- comment within root -->
</shapes>
<!-- comment after root -->r   r   r   r&   r   r\   r   r  r   r   r&   r   r   r   r2   r2   r3   test_comment  s   rU  c                 C  rS  )Na8  <?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE non-profits [
    <!ELEMENT shapes (shape*) >
    <!ELEMENT shape ( name, type )>
    <!ELEMENT name (#PCDATA)>
]>
<shapes>
  <shape>
    <name>circle</name>
    <type>2D</type>
  </shape>
  <shape>
    <name>sphere</name>
    <type>3D</type>
  </shape>
</shapes>r   r   r   r&   r   r\   r   r  r   r   rT  r   r   r2   r2   r3   test_dtd$     rV  c                 C  rS  )Nam  <?xml version="1.0" encoding="UTF-8"?>
<?xml-stylesheet type="text/xsl" href="style.xsl"?>
<?display table-view?>
<?sort alpha-ascending?>
<?textinfo whitespace is allowed ?>
<?elementnames <shape>, <name>, <type> ?>
<shapes>
  <shape>
    <name>circle</name>
    <type>2D</type>
  </shape>
  <shape>
    <name>sphere</name>
    <type>3D</type>
  </shape>
</shapes>r   r   r   r&   r   r\   r   r  r   r   rT  r   r   r2   r2   r3   test_processing_instructionH  rW  rX  c                 C  J   t jtdd t| |dg did W d    d S 1 sw   Y  d S )Nz+No result from selected items in iterparse.r)   nodeZattr1Zelem1Zelem2Zelem3r\   )r+   r   r   r   r   r2   r2   r3   test_no_resultl  s   
"r\  c                 C  rY  )NzNo columns to parse from filer)   rU   r[  r\   )r+   r   r   r   r   r2   r2   r3   test_empty_dataw  s   
"r]  c                  C  s`   t d d} d}tt| dddg|d}tdd	d
dddddddddd}t|| d S )Nr'   a  <?xml version="1.0" encoding="UTF-8"?>
<catalog>
  <cd>
    <title>Empire Burlesque</title>
    <artist>Bob Dylan</artist>
    <country>USA</country>
    <company>Columbia</company>
    <price>10.90</price>
    <year>1985</year>
  </cd>
  <cd>
    <title>Hide your heart</title>
    <artist>Bonnie Tyler</artist>
    <country>UK</country>
    <company>CBS Records</company>
    <price>9.90</price>
    <year>1988</year>
  </cd>
  <cd>
    <title>Greatest Hits</title>
    <artist>Dolly Parton</artist>
    <country>USA</country>
    <company>RCA</company>
    <price>9.90</price>
    <year>1982</year>
  </cd>
  <cd>
    <title>Still got the blues</title>
    <artist>Gary Moore</artist>
    <country>UK</country>
    <company>Virgin records</company>
    <price>10.20</price>
    <year>1990</year>
  </cd>
  <cd>
    <title>Eros</title>
    <artist>Eros Ramazzotti</artist>
    <country>EU</country>
    <company>BMG</company>
    <price>9.90</price>
    <year>1997</year>
  </cd>
  <cd>
    <title>One night only</title>
    <artist>Bee Gees</artist>
    <country>UK</country>
    <company>Polydor</company>
    <price>10.90</price>
    <year>1998</year>
  </cd>
  <cd>
    <title>Sylvias Mother</title>
    <artist>Dr.Hook</artist>
    <country>UK</country>
    <company>CBS</company>
    <price>8.10</price>
    <year>1973</year>
  </cd>
  <cd>
    <title>Maggie May</title>
    <artist>Rod Stewart</artist>
    <country>UK</country>
    <company>Pickwick</company>
    <price>8.50</price>
    <year>1990</year>
  </cd>
  <cd>
    <title>Romanza</title>
    <artist>Andrea Bocelli</artist>
    <country>EU</country>
    <company>Polydor</company>
    <price>10.80</price>
    <year>1996</year>
  </cd>
  <cd>
    <title>When a man loves a woman</title>
    <artist>Percy Sledge</artist>
    <country>USA</country>
    <company>Atlantic</company>
    <price>8.70</price>
    <year>1987</year>
  </cd>
  <cd>
    <title>Black angel</title>
    <artist>Savage Rose</artist>
    <country>EU</country>
    <company>Mega</company>
    <price>10.90</price>
    <year>1995</year>
  </cd>
  <cd>
    <title>1999 Grammy Nominees</title>
    <artist>Many</artist>
    <country>USA</country>
    <company>Grammy</company>
    <price>10.20</price>
    <year>1999</year>
  </cd>
  <cd>
    <title>For the good times</title>
    <artist>Kenny Rogers</artist>
    <country>UK</country>
    <company>Mucik Master</company>
    <price>8.70</price>
    <year>1995</year>
  </cd>
  <cd>
    <title>Big Willie style</title>
    <artist>Will Smith</artist>
    <country>USA</country>
    <company>Columbia</company>
    <price>9.90</price>
    <year>1997</year>
  </cd>
  <cd>
    <title>Tupelo Honey</title>
    <artist>Van Morrison</artist>
    <country>UK</country>
    <company>Polydor</company>
    <price>8.20</price>
    <year>1971</year>
  </cd>
  <cd>
    <title>Soulsville</title>
    <artist>Jorn Hoel</artist>
    <country>Norway</country>
    <company>WEA</company>
    <price>7.90</price>
    <year>1996</year>
  </cd>
  <cd>
    <title>The very best of</title>
    <artist>Cat Stevens</artist>
    <country>UK</country>
    <company>Island</company>
    <price>8.90</price>
    <year>1990</year>
  </cd>
  <cd>
    <title>Stop</title>
    <artist>Sam Brown</artist>
    <country>UK</country>
    <company>A and M</company>
    <price>8.90</price>
    <year>1988</year>
  </cd>
  <cd>
    <title>Bridge of Spies</title>
    <artist>T`Pau</artist>
    <country>UK</country>
    <company>Siren</company>
    <price>7.90</price>
    <year>1987</year>
  </cd>
  <cd>
    <title>Private Dancer</title>
    <artist>Tina Turner</artist>
    <country>UK</country>
    <company>Capitol</company>
    <price>8.90</price>
    <year>1983</year>
  </cd>
  <cd>
    <title>Midt om natten</title>
    <artist>Kim Larsen</artist>
    <country>EU</country>
    <company>Medley</company>
    <price>7.80</price>
    <year>1983</year>
  </cd>
  <cd>
    <title>Pavarotti Gala Concert</title>
    <artist>Luciano Pavarotti</artist>
    <country>UK</country>
    <company>DECCA</company>
    <price>9.90</price>
    <year>1991</year>
  </cd>
  <cd>
    <title>The dock of the bay</title>
    <artist>Otis Redding</artist>
    <country>USA</country>
    <COMPANY>Stax Records</COMPANY>
    <PRICE>7.90</PRICE>
    <YEAR>1968</YEAR>
  </cd>
  <cd>
    <title>Picture book</title>
    <artist>Simply Red</artist>
    <country>EU</country>
    <company>Elektra</company>
    <price>7.20</price>
    <year>1985</year>
  </cd>
  <cd>
    <title>Red</title>
    <artist>The Communards</artist>
    <country>UK</country>
    <company>London</company>
    <price>7.80</price>
    <year>1987</year>
  </cd>
  <cd>
    <title>Unchain my heart</title>
    <artist>Joe Cocker</artist>
    <country>USA</country>
    <company>EMI</company>
    <price>8.20</price>
    <year>1987</year>
  </cd>
</catalog>
a7  <?xml version="1.0" encoding="UTF-8"?>
<xsl:stylesheet version="1.0" xmlns:xsl="http://www.w3.org/1999/XSL/Transform">
<xsl:template match="/">
<html>
<body>
  <h2>My CD Collection</h2>
  <table border="1">
    <tr bgcolor="#9acd32">
      <th style="text-align:left">Title</th>
      <th style="text-align:left">Artist</th>
    </tr>
    <xsl:for-each select="catalog/cd">
    <tr>
      <td><xsl:value-of select="title"/></td>
      <td><xsl:value-of select="artist"/></td>
    </tr>
    </xsl:for-each>
  </table>
</body>
</html>
</xsl:template>
</xsl:stylesheet>
z.//tr[td and position() <= 6]rX   artist)r   r   r(  zEmpire BurlesquezHide your heartzGreatest HitszStill got the bluesZErosr!   z	Bob DylanzBonnie TylerzDolly Partonz
Gary MoorezEros Ramazzotti)rX   r^  )r+   r,   r   r   r   r-   r_   )r   r4  Zdf_xslrx   r2   r2   r3   test_online_stylesheet  s4   
 Vr_  c                 C  s~   t  %}tj|d| |d t|| |d}t||| dg di|d}W d    n1 s,w   Y  t |t t |t d S )NF)r   rA   rN   rK  rf   r   rL  )r-   rF   rM  rN  r   rS   r_   )rA   rR   rQ   rj   rk   r2   r2   r3   test_compression_read  s   

r`  c              	   C  s  |}|}||krt | d|  tdftdftdftdfd}tddd	}|d ur2|jd
f|d< tddd	}|d urBtdf|d< || \}}	t	 7}
t
j|
| |d t j||	d t|
| |d W d    n1 snw   Y  W d    d S W d    d S 1 sw   Y  d S )Nz == zInvalid data streamzNot a gzipped filezFile is not a zip filez%file could not be opened successfully)bz2gzipziptarZ	zstandardignore)errorszUnknown frame descriptorzstdlzmaz%Input format not supported by decoderxzrK  r)   )r+   skipOSErrorr	   r   r
   Z	ZstdErrorr   r-   rF   rM  rN  r   r   )rA   rN   rR   Zactual_compressionZattempted_compressionrf  rg  rh  	error_clsZ	error_strrK   r2   r2   r3   test_wrong_compression  s0   
"rm  c              	   C  sz   t jtdd, t }t|| dd W d    n1 sw   Y  W d    d S W d    d S 1 s6w   Y  d S )NzUnrecognized compression typer)   Z7zrK  )r+   r   r   r-   rF   r   )rA   rK   r2   r2   r3   test_unsuported_compression  s   
"rn  c                 C  sN   t d t d d| j d}t|d|d}t|d|d}t|| d S )NZs3fsr'   zs3://z
/books.xml)rA   Zstorage_optionsr@   )r+   r,   r&   r   r-   r_   )Zs3_public_bucket_with_dataZs3soZs3r   r   r2   r2   r3   test_s3_parser_consistency  s   

ro  c                   sz  d}|dkrt tjddgtjd}t tjdtgtjd}ntdtddg}tdd g}t	d| t
t|| |d}W d    n1 sPw   Y  t|td	d
gddtddgdd|td
tgddtdtgddtttgddtddgddtdtgddd	|dkrtdddlm  t fddjD  d d gd< t| d S )Na;  <?xml version='1.0' encoding='utf-8'?>
<data xmlns="http://example.com">
<row>
  <a>x</a>
  <b>1</b>
  <c>4.0</c>
  <d>x</d>
  <e>2</e>
  <f>4.0</f>
  <g></g>
  <h>True</h>
  <i>False</i>
</row>
<row>
  <a>y</a>
  <b>2</b>
  <c>5.0</c>
  <d></d>
  <e></e>
  <f></f>
  <g></g>
  <h>False</h>
  <i></i>
</row>
</data>pythonxyr   Zpyarrowzmode.string_storage)rA   dtype_backendr"   r#   ZInt64r   g      @ZFloat64TFboolean)	r   rD  r  derL   ghir   )ArrowExtensionArrayc                   s$   i | ]}| j | d dqS )T)Zfrom_pandas)array).0colrz  r   par2   r3   
<dictcomp>  s    z1test_read_xml_nullable_dtypes.<locals>.<dictcomp>rw  )r   npr{  Zobject_r   r+   r,   r   pdZoption_contextr   r   r   r   Zpandas.arraysrz  columnsr-   r_   )rA   Zstring_storagers  rI   Zstring_arrayZstring_array_nar  r2   r~  r3   test_read_xml_nullable_dtypes  s@   

r  c                  C  sD   d} t jt| d tddd W d    d S 1 sw   Y  d S )NzPdtype_backend numpy is invalid, only 'numpy_nullable' and 'pyarrow' are allowed.r)   testnumpy)rs  r   r0   r2   r2   r3   test_invalid_dtype_backend  s
   "r  )
__future__r   ior   r   rh  r   r   tarfiler   urllib.errorr   Zxml.etree.ElementTreer   zipfiler	   r  r  r+   Zpandas.compat._optionalr
   Zpandas.errorsr   r   Zpandas.util._test_decoratorsutilZ_test_decoratorstdZpandasr  r   r   r   Zpandas._testingZ_testingr-   Zpandas.core.arraysr   r   Zpandas.io.commonr   Zpandas.io.xmlr   r   rM  r/   r   r)  r4   Zfixturer>   r;   Z
skip_if_norA   rM   rS   rc   marknetworkZ
single_cpurl   ry   r}   r   r   r   r   r   Zparametrizer   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r  r  r
  r  r  r  r  r  r  r  r  r!  Zskip_if_installedr#  r$  r,  r.  r1  r2  r5  r7  r;  r<  r=  r>  r?  r@  rA  rC  rE  rI  rO  rP  rQ  rR  rU  rV  rX  r\  r]  r_  r`  rm  rn  ro  r  r  r2   r2   r2   r3   <module>   sJ   6
	m




	4

	%&*1

	
"*	)#$$	  	E