
    [6g                     0   d Z ddlmZ ddlZddlmZ ddlZddlmc m	Z
 ddlZddlmZ ej        j        ej        j        ej                            d          gZd Zd Zd Zd	 Zd
 Zd Zd Zd Zd Zd Zej                            deej         feej!        fed f ej"        e eej#        d           e
j$        d                     ej"        e eej#        d           e
j$        d           e
j$        d          e
j%        g          eej&        feej'        feej         feej!        fg	          ej                            ddddidddg          d                         Z(ej                            dddg          d             Z)dS )z9
Tests for the pandas custom headers in http(s) requests
    )partialN)BytesIOz=ignore:Passing a BlockManager to DataFrame:DeprecationWarningc                     t                      5 }t          j        |d          5 }|                    |            d d d            n# 1 swxY w Y   |                                cd d d            S # 1 swxY w Y   d S )Nw)fileobjmode)r   gzipGzipFilewritegetvalue)response_bytesbiozippers      `/var/www/surfInsights/venv3-11/lib/python3.11/site-packages/pandas/tests/io/test_http_headers.py
gzip_bytesr      s    	 c]3S111 	)VLL(((	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	)||~~                 s3   A3AA3A	A3A	A33A7:A7c                 T    |                      d                              d          S NF)indexutf-8)to_csvencodedfs    r   csv_responderr      s$    9959!!((111    c                 :    t          t          |                     S N)r   r   r   s    r   gz_csv_responderr   #   s    mB''(((r   c                 P    |                                                      d          S )Nr   )to_jsonr   r   s    r   json_responderr!   '   s    ::<<w'''r   c                 :    t          t          |                     S r   )r   r!   r   s    r   gz_json_responderr#   +   s    nR(()))r   c                 T    |                      d                              d          S r   )to_htmlr   r   s    r   html_responderr&   /   s$    ::E:""))'222r   c                 0    |                      dd          S )NFpyarrow)r   engine)
to_parquetr   s    r   parquetpyarrow_reponderr+   3   s    ==uY=777r   c                     dd l }|                     dddd            |                    dd          5 }|                                cd d d            S # 1 swxY w Y   d S )Nr   z'memory://fastparquet_user_agent.parquetFfastparquet)r   r)   compressionrb)fsspecr*   openread)r   r0   fs      r   parquetfastparquet_responderr4   7   s     MMMMM1	     
>	E	E vvxx                 s   AAAc                     t                      5 }|                     |           |                                cd d d            S # 1 swxY w Y   d S r   )r   	to_pickler   r   r   s     r   pickle_respnderr8   J   s    	 c
S||~~                 s   )AA	A	c                     t                      5 }|                     |d           |                                cd d d            S # 1 swxY w Y   d S )NF)write_index)r   to_statar   r7   s     r   stata_responderr<   P   s    	 c
CU+++||~~                 s   +AAAzresponder, read_methodc                  0    t          j        | i |d         S )Nr   )pd	read_html)argskwargss     r   <lambda>rB   ]   s    BL$$A&$A$A!$D r   r(   )r)   )marksr-   r0   storage_options
User-AgentfoobarrE   Authc                    t          j        ddgi          }g d}d| j        v rddi}||}n||z  }nd }t          |                              |r|                                ng           }|                     | |          |            ||j        |          }t          j	        ||           t          |j        d	         j                  }	|D ]+}
|	                    |
          }|r|
|v r|||
         k    sJ ,|	rJ d S )
Nab)zAccept-EncodingHost
ConnectionrE   gzzContent-Encodingr	   )contentheaders)rD   r   )r>   	DataFrame__name__setunionkeysserve_contenturltmassert_frame_equaldictrequestsrQ   pop)	responderread_method
httpserverrD   expecteddefault_headersextraexpected_headersresultrequest_headersheaderexps               r   test_request_headersri   V   sU   N |S3%L))HMMMOy!!!#V,"#OOu$OO?++11"19r  YYx%8%8%HHH[IIIF&(+++:.q19::O" 2 2!!&)) 	2v88/&11111r   r)   c                    ddd}t          j        |            t          j        ddgi          }d}t          j        t
          |          5  |                    d|| 	           d d d            d S # 1 swxY w Y   d S )
Ncustomother_customrH   column_namecolumn_valuezxstorage_options passed with file object or non-fsspec file path|storage_options passed with buffer, or non-supported URL)matchz/tmp/junk.parquet)rD   r)   )pytestimportorskipr>   rR   raises
ValueErrorr*   )r)   rQ   true_dfmsgs       r   ,test_to_parquet_to_disk_with_storage_optionsrv      s      G
 lMN+;<==G	C  
z	-	-	- X X.PVWWWX X X X X X X X X X X X X X X X X Xs   A44A8;A8)*__doc__	functoolsr   r	   ior   rp   pandas.util._test_decoratorsutil_test_decoratorstdpandasr>   pandas._testing_testingrY   mark
single_cpunetworkfilterwarnings
pytestmarkr   r   r   r!   r#   r&   r+   r4   r8   r<   parametrizeread_csv	read_jsonparamread_parquet
skip_if_no&skip_array_manager_not_yet_implementedread_pickle
read_statari   rv    r   r   <module>r      s                  ) ) ) ) ) ) ) ) )           K
K
KG 
  2 2 2) ) )( ( (* * *3 3 38 8 8  &     	$	&DD	
 	#GBOI666"-	**	
 	
 	

 	(GBOM::: m,,h''9			
 		
 		
 
".)	"-(	2;'	BL)3 < 	ue,,   = L6  X X X X Xr   