K i&ddlZddlZddlZddlmZddlmZddlmZddlm Z ddl Z gdZ iddd dd ejd d ejd d dd ddddddddddddejddddddddejdddiddddddddddddd dd!dd"dd#dd$dd%ejd%d&dd'dd(ed)dd*ejd*Z e jd+k(rde d<de d!<de d&<e jd,k(r e e d)<n-e jd+k(rejj!d-e d)< ddlZde d < ddlZde d< ddlZde d< ddlZde d< ddlZde d < ddlZe jd,k(rde d<ne d)e d< ddlZde d< ddlZde d< ddlZde d< ddlZde d < ddlZde d< dd.lmZde d < dd/lmZde d< dd0lm Z de d#< dd1lm!Z!de d< ddl"Zde d'<d2Z#ejHd3d4Z%ejHd3d5Z&ejHd67d8Z'ejHd67d9Z(ejHd67d:Z)y#e$rY?wxYw#e$rYAwxYw#e$rYCwxYw#e$rYEwxYw#e$rYGwxYw#e$rY1wxYw#e$r de d<Y8wxYw#e$r de d<Y?wxYw#e$rYAwxYw#e$rYCwxYw#e$rYEwxYw#e$rYEwxYw#e$rYEwxYw#e$rYEwxYw#e$rYEwxYw#e$rYGwxYw);N)Codec)fs)is_threading_enabled)windows_has_tzdata)"aceroazurebrotlibz2cythondataset hypothesis fastparquetflightgandivagcsgdbgziphdfs large_memorylz4 memory_leaknopandasnonumpynumpyorcpandasparquetparquet_encryption processesrequires_testing_datas3slowsnappysockets substrait threading timezone_datazstdrFrr r r r rrrrrTrrr rrrrrrrrrrrr r!r"r#r$r%r&r'r( emscriptenwin32z/usr/share/zoneinfo)AzureFileSystem) GcsFileSystem S3FileSystem)HadoopFileSystemc|jjrtdt|vrygd}|D]}d|t|vst|rydt|vr tdsydt|vr ddl}y d t|vr dd lm}y t|jd d rdt|vryd t|vryy #t $rYywxYw#t $rYywxYw)Nz/pyarrow/tests/T)r rrrr%zpyarrow/zpyarrow/parquet/encryptionrz pyarrow/cudarFz pyarrow/fsr-doctest_cythonz/pyarrow/_parquet_encryption) optiondoctestmodulesstrdefaults pyarrow.cuda ImportError pyarrow.fsr.getattr)collection_pathconfigdoctest_groupsgrouppyarrowr.s V/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/pyarrow/conftest.pypytest_ignore_collectr@s }}## O 4 4 $ E%!S%99 (3+? ?01 S1 1 # 3/ / 3v}}.6 O 4 4 )S-A A !    s$1B?C? C  C  CC)autousec#K|jjj}t|jjdd}|s|r/|j d}|j 5ddddydy#1swYyxYww)Nr1Ftmpdir)r;r2r3r9getfixturevalueas_cwd)request doctest_m doctest_crCs r?_docdirrI sy%%44I--/?GII((2]]_        sA'B)A<.B<BBc#K|jjj}t|jjdd}|s|rwt|d<t j }|dz }|j t|5}|jdddd||d<t||d<t||d<dy#1swY/xYww) Nr1Frzpyarrow-fs-example.datsdatalocal local_pathpath) r;r2r3r9rLocalFileSystemopen_output_streamr4write)doctest_namespacerFtmp_pathrGrHrKrMstreams r?add_fsrT!s%%44I--/?GII"$$""$22  % %c$i 0 "F LL ! "%*'"*-h-,'$'I&!  " "sBCC.CC  Csession)scopecddlmfd}d}ddd}j|||dtjitj||fS) z+ Register a unary scalar function. rcomputecDjd|dg|jS)Nadd) memory_pool) call_functionr])ctxxpcs r?unary_functionz*unary_func_fixture..unary_function?s*1v,/OO = =zy=x+1z add functionztest add functionsummary descriptionarray)r>rYregister_scalar_functionpaint64)rb func_name unary_docras @r?unary_func_fixturerm8s[ &=I* 35I ) )!("((* 5 " , 9 $$rccddlm}ddlfd}d}ddd}|j|||d t j it j ||fS) z4 Register a unary aggregate function (mean) rrXNcLtjj|S)N)riscalarnanmean)r_r`nps r?funcz$unary_agg_func_fixture..funcUsyyA''rcmean_udfzy=avg(x)zfind mean of xrdr`)r>rYrregister_aggregate_functionrifloat64rarsrkfunc_docrrs @r?unary_agg_func_fixtureryMsc &(I%/1H""4#,#+'*BJJL$%$&::< % ?rccddlm}ddlfd}d}ddd}|j|||t j t j dt j ||fS) z- Register a unary aggregate function rrXNchd}|D]}|j|z }tj|S)Ng)rqrirp)r_argssumargrrs r?rsz&varargs_agg_func_fixture..funcos7 #C 2::c? "C #yy~rcsum_meanzVarargs aggregaterd)r`y)r>rYrrurirjrvrws @r?varargs_agg_func_fixturergsj & I.24H""4#,#+,.HHJ,.JJL$%$&::<% ?rc)*pytestosr>rirr pyarrow.librpyarrow.tests.utilrsysgroups is_availabler5platformrMexistsr r7rpyarrow.gandiva pyarrow.aceropyarrow.dataset pyarrow.orcrrpyarrow.parquetpyarrow.parquet.encryptionpyarrow.flightr8r+r,r.r/pyarrow.substraitr@fixturerIrTrmryrrcr?rs$ ,1 # 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<<7 2 4H_\\\! "/D EH_ HX "H] HY HW HY   ||w #?3 HXHW HY %%)H !" HX *HW (HUO 'HTN +HV  H[ ,`(  ,i %!%(i !2i !c             HZ HY                 s " J", J.6 J: K K!K6 K* K; L  L L$( L04 L< M M M "J+*J+.J76J7:KKKKKKK'&K'* K87K8; L L  LLL! L!$L-,L-0L98L9<MMMMMM M)(M)