
    Mh                        S r SSKJr  SSKrSSKJr  SSKrSSKJs  J	r
  SSKrSSKJr  \R                   R"                  \R                   R$                  \R                   R'                  S5      /rS rS rS rS	 rS
 rS rS rS rS rS r\R                   R?                  S\\R@                  4\\RB                  4\S 4\RD                  " \\" \RF                  SS9\
RH                  " S5      S9\RD                  " \\" \RF                  SS9\
RH                  " S5      \
RH                  " S5      \
RJ                  /S9\\RL                  4\\RN                  4\\R@                  4\\RB                  4/	5      \R                   R?                  SSSS0SSS./5      S 5       5       r(\R                   R?                  SSS/5      S 5       r)g)z9
Tests for the pandas custom headers in http(s) requests
    )partialN)BytesIOz=ignore:Passing a BlockManager to DataFrame:DeprecationWarningc                     [        5        n[        R                  " USS9 nUR                  U 5        S S S 5        UR	                  5       sS S S 5        $ ! , (       d  f       N'= f! , (       d  f       g = f)Nw)fileobjmode)r   gzipGzipFilewritegetvalue)response_bytesbiozippers      S/var/www/html/env/lib/python3.13/site-packages/pandas/tests/io/test_http_headers.py
gzip_bytesr      sI    	c]]3S1VLL( 2||~ 
11 
s!   A%AA%
A"	A%%
A3c                 >    U R                  SS9R                  S5      $ NF)indexutf-8)to_csvencodedfs    r   csv_responderr      s    9959!((11    c                 *    [        [        U 5      5      $ N)r   r   r   s    r   gz_csv_responderr   #   s    mB'((r   c                 @    U R                  5       R                  S5      $ )Nr   )to_jsonr   r   s    r   json_responderr!   '   s    ::<w''r   c                 *    [        [        U 5      5      $ r   )r   r!   r   s    r   gz_json_responderr#   +   s    nR())r   c                 >    U R                  SS9R                  S5      $ r   )to_htmlr   r   s    r   html_responderr&   /   s    ::E:"))'22r   c                 "    U R                  SSS9$ )NFpyarrow)r   engine)
to_parquetr   s    r   parquetpyarrow_reponderr+   3   s    ==uY=77r   c                     SS K nU R                  SSSS S9  UR                  SS5       nUR                  5       sS S S 5        $ ! , (       d  f       g = f)Nr   z'memory://fastparquet_user_agent.parquetFfastparquet)r   r)   compressionrb)fsspecr*   openread)r   r0   fs      r   parquetfastparquet_responderr4   7   sM     MM1	   
>	Evvx 
F	E	Es   A
Ac                     [        5        nU R                  U5        UR                  5       sS S S 5        $ ! , (       d  f       g = fr   )r   	to_pickler   r   r   s     r   pickle_respnderr8   J   s(    	c
S||~ 
s	   !6
Ac                     [        5        nU R                  USS9  UR                  5       sS S S 5        $ ! , (       d  f       g = f)NF)write_index)r   to_statar   r7   s     r   stata_responderr<   P   s,    	c
CU+||~ 
s	    5
Azresponder, read_methodc                  4    [         R                  " U 0 UD6S   $ )Nr   )pd	read_html)argskwargss     r   <lambda>rB   ]   s    BLL$$A&$A!$Dr   r(   )r)   )marksr-   r0   storage_options
User-AgentfoobarrE   Authc                    [         R                  " SS/05      n/ SQnSU R                  ;   a  SS0nUc  UnOX6-  nOS n[        U5      R	                  U(       a  UR                  5       O/ 5      nUR                  U " U5      US9  U" UR                  US9n[        R                  " X5        [        UR                  S	   R                  5      n	U H.  n
U	R                  U
5      nU(       d  M  X;   d  M$  XU
   :X  a  M.   e   U	(       a   eg )
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   s	   N ||S3%L)HMOy!!!#V,"#O$O?+11"1r Yx%8%HIF&+:..q199:O"!!&)?v8&1111 #
 r   r)   c                     SSS.n[         R                  " U 5        [        R                  " SS/05      nSn[         R                  " [
        US9   UR                  SXS	9  S S S 5        g ! , (       d  f       g = f)
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      sr     G
 llMN+;<=G	C  
z	-.W 
.	-	-s   A**
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       ) )   KK
KK
KKG
2)(*38& 	$	&D	
 	#BOOI6--	*	

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