U é¥¡c²ã@s˜ddlZddlmZddlmZdddddd d ddd ddddddddddddddddddgZe 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"œZzddlZd!ed<Wnek ræYnXzddl Z d!ed <Wnek rYnXzddl Zd!ed <Wnek r>YnXzddlZd!ed<Wnek rjYnXzddlZd!ed<Wnek r–YnXzddl Z d!ed<Wnek rÊd!ed<YnXzddlZd!ed<Wnek röYnXzddlZd!ed<Wnek r"YnXzddlZd!ed<Wnek rNYnXzddlZd!ed<Wnek rzYnXzddlZd!ed<Wnek r¦YnXzdd#lmZd!ed<Wnek rÖYnXzdd$lmZd!ed<Wnek rYnXzdd%lmZd!ed<Wnek r6YnXzddlZd!ed<Wnek rbYnXd&d'„Zejd!d(d)d*„ƒZejd!d(d+d,„ƒZdS)-é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ÚzstdFT)rrrrr rr rrr rrrrrrrrrrrrrrrrrr)Ú GcsFileSystem©ÚS3FileSystem)ÚHadoopFileSystemcCs |jjrÒdt|ƒkrdSddddddg}|D]$}d |¡t|ƒkr,t|s,dSq,d t|ƒkrjtdsjdSdt|ƒkrœzd dl}WdStk ršYdSXdt|ƒkrÒzd dlm}WdStk rÐYdSXt |jddƒrdt|ƒkrôdSdt|ƒkrdSdS)Nz/pyarrow/tests/Trrrrrrz pyarrow/{}zpyarrow/parquet/encryptionrzpyarrow/cudarFz pyarrow/fsr!Údoctest_cythonz/pyarrow/_parquet_encryption) ÚoptionÚdoctestmodulesÚstrÚformatÚdefaultsZpyarrow.cudaÚImportErrorÚ pyarrow.fsr"Úgetattr)ÚpathÚconfigZdoctest_groupsÚgroupÚpyarrowr"©r1ú4/tmp/pip-unpacked-wheel-qoi2rb4q/pyarrow/conftest.pyÚpytest_ignore_collect²sFú r3)Zautousec csR|jjj}t|jjddƒ}|s"|rH| d¡}| ¡dVW5QRXndVdS)Nr$FÚtmpdir)r.r%r&r,ZgetfixturevalueZas_cwd)ÚrequestÚ doctest_mÚ doctest_cr4r1r1r2Ú_docdirãs r8c csˆ|jjj}t|jjddƒ}|s"|r~t|d<t ¡}|d}| t|ƒ¡}| d¡W5QRX||d<t|ƒ|d<t|ƒ|d<dVdS) Nr$Frzpyarrow-fs-example.datsdataÚlocalÚ local_pathr-) r.r%r&r,rZLocalFileSystemZopen_output_streamr'Úwrite)Zdoctest_namespacer5Ztmp_pathr6r7r9r-Ústreamr1r1r2Úadd_fsøs r=)Zpytestr0rrÚgroupsZis_availabler)rr*r Zpyarrow.gandivaZpyarrow.datasetZpyarrow.orcrZpyarrow.parquetZpyarrow.parquet.encryptionZpyarrow.plasmarZpyarrow.flightr+r r"r#Zpyarrow.substraitr3Zfixturer8r=r1r1r1r2Úsä ä1