
    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                            d+          ed(<   	 d dlZded<   n# e$ r Y nw xY w	 d dlZded<   n# e$ r Y nw xY w	 d dlZded<   n# e$ r Y nw xY w	 d dlZded<   n# e$ r Y nw xY w	 d dlZded<   n# e$ r Y nw xY w	 d dlZej        d,k    rded<   ned(         ed<   n# e$ r Y nw xY w	 d dlZded<   n# e$ r ded<   Y nw xY w	 d dlZded<   n# e$ r ded<   Y nw xY w	 d dlZded<   n# e$ r Y nw xY w	 d dlZded<   n# e$ r Y nw xY w	 d dlZded<   n# e$ r Y nw xY w	 d d-lmZ ded<   n# e$ r Y nw xY w	 d d.lmZ ded<   n# e$ r Y nw xY w	 d d/lmZ ded"<   n# e$ r Y nw xY w	 d d0lmZ ded<   n# e$ r Y nw xY w	 d dl Zded&<   n# e$ r Y nw xY wd1 Z! e j"        d2          d3             Z# e j"        d2          d4             Z$ e j"        d56          d7             Z% e j"        d56          d8             Z& e j"        d56          d9             Z'dS ):    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        rdt          |           v rdS g d}|D ]&}d| t          |           v rt          |         s dS 'dt          |           v rt          d         sdS dt          |           v r |             S d	t          |           v r	 d
dlm} dS # t          $ r Y dS w xY wt          |j         dd          rBdt          |           v rdS dt          |           v rdS dt          |           v r |             S dS )Nc                  2    	 dd l } dS # t          $ r Y dS w xY w)Nr   TF)pyarrow.cudaImportError)pyarrows    Z/lsinfo/ai/hellotax_ai/base_platform/venv/lib/python3.11/site-packages/pyarrow/conftest.py_cuda_is_availablez1pytest_ignore_collect.<locals>._cuda_is_available   s:    	4 	 	 	55	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444
 
 
 $ 	  	 E!%!!S%9%999  44'3+?+???01 tS1111))++++3////333333u   tt v}.66 ,O 4 4444)S-A-AAA4s?3333))++++5s   B& &
B43B4)autousec              #     K   | j         j        j        }t          | j         j        dd          }|s|rH|                     d          }|                                5  d V  d d d            d S # 1 swxY w Y   d S d V  d S )Nr6   Ftmpdir)r>   r7   r8   r<   getfixturevalueas_cwd)request	doctest_m	doctest_crD   s       r4   _docdirrJ     s       %4I-/?GGI 
I 
 ((22 ]]__ 	 	EEE	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	s   A..A25A2c              #     K   |j         j        j        }t          |j         j        dd          }|s|rt          | d<   t	          j                    }|dz  }|                    t          |                    5 }|                    d           d d d            n# 1 swxY w Y   || d<   t          |          | d<   t          |          | d<   d V  d S )	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      %4I-/?GGI .I ."$$ "$$22%%c$ii00 	"FLL!!!	" 	" 	" 	" 	" 	" 	" 	" 	" 	" 	" 	" 	" 	" 	"%*'"*-h--,'$'II&!	EEEEEs   7BB Bsession)scopec                      ddl m fd} d}ddd}                    | ||dt          j                    it          j                               | |fS )	z+
    Register a unary scalar function.
    r   computec                 B                         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,/O   = = 	=    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5 5I ) )!("(** 5 "

	, , ,
 9$$rd   c                      ddl m}  ddlfd}d}ddd}|                     |||d	t	          j                    it	          j                               ||fS )
z4
    Register a unary aggregate function (mean)
    r   rY   Nc                 R    t          j                            |                    S )N)rj   scalarnanmean)r`   ra   nps     r4   funcz$unary_agg_func_fixture.<locals>.funcW  s    y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  s    
 &%%%%%( ( ( ( ( I%/1 1H ""4#,#+'*BJLL$% $&:<<% % % ?rd   c                      ddl m}  ddlfd}d}ddd}|                     |||t	          j                    t	          j                    dt	          j                               ||fS )	z-
    Register a unary aggregate function
    r   rY   Nc                 j    d}|D ]}|                     |          z  }t          j        |          S )Ng        )rr   rj   rq   )r`   argssumargrs   s       r4   rt   z&varargs_agg_func_fixture.<locals>.funcq  s;     	# 	#C2::c??"CCy~~rd   sum_meanzVarargs aggregatere   )ra   y)r3   rZ   r   rv   rj   rk   rw   rx   s       @r4   varargs_agg_func_fixturer   i  s    
 &%%%%%     I.24 4H ""4#,#+,.HJJ,.JLL$% $% $&:<<% % % ?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 E EH_	MMMHX 	 	 	D		"H] 	 	 	D		HY 	 	 	D		HW 	 	 	D		HY 	 	 	D	
	
|w
 #?3 	 	 	D	 MMMHX      HZ LLLHW   HY	HY 	 	 	D		%%%%%)H!"" 	 	 	D		HX 	 	 	D		******HW 	 	 	D		((((((HUOO 	 	 	D		''''''HTNN 	 	 	D		++++++HV 	 	 	D		 H[ 	 	 	D	
1 1 1j   ( 
 
 
, i   % % ! %( i     ! 2 i     !   s   0	D: :EE	E EE	E& &E.-E.2	E< <FF	F FF F? ?GG	G 
G"!G"&	G0 0
G=<G=	H HH	H! !H)(H)-	H7 7H?>H?I III' 'I/.I/3I? ?JJJ JJ#	J- -J54J5