
    .i                        d dl Z d dlZd dlmZmZmZmZ d dlmc m	Z
 d dlmc mZ d dlmZ ej"                  d        Zej"                  d        Zej"                  d        Zej"                  d        Zej"                  d        Zej"                  d	        Z ej"                  d
      d        Z ej"                  d
      d        Zej"                  d        Z ej"                  d
      d        Zej"                  d        Zej"                  d        Zej"                  d        Zej"                  d        Zdddddddddd ej@                  d e
jB                  d              ej@                  d! e
jB                  d             gZ" ej"                  e"d"d #      d$        Z# ej"                  e"#      d%        Z$y)&    N)is_ci_environmentis_platform_armis_platform_macis_platform_windowsread_csvc                  r    t         j                  j                         D  ci c]  \  } }|| 
 c}} S c c}} w N)icomextension_to_compressionitems)keyvalues     V/var/www/app/trading-bot/venv/lib/python3.12/site-packages/pandas/tests/io/conftest.pycompression_to_extensionr      s-    )-)F)F)L)L)NO:3E3JOOOs   3c                      | dddd      S )zPath to the tips datasetiodatacsvtips.csv datapaths    r   	tips_filer      s     D&%44    c                      | dddd      S )zPath to a JSONL datasetr   parserr   items.jsonlr   r   s    r   
jsonl_filer      s     D(FM::r   c                 .    t         | dddd      d      S )z#DataFrame with the salaries datasetr   r   r   zsalaries.csv	)sepr   r   s    r   salaries_tabler#   "   s     HT8V^D$OOr   c                      | dddd      S )Nr   r   featherzfeather-0_3_1.featherr   r   s    r   feather_filer&   (   s    D&)-DEEr   c                      | dddd      S )Nr   r   xml	books.xmlr   r   s    r   xml_filer*   -   s    D&%55r   session)scopec                     | j                  dd       | j                  dd       | j                  dd       | j                  dd       | j                  dd       y)	z Mocked AWS Credentials for moto.AWS_ACCESS_KEY_IDtestingAWS_SECRET_ACCESS_KEYAWS_SECURITY_TOKENAWS_SESSION_AWS_SESSION_TOKENAWS_DEFAULT_REGIONz	us-east-1N)setenv)monkeysessions    r   aws_credentialsr6   2   sY     ,i80)<-y98)D-{;r   c              #   .  K   t               r#t               st               st               sd y t	        j
                  d      }|j                  d      }|j                          |j                         \  }}d| d|  |j                          y w)Nzhttp://localhost:5000zmoto.serverr   )portzhttp://:)
r   r   r   r   pytestimportorskipThreadedMotoServerstartget_host_and_portstop)r6   moto_serverserverhostr8   s        r   r@   r@   <   s      _.2E2G%%))-8//Q/7--/
dvQtf%%s   BBc                 V    t        j                  d      }|j                  d|       }|S )Nboto3s3)endpoint_url)r:   r;   resource)r@   rD   rE   s      r   moto_s3_resourcerH   L   s)    (E	;	7BIr   c                     dd| iiS )Nclient_kwargsrF   r   )r@   s    r   s3sorK   S   s     	K
 r   c              #      K   dt        j                          }| j                  |      }|j                  d       | |j                  j                          |j                          yw)z/
    Create a public S3 bucket using moto.
    zpandas-test-zpublic-readACLNuuiduuid4BucketcreateobjectsdeleterH   bucket_namebuckets      r   s3_bucket_publicrY   \   sV     
 !/K$$[1F
MMmM$
L
NN
MMO   A*A,c              #      K   dt        j                          }| j                  |      }|j                  d       | |j                  j                          |j                          yw)z0
    Create a private S3 bucket using moto.
    zcant_get_it-privaterM   NrO   rV   s      r   s3_bucket_privater]   i   sV     
 !/K$$[1F
MMiM 
L
NN
MMOrZ   c                     d|fd|fd|dz   fd|dz   fd|fd|fd	|fg}|D ]-  \  }}t        |d
      5 }| j                  ||       ddd       / | S # 1 sw Y   ;xY wzw
    The following datasets
    are loaded.

    - tips.csv
    - tips.csv.gz
    - tips.csv.bz2
    - items.jsonl
    z
tips#1.csvr   ztips.csv.gz.gzztips.csv.bz2.bz2r   zsimple_dataset.featherr)   rb)KeyBodyNopen
put_object)	rY   r   r   r&   r*   test_s3_filess3_key	file_namefs	            r   s3_bucket_public_with_datarl   v   s     
y!	Y		E)*	V+,	
#	!<0	hM + <	)T" 	<a''F';	< 	<< 	< 	<   AA	c                     d|fd|fd|dz   fd|dz   fd|fd|fd	|fg}|D ]-  \  }}t        |d
      5 }| j                  ||       ddd       / | S # 1 sw Y   ;xY wr_   re   )	r]   r   r   r&   r*   rh   ri   rj   rk   s	            r   s3_bucket_private_with_dataro      s     
y!	Y		E)*	V+,	
#	!<0	hM + =	)T" 	=a((V!(<	= 	== 	= 	=rm   )z.no_compressN) N)r`   gzip)z.GZrq   )ra   bz2)z.BZ2rr   )z.zipzip)z.ZIPrs   )z.xzxz)z.XZrt   )z.zstzstd	zstandard)marks)z.ZSTru      )paramsc                     | j                   S r
   paramrequests    r   compression_formatr      s    ==r   c                      | j                   d   S )Nr   r{   r}   s    r   compression_extr      s    ==r   )%rP   r:   pandas.compatr   r   r   r   pandas.util._test_decoratorsutil_test_decoratorstdpandas.io.commonr   commonr   pandas.io.parsersr   fixturer   r   r   r#   r&   r*   r6   r@   rH   rK   rY   r]   rl   ro   r|   
skip_if_no_compression_formats_paramsr   r   r   r   r   <module>r      s8      * )   & P P 5 5
 ; ;
 P P
 F F 6 6 i < !< i  !   i  ! 	 	 	 	  6  8 FLL!{)CDFLL!{)CD   21267 8 23 4r   