a
    ,gh                     @   sD  d dl Z d dlmZ d dlmZ g dZededddddddded	ddded
ddddddddddedddeddZzd dlZded< W n ey   Y n0 zd dl	Z	ded< W n ey   Y n0 zd dl
Zded< W n ey   Y n0 zd dlZded< W n ey,   Y n0 zd dlZded< W n eyV   Y n0 zd dlZded< W n ey   ded< Y n0 zd dlZded< W n ey   Y n0 zd dlZded< W n ey   Y n0 zd dlZded< W n ey   Y n0 zd dlZded< W n ey0   Y n0 zd dlZded< W n eyZ   Y n0 zd dlmZ ded< W n ey   Y n0 zd dlmZ ded< W n ey   Y n0 zd dlmZ ded< W n ey   Y n0 zd dlZded < W n ey   Y n0 d!d" Ze jdd#d$d% Ze jdd#d&d' ZdS )(    N)Codec)fs)brotlibz2cythondataset
hypothesisfastparquetgandivagcsgdbgziphdfslarge_memorylz4memory_leaknopandasorcpandasparquetparquet_encryptionplasmas3snappy	substrait
tensorflowflightslowrequires_testing_datazstdr   r   FTr   r   r   r   )r   r   r   r   r	   r   r
   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r	   r
   r   r   r   r   r   r   r   r   r   )GcsFileSystemr   S3FileSystemr   )HadoopFileSystemr   r   c                 C   s   |j jrdt| v rdS g d}|D ]$}d|t| v r$t| s$ dS q$dt| v rbtd sbdS dt| v rzdd l}W d	S  ty   Y dS 0 d
t| v rzddlm} W d	S  ty   Y dS 0 t	|j dd	 rdt| v rdS dt| v  rdS d	S )Nz/pyarrow/tests/T)r   r   r   r   r   r   z
pyarrow/{}zpyarrow/parquet/encryptionr   zpyarrow/cudar   Fz
pyarrow/fsr!   doctest_cythonz/pyarrow/_parquet_encryption)
optiondoctestmodulesstrformatdefaultsZpyarrow.cudaImportError
pyarrow.fsr"   getattr)pathconfigZdoctest_groupsgrouppyarrowr"    r1   `/mounts/lovelace/software/anaconda3/envs/metaDMG/lib/python3.9/site-packages/pyarrow/conftest.pypytest_ignore_collect   s:    
r3   )Zautousec                 c   sf   | j jj}t| j jdd}|s"|r\| d}|  d V  W d    qb1 sP0    Y  nd V  d S )Nr$   Ftmpdir)r.   r%   r&   r,   ZgetfixturevalueZas_cwd)request	doctest_m	doctest_cr4   r1   r1   r2   _docdir   s    


&r8   c                 c   s   |j jj}t|j jdd}|s"|rt| d< t }|d }|t|}|d W d    n1 sh0    Y  || d< t|| d< t|| d< d V  d S )	Nr$   Fr   zpyarrow-fs-example.dats   datalocalZ
local_pathr-   )	r.   r%   r&   r,   r   ZLocalFileSystemZopen_output_streamr'   write)Zdoctest_namespacer5   Ztmp_pathr6   r7   r9   r-   streamr1   r1   r2   add_fs   s    
(r<   )Zpytestr0   r   r   groupsZis_availabler)   r   r*   r	   Zpyarrow.gandivaZpyarrow.datasetZpyarrow.orcr   Zpyarrow.parquetZpyarrow.parquet.encryptionZpyarrow.plasmar   Zpyarrow.flightr+   r    r"   r#   Zpyarrow.substraitr3   Zfixturer8   r<   r1   r1   r1   r2   <module>   s    1


