d}i:(~ddlZddlZddlZddlmZddlmZddlmZddlZgdZ iddddd ej d d ej d d dd dd ddddddddddej ddddddddej dddiddddddddddddddd dd!dd"dd#dd$ej d$d%dd&dd'ed(dd)ej d)Z ej d*krde d<de d <de d%<ej d*krej d+e d(< ddlZde d <n #e$rYnwxYw ddlZde d <n #e$rYnwxYw ddlZde d<n #e$rYnwxYw ddlZde d<n #e$rYnwxYw ddlZde d <n #e$rYnwxYw ddlZej d,krde d<n e d(e d<n #e$rYnwxYw ddlZde d<n#e$rde d<YnwxYw ddlZde d<n#e$rde d<YnwxYw ddlZde d<n #e$rYnwxYw ddlZde d<n #e$rYnwxYw ddlZde d<n #e$rYnwxYw dd-lmZde d<n #e$rYnwxYw dd.lmZde d<n #e$rYnwxYw dd/lmZde d"<n #e$rYnwxYw dd0lmZde d<n #e$rYnwxYw ddl Zde d&<n #e$rYnwxYwd1Z!ej"d2d3Z#ej"d2d4Z$ej"d56d7Z%ej"d56d8Z&ej"d56d9Z'dS):N)Codec)fs)is_threading_enabled)"aceroazurebrotlibz2cythondataset hypothesis fastparquetflightgandivagcsgdbgziphdfs large_memorylz4 memory_leaknopandasnonumpynumpyorcpandasparquetparquet_encryption processesrequires_testing_datas3slowsnappysockets substrait threading timezone_datazstdrFrrr r r r rrrrTrrr rrrrrrrrrrrrr r!r"r#r$r%r&r' emscriptenz/usr/share/zoneinfowin32)AzureFileSystem) GcsFileSystem S3FileSystem)HadoopFileSystemc"d}|jjrdt|vrdSgd}|D]&}d|t|vrt|sdS'dt|vrtdsdSdt|vr | Sd t|vr d d lm}d S#t $rYdSwxYwt|jd d rBdt|vrdSdt|vrdSdt|vr | Sd S)Nc2 ddl}dS#t$rYdSwxYw)NrTF) pyarrow.cuda ImportError)pyarrows >C:\PYTHON\_runtimes\venv\Lib\site-packages\pyarrow/conftest.py_cuda_is_availablez1pytest_ignore_collect.._cuda_is_availables:     4   55 s  z/pyarrow/tests/T)r rrrr$zpyarrow/zpyarrow/parquet/encryptionrz pyarrow/cudaz pyarrow/fsrr,Fdoctest_cythonz/pyarrow/_parquet_encryptionz/pyarrow/_cuda)optiondoctestmodulesstrdefaults pyarrow.fsr-r2getattr)collection_pathconfigr5doctest_groupsgroupr-s r4pytest_ignore_collectrAs}# O 4 4 4 44   $  E!%!!S%9%999 44 '3+?+? ? ?01 t S11 1 1))+++ + 3// / / 333333u   tt v}.66, O 4 4 4 44 )S-A-A A A4 s?33 3 3))+++ + 5sB&& B43B4)autousec#K|jjj}t|jjdd}|s|rH|d}|5dVddddS#1swxYwYdSdVdS)Nr6Ftmpdir)r>r7r8r<getfixturevalueas_cwd)request doctest_m doctest_crDs r4_docdirrJs%4I-/?GGI I ((22]]__   EEE                   sA..A25A2c#K|jjj}t|jjdd}|s|rt|d<t j}|dz }|t|5}|ddddn #1swxYwY||d<t||d<t||d<dVdS) Nr6Frzpyarrow-fs-example.datsdatalocal local_pathpath) r>r7r8r<rLocalFileSystemopen_output_streamr9write)doctest_namespacerGtmp_pathrHrIrLrNstreams r4add_fsrU#s%4I-/?GGI .I ."$$"$$22  % %c$ii 0 0 "F LL ! ! ! " " " " " " " " " " " " " " "%*'"*-h--,'$'II&! EEEEEs7BB Bsession)scopecddlmfd}d}ddd}|||dtjitj||fS) z+ Register a unary scalar function. rcomputecBd|dg|jS)Nadd) memory_pool) call_functionr^)ctxxpcs r4unary_functionz*unary_func_fixture..unary_functionAs-1v,/O == =zy=x+1z add functionztest add functionsummary descriptionarray)r3rZregister_scalar_functionpaint64)rc func_name unary_docrbs @r4unary_func_fixturern:s &%%%%%=====I* 355I ) )!("(** 5 " ,,, 9 $$rdcddlm}ddlfd}d}ddd}||||d t jit j||fS) z4 Register a unary aggregate function (mean) rrYNcRtj|S)N)rjscalarnanmean)r`ranps r4funcz$unary_agg_func_fixture..funcWsyA'''rdmean_udfzy=avg(x)zfind mean of xrera)r3rZrregister_aggregate_functionrjfloat64rbrtrlfunc_docrss @r4unary_agg_func_fixturerzOs &%%%%%(((((I%/11H""4#,#+'*BJLL$%$&:<< %%% ?rdcddlm}ddlfd}d}ddd}||||t jt jdt j||fS) z- Register a unary aggregate function rrYNcjd}|D]}||z }tj|S)Ng)rrrjrq)r`argssumargrss r4rtz&varargs_agg_func_fixture..funcqs; # #C 2::c?? "CCy~~rdsum_meanzVarargs aggregatere)ray)r3rZrrvrjrkrwrxs @r4varargs_agg_func_fixtureris &%%%%% I.244H""4#,#+,.HJJ,.JLL$%$%$&:<<%%% ?rd)(pytestosr3rjrr pyarrow.librsysgroups is_availabler:platformrNexistsr r2r pyarrow.gandiva pyarrow.aceropyarrow.dataset pyarrow.orcrrpyarrow.parquetpyarrow.parquet.encryptionpyarrow.flightr;r*r+r-r.pyarrow.substraitrAfixturerJrUrnrzrrdr4rs$ ,,,,,, # # # J#  U#  U#   e **#  5 e $ $ #   e #  u # 5#  e# u#  5#  4#  E v & &#  E# %# E#   5 e $ $!# "5## # $%# &u'# ( U)# * 5+# , e-# .u/# 0%1# 23# 4T5# 6 %7# 8 E9# :  e **;# <t=# >?# @%%''A# BTC# D E v & &E# # J<<HUO!H[HY<< "/D E EH_ MMMHX   D  "H]   D  HY   D  HW   D  HY   D    |w #?3   D  MMMHX   HZ LLLHWHY HY   D  %%%%%)H !""   D  HX   D  ******HW   D  ((((((HUOO   D  ''''''HTNN   D  ++++++HV   D   H[   D 111j(   ,i   %%! %(i   ! 2i   ! s 0 D::EE EEE E&&E.-E.2 E<<FF FFF F??GG G G"!G"& G00 G=<G= H HH H!!H)(H)- H77H?>H? III I''I/.I/3 I??JJ JJJ# J--J54J5