B lnd@sddlZddlZddlmZddlmZdddddd d d d d dddddddddddddddddgZdededddddddd eddddedddddddd ddeddedd!ZyddlZd ed<Wne k rYnXyddl Z d ed <Wne k rYnXyddl Zd ed <Wne k rBYnXyddl Zd ed<Wne k rnYnXyddl Zd ed<Wne k rYnXyddlZd ed<Wne k rYnXyddlZd ed<Wne k rd ed<YnXyddlZd ed<Wne k r&YnXyddlZd ed<Wne k rRYnXyddlZd ed<Wne k r~YnXydd"lmZd ed <Wne k rYnXydd#lmZd ed<Wne k rYnXydd$lmZd ed<Wne k rYnXyddlZd ed<Wne k r:YnXd%d&Zejd d'd(d)Zejd d'd*d+Zejd,d-d.d/ZdS)0N)Codec)fsacerobrotlibz2cythondataset hypothesis fastparquetgandivagcsgdbgziphdfs large_memorylz4 memory_leaknopandasorcpandasparquetparquet_encryptions3snappy substraitflightslowrequires_testing_datazstdFT)rrrrrr rr r r rrr rrrrrrrrrrrrrr) GcsFileSystem) S3FileSystem)HadoopFileSystemcCs|jjrdt|krdSdddddg}x*|D]"}d|t|kr,t|s,dSq,Wd t|krjtd sjdSd t|kry d dl}d Stk rdSXdt|kryd dlm}d Stk rdSXt |jdd rdt|krdSdt|krdSd S)Nz/pyarrow/tests/Trrrrrz pyarrow/{}zpyarrow/parquet/encryptionrz pyarrow/cudarFz pyarrow/fs)r doctest_cythonz/pyarrow/_parquet_encryption) optiondoctestmodulesstrformatdefaultsZ pyarrow.cuda ImportError pyarrow.fsr getattr)pathconfigZdoctest_groupsgrouppyarrowr r/4/tmp/pip-unpacked-wheel-i2z0tf32/pyarrow/conftest.pypytest_ignore_collectsB       r1)Zautousec csR|jjj}t|jjdd}|s"|rH|d}| dVWdQRXndVdS)Nr"Ftmpdir)r,r#r$r*ZgetfixturevalueZas_cwd)request doctest_m doctest_cr2r/r/r0_docdirs   r6c cs|jjj}t|jjdd}|s"|r~t|d<t}|d}|t|}|dWdQRX||d<t||d<t||d<dVdS) Nr"Frzpyarrow-fs-example.datsdatalocal local_pathr+) r,r#r$r*rZLocalFileSystemZopen_output_streamr%write)Zdoctest_namespacer3Ztmp_pathr4r5r7r+streamr/r/r0add_fss   r;session)scopecsLddlmfdd}d}ddd}|||d tit||fS) z+ Register a unary scalar function. r)computecsjd|dg|jdS)Nadd) memory_pool) call_functionrA)ctxx)pcr/r0unary_function s z*unary_func_fixture..unary_functionzy=x+1z add functionztest add function)summary descriptionarray)r.r>Zregister_scalar_functionpaZint64)rF func_nameZ unary_docr/)rEr0unary_func_fixtures    rL)Zpytestr.rJrrgroupsZ is_availabler'rr(r Zpyarrow.gandivaZ pyarrow.aceroZpyarrow.datasetZ pyarrow.orcrZpyarrow.parquetZpyarrow.parquet.encryptionZpyarrow.flightr)rr r!Zpyarrow.substraitr1Zfixturer6r;rLr/r/r/r0s                   0