o
    Gh                     @   s   d dl Z d dlZd dlZd dlZd dlmZ ejdddd Zejdddd Zejd	d
 Z	ejdd Z
ejdd ZdS )    N)guidmodule)scopec                 C   s   | d S )Nparquet )base_datadirr   r   _/var/www/html/Persson_Maskin/env/lib/python3.10/site-packages/pyarrow/tests/parquet/conftest.pydatadir   s   r	   c                  C   s6   t jdkr
td tjd} | stdt	| S )N
emscriptenz$needs PARQUET_TEST_DATA files accessPARQUET_TEST_DATAzRPlease point the PARQUET_TEST_DATA environment variable to the test data directory)
sysplatformpytestskiposenvirongetRuntimeErrorpathlibPath)resultr   r   r   parquet_test_datadir    s   


r   c           	      C   s   t d}t d}d}| d \}}}}|jdd| d| |||jjdd	d
d}z z|j|d W n	 ty<   Y nw W |  |S W |  |S |  w )Nboto3botocorez	test-s3fs
connections3http://:s3v4)signature_versionz	us-east-1)endpoint_urlaws_access_key_idaws_secret_access_keyconfigregion_name)Bucket)r   importorskipclientConfigcreate_bucket	Exceptionclose)		s3_serverr   r   s3_bucket_namehostport
access_key
secret_key	s3_clientr   r   r   	s3_bucket+   s0   

	
r3   c           	      c   s    t d}| d \}}}}|j||dd| d| id}| dt  }|| ||fV  z
|j|dd	 W d S  tyD   Y d S w )
Ns3fsr   r    r   r   )keysecretclient_kwargs/T)	recursive)r   r&   S3FileSystemr   mkdirrmFileNotFoundError)	r,   r3   r4   r.   r/   r0   r1   fs	test_pathr   r   r   s3_example_s3fsE   s"   


r@   c           	   	   c   sd    ddl m} | d \}}}}d| d| d| d| d	}||\}}|d |||fV  d S )	Nr   )
FileSystemr   zs3://r   z5@mybucket/data.parquet?scheme=http&endpoint_override=z&allow_bucket_creation=Truemybucket)
pyarrow.fsrA   from_uri
create_dir)	r,   rA   r.   r/   r0   r1   urir>   pathr   r   r   s3_example_fs\   s   
rH   )r   r   r   r   pyarrow.utilr   fixturer	   r   r3   r@   rH   r   r   r   r   <module>   s   






