
    j)'                        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 ej&                         5   ej(                  d,e       d dlZddd       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 d0lm"Z" ded"<   	 d d1lm#Z# ded<   	 d dl$Zded&<   d2 Z% e jL                  d3      d4        Z' e jL                  d3      d5        Z( e jL                  d67      d8        Z) e jL                  d67      d9        Z* e jL                  d67      d:        Z+y# e$ r Y pw xY w# e$ r Y rw xY w# 1 sw Y   JxY w# e$ r Y Pw xY w# e$ r Y Rw xY w# e$ r Y Tw xY w# e$ r Y >w xY w# e$ r	 ded<   Y Ew xY w# e$ r	 ded<   Y Lw xY w# e$ r Y Nw xY w# e$ r Y Pw xY w# e$ r Y Rw xY w# e$ r Y Rw xY w# e$ r Y Rw xY w# e$ r Y Rw xY w# e$ r Y Rw xY w# e$ r Y Tw 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ignore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    7/tmp/pip-target-jj7clzyk/lib/python/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.   r3   getattr)collection_pathconfigr6   doctest_groupsgroupr.   s         r5   pytest_ignore_collectrB      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)Nr7   Ftmpdir)r?   r8   r9   r=   getfixturevalueas_cwd)request	doctest_m	doctest_crE   s       r5   _docdirrK     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)	Nr7   Fr   zpyarrow-fs-example.dats   datalocal
local_pathpath)	r?   r8   r9   r=   r   LocalFileSystemopen_output_streamr:   write)doctest_namespacerH   tmp_pathrI   rJ   rM   rO   streams           r5   add_fsrV   &  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     r5   unary_functionz*unary_func_fixture.<locals>.unary_functionD  s*    1v,/OO   = 	=    zy=x+1zadd functionztest add functionsummarydescriptionarray)r4   r[   register_scalar_functionpaint64)rd   	func_name	unary_docrc   s      @r5   unary_func_fixturero   =  s[    
 &= I* 35I ) )!("((* 5 "
	,
 9$$re   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   rZ   Nc                 L    t        j                  j                  |            S )N)rk   scalarnanmean)ra   rb   nps     r5   funcz$unary_agg_func_fixture.<locals>.funcZ  s    yyA''re   mean_udfzy=avg(x)zfind mean of xrf   rb   )r4   r[   r   register_aggregate_functionrk   float64rc   ru   rm   func_docrt   s       @r5   unary_agg_func_fixturer{   R  sc    
 &( I%/1H ""4#,#+'*BJJL$% $&::<% ?re   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   rZ   Nc                 h    d}|D ]  }|j                  |      z  } t        j                  |      S )Ng        )rs   rk   rr   )ra   argssumargrt   s       r5   ru   z&varargs_agg_func_fixture.<locals>.funct  s7     	#C2::c?"C	#yy~re   sum_meanzVarargs aggregaterf   )rb   y)r4   r[   r   rw   rk   rl   rx   ry   s       @r5   varargs_agg_func_fixturer   l  sj    
 & I.24H ""4#,#+,.HHJ,.JJL$% $&::<% ?re   ),pytestosr4   rk   r   r   pyarrow.libr   sysgroupsis_availabler;   platformrO   existsr
   r3   r   warningscatch_warningssimplefilterFutureWarningpyarrow.gandivapyarrow.aceropyarrow.datasetpyarrow.orcr   r   pyarrow.parquetpyarrow.parquet.encryptionpyarrow.flightr<   r+   r,   r.   r/   pyarrow.substraitrB   fixturerK   rV   ro   r{   r    re   r5   <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]		 	 	 	" h6 HY	HW	HY
	
||w
 #?3 HXHW	HY	%%)H!"	HX	*HW	(HUO	'HTN	+HV	 H[1j  ( 
 
, i % !%( i  !2 i  !s  		  		
   		  		  		  		   HZ   HY  		  		  		  		  		  		  		  		s8  	J3 	J? K +KK 	K$ 	K0 %!K< 	L 	L 	L* %	L6 /	M 9M M M& M2 )	M> 3J<;J<?KKKK K! K!$K-,K-0K98K9<LL
LL
L'&L'*L32L36L?>L?M
MMMM#"M#&M/.M/2M;:M;>NN