
    i&                        d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZ d dlZg dZ	i ddddd	 ej                  d	      d
 ej                  d
      ddddddddddddddd ej                  d      ddddddd ej                  d      ddi ddddddddddddddd dd!dd"dd#dd$ ej                  d$      d%dd&dd' e       d(dd) ej                  d)      Zej                  d*k(  rded<   ded <   ded%<   ej                  d*k(  rej                  j                  d+      ed(<   	 d dlZded<   	 d dlZded<   	 d dlZded<   	 d dlZded<   	 d dlZded<   	 d dlZej                  d,k(  rded<   ned(   ed<   	 d dlZded<   	 d dlZded<   	 d dlZded<   	 d dlZded<   	 d dlZded<   	 d d-lmZ ded<   	 d d.lmZ ded<   	 d d/lmZ ded"<   	 d d0lmZ ded<   	 d dl Zded&<   d1 Z! e jD                  d2      d3        Z# e jD                  d2      d4        Z$ e jD                  d56      d7        Z% e jD                  d56      d8        Z& e jD                  d56      d9        Z'y# e$ r Y ?w xY w# e$ r Y Aw xY w# e$ r Y Cw xY w# e$ r Y Ew xY w# e$ r Y Gw xY w# e$ r Y 1w xY w# e$ r	 ded<   Y 8w xY w# e$ r	 ded<   Y ?w xY w# e$ r Y Aw xY w# e$ r Y Cw xY w# e$ r Y Ew xY w# e$ r Y Ew xY w# e$ r Y Ew xY w# e$ r Y Ew xY w# e$ r Y Ew xY w# e$ r Y Gw xY w):    N)Codec)fs)is_threading_enabled)"aceroazurebrotlibz2cythondataset
hypothesisfastparquetflightgandivagcsgdbgziphdfslarge_memorylz4memory_leaknopandasnonumpynumpyorcpandasparquetparquet_encryption	processesrequires_testing_datas3slowsnappysockets	substrait	threadingtimezone_datazstdr   Fr   r   r	   r
   r   r   r   r   r   r   Tr   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r    r!   r"   r#   r$   r%   r&   r'   
emscriptenz/usr/share/zoneinfowin32)AzureFileSystem)GcsFileSystemS3FileSystem)HadoopFileSystemc                    d }|j                   j                  rvdt        |       v ryg d}|D ]  }d| t        |       v st        |   r y dt        |       v r
t        d   sydt        |       v r |        S d	t        |       v r	 d
dlm} yt        |j                   dd      r1dt        |       v rydt        |       v rydt        |       v r |        S y# t        $ r Y yw xY w)Nc                  ,    	 dd l } y# t        $ r Y yw xY w)Nr   TF)pyarrow.cudaImportError)pyarrows    I/home/work/apex_v16/venv/lib/python3.12/site-packages/pyarrow/conftest.py_cuda_is_availablez1pytest_ignore_collect.<locals>._cuda_is_available   s    	 		s    	z/pyarrow/tests/T)r   r   r   r   r$   zpyarrow/zpyarrow/parquet/encryptionr   zpyarrow/cudaz
pyarrow/fsr   r,   Fdoctest_cythonz/pyarrow/_parquet_encryptionz/pyarrow/_cuda)optiondoctestmodulesstrdefaults
pyarrow.fsr-   r2   getattr)collection_pathconfigr5   doctest_groupsgroupr-   s         r4   pytest_ignore_collectrA      s    }}##O 44
 $ 	 E%!S%99	 
 (3+??01S11)+++3//3 v}}.6O 44)S-AAs?33)+++  s   	C 	C%$C%)autousec              #     K   | j                   j                  j                  }t        | j                   j                  dd      }|s|r/| j	                  d      }|j                         5  d  d d d        y d  y # 1 sw Y   y xY ww)Nr6   Ftmpdir)r>   r7   r8   r<   getfixturevalueas_cwd)request	doctest_m	doctest_crD   s       r4   _docdirrJ     sy      %%44I--/?GII ((2 ]]_ 		 	 			 	s   A'B)A<.B<BBc              #     K   |j                   j                  j                  }t        |j                   j                  dd      }|s|rwt        | d<   t	        j
                         }|dz  }|j                  t        |            5 }|j                  d       d d d        || d<   t        |      | d<   t        |      | d<   d  y # 1 sw Y   /xY ww)	Nr6   Fr   zpyarrow-fs-example.dats   datalocal
local_pathpath)	r>   r7   r8   r<   r   LocalFileSystemopen_output_streamr9   write)doctest_namespacerG   tmp_pathrH   rI   rL   rN   streams           r4   add_fsrU   #  s      %%44I--/?GII"$$ ""$22%%c$i0 	"FLL!	"%*'"*-h-,'$'I&!		" 	"s   BCC.CC	Csession)scopec                      ddl m fd} d}ddd}j                  | ||dt        j                         it        j                                | |fS )	z+
    Register a unary scalar function.
    r   computec                 D    j                  d|dg| j                        S )Nadd   )memory_pool)call_functionr^   )ctxxpcs     r4   unary_functionz*unary_func_fixture.<locals>.unary_functionA  s*    1v,/OO   = 	=    zy=x+1zadd functionztest add functionsummarydescriptionarray)r3   rZ   register_scalar_functionpaint64)rc   	func_name	unary_docrb   s      @r4   unary_func_fixturern   :  s[    
 &= I* 35I ) )!("((* 5 "
	,
 9$$rd   c                      ddl m}  ddlfd}d}ddd}| j                  |||d	t	        j
                         it	        j
                                ||fS )
z4
    Register a unary aggregate function (mean)
    r   rY   Nc                 L    t        j                  j                  |            S )N)rj   scalarnanmean)r`   ra   nps     r4   funcz$unary_agg_func_fixture.<locals>.funcW  s    yyA''rd   mean_udfzy=avg(x)zfind mean of xre   ra   )r3   rZ   r   register_aggregate_functionrj   float64rb   rt   rl   func_docrs   s       @r4   unary_agg_func_fixturerz   O  sc    
 &( I%/1H ""4#,#+'*BJJL$% $&::<% ?rd   c                      ddl m}  ddlfd}d}ddd}| j                  |||t	        j
                         t	        j                         dt	        j                                ||fS )	z-
    Register a unary aggregate function
    r   rY   Nc                 h    d}|D ]  }|j                  |      z  } t        j                  |      S )Ng        )rr   rj   rq   )r`   argssumargrs   s       r4   rt   z&varargs_agg_func_fixture.<locals>.funcq  s7     	#C2::c?"C	#yy~rd   sum_meanzVarargs aggregatere   )ra   y)r3   rZ   r   rv   rj   rk   rw   rx   s       @r4   varargs_agg_func_fixturer   i  sj    
 & I.24H ""4#,#+,.HHJ,.JJL$% $&::<% ?rd   )(pytestosr3   rj   r   r   pyarrow.libr   sysgroupsis_availabler:   platformrN   existsr
   r2   r   pyarrow.gandivapyarrow.aceropyarrow.datasetpyarrow.orcr   r   pyarrow.parquetpyarrow.parquet.encryptionpyarrow.flightr;   r*   r+   r-   r.   pyarrow.substraitrA   fixturerJ   rU   rn   rz   r    rd   r4   <module>r      s  $  	    , 
#
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#2 3#4 T5#6 	%7#8 E9#:  e  *;#< t=#> ?#@ %'A#B TC#D Ev&E#J <<< HUO!H[HY<<< "/D EH_	HX	"H]	HY	HW	HY
	
||w
 #?3 HXHW	HY	%%)H!"	HX	*HW	(HUO	'HTN	+HV	 H[1j  ( 
 
, i % !%( i  !2 i  !m  		  		  		  		  		  		   HZ   HY  		  		  		  		  		  		  		  		s   	J 	J 	J  	J& *	J2 4!J> 	K
  	K *	K, 4	K8 >	L L L  L( ,L4 8	M  J
JJJJ#"J#&J/.J/2J;:J;>KK

KK
K)(K),K54K58L LLLLLL%$L%(L10L14L=<L= M	M	