
    \
qi                     f   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
j!        d                      ej         d! e
j!        d                     gZ" ej        e"d"d         #          d$             Z# ej        e"#          d%             Z$dS )&    N)is_ci_environmentis_platform_armis_platform_macis_platform_windowsread_csvc                  R    d t           j                                        D             S )Nc                     i | ]\  }}||	S  r   ).0keyvalues      x/var/www/html/bestrading.cuttalo.com/services/ml-inference/venv/lib/python3.11/site-packages/pandas/tests/io/conftest.py
<dictcomp>z,compression_to_extension.<locals>.<dictcomp>   s    OOO:3E3OOO    )icomextension_to_compressionitemsr   r   r   compression_to_extensionr      s%    OO)F)L)L)N)NOOOOr   c                      | dddd          S )zPath to the tips datasetiodatacsvtips.csvr   datapaths    r   	tips_filer      s     8D&%444r   c                      | dddd          S )zPath to a JSONL datasetr   parserr   items.jsonlr   r   s    r   
jsonl_filer!      s     8D(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'     HHT8V^DD$OOOOr   c                      | dddd          S )Nr   r   featherzfeather-0_3_1.featherr   r   s    r   feather_filer(   (   s    8D&)-DEEEr   c                      | dddd          S )Nr   r   xml	books.xmlr   r   s    r   xml_filer,   -   s    8D&%555r   session)scopec                     |                      dd           |                      dd           |                      dd           |                      dd           |                      dd           dS )	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_credentialsr8   2   s}     ,i8880)<<<-y9998)DDD-{;;;;;r   c              #   l  K   t                      r0t                      s"t                      st                      sdV  d S t	          j        d          }|                    d          }|                                 |                                \  }}d| d| V  |	                                 d S )Nzhttp://localhost:5000zmoto.serverr   )portzhttp://:)
r   r   r   r   pytestimportorskipThreadedMotoServerstartget_host_and_portstop)r8   moto_serverserverhostr:   s        r   rB   rB   <   s        

,..
2E2G2G
 &%%%%%)-88//Q/77--//
d%%%t%%%%%r   c                 \    t          j        d          }|                    d|           }|S )Nboto3s3)endpoint_url)r<   r=   resource)rB   rF   rG   s      r   moto_s3_resourcerJ   L   s,    ((E	;	7	7BIr   c                     dd| iiS )Nclient_kwargsrH   r   )rB   s    r   s3sorM   S   s     	K
 r   c              #      K   dt          j                     }|                     |          }|                    d           |V  |j                                         |                                 dS )z/
    Create a public S3 bucket using moto.
    zpandas-test-zpublic-readACLNuuiduuid4BucketcreateobjectsdeleterJ   bucket_namebuckets      r   s3_bucket_publicr[   \   sp      
 0//K$$[11F
MMmM$$$
LLL
N
MMOOOOOr   c              #      K   dt          j                     }|                     |          }|                    d           |V  |j                                         |                                 dS )z0
    Create a private S3 bucket using moto.
    zcant_get_it-privaterO   NrQ   rX   s      r   s3_bucket_privater^   i   sp      
 0//K$$[11F
MMiM   
LLL
N
MMOOOOOr   c                     d|fd|fd|dz   fd|dz   fd|fd|fd	|fg}|D ]D\  }}t          |d
          5 }|                     ||           ddd           n# 1 swxY w Y   E| S 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)	r[   r   r!   r(   r,   test_s3_filess3_key	file_namefs	            r   s3_bucket_public_with_datarm   v   s     
y!	Y		E)*	V+,	
#	!<0	hM + < <	)T"" 	<a''F';;;	< 	< 	< 	< 	< 	< 	< 	< 	< 	< 	< 	< 	< 	< 	<   AA	 A	c                     d|fd|fd|dz   fd|dz   fd|fd|fd	|fg}|D ]D\  }}t          |d
          5 }|                     ||           ddd           n# 1 swxY w Y   E| S r`   rf   )	r^   r   r!   r(   r,   ri   rj   rk   rl   s	            r   s3_bucket_private_with_datarp      s     
y!	Y		E)*	V+,	
#	!<0	hM + = =	)T"" 	=a((V!(<<<	= 	= 	= 	= 	= 	= 	= 	= 	= 	= 	= 	= 	= 	= 	=rn   )z.no_compressN) N)ra   gzip)z.GZrr   )rb   bz2)z.BZ2rs   )z.zipzip)z.ZIPrt   )z.xzxz)z.XZru   )z.zstzstd	zstandard)marks)z.ZSTrv      )paramsc                     | j         S )Nparamrequests    r   compression_formatr      s
    =r   c                     | j         d         S )Nr   r|   r~   s    r   compression_extr      s    =r   )%rR   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,   r8   rB   rJ   rM   r[   r^   rm   rp   r}   
skip_if_no_compression_formats_paramsr   r   r   r   r   <module>r      s;                * ) ) ) ) ) ) ) )          & & & & & & P P P 5 5 5
 ; ; ;
 P P P
 F F F 6 6 6 i   < < ! < i     !     i     !  	 	 	 	 	 	   6   8 FL!{)C)CDDDFL!{)C)CDDD   21226777  87 2333  43  r   