
    'eF!                        d dl Z d dlZd dlmZ d dlmZ d dlZg dZi 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ddddddddd ej        d          d ej        d          d
Z		 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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 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 dlZde	d#<   n# e$ r Y nw xY wd$ Z e j        d%          d&             Z e j        d%          d'             Z e j        d()          d*             Z e j        d()          d+             Z e j        d()          d,             Z dS )-    N)Codec)fs)acerobrotlibz2cythondataset
hypothesisfastparquetgandivagcsgdbgziphdfslarge_memorylz4memory_leaknopandasorcpandasparquetparquet_encryptions3snappy	substraitflightslowrequires_testing_datazstdr   Fr   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   )GcsFileSystemS3FileSystemr   )HadoopFileSystemr   c                    |j         j        rdt          |           v rdS g d}|D ]6}d                    |          t          |           v rt          |         s dS 7dt          |           v rt          d         sdS dt          |           v r	 dd l}d	S # t          $ r Y dS w xY wd
t          |           v r	 ddlm} d	S # t          $ r Y dS w xY wt          |j         dd	          r&dt          |           v rdS dt          |           v rdS d	S )Nz/pyarrow/tests/T)r	   r   r   r   r   z
pyarrow/{}zpyarrow/parquet/encryptionr   zpyarrow/cudar   Fz
pyarrow/fsr!   doctest_cythonz/pyarrow/_parquet_encryption)
optiondoctestmodulesstrformatdefaultspyarrow.cudaImportError
pyarrow.fsr"   getattr)pathconfigdoctest_groupsgrouppyarrowr"   s         0  /croot/pyarrow_1707330824290/_h_env_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold/lib/python3.11/site-packages/pyarrow/conftest.pypytest_ignore_collectr5      s   }# #D		))4
 
 
 $ 	  	 E""5))SYY66  44'3t994401 tSYY&&####u   tt 3t99$$333333u   tt v}.66 D		))4)SYY6645s$   B 
B#"B#8C   
CC)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 )Nr%   Ftmpdir)r0   r&   r'   r.   getfixturevalueas_cwd)request	doctest_m	doctest_cr8   s       r4   _docdirr>      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 )	Nr%   Fr   zpyarrow-fs-example.dats   datalocal
local_pathr/   )	r0   r&   r'   r.   r   LocalFileSystemopen_output_streamr(   write)doctest_namespacer;   tmp_pathr<   r=   r@   r/   streams           r4   add_fsrH      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_functionrQ   )ctxxpcs     r4   unary_functionz*unary_func_fixture.<locals>.unary_function  s-    1v,/O   = = 	=    zy=x+1zadd functionztest add functionsummarydescriptionarray)r3   rM   register_scalar_functionpaint64)rV   	func_name	unary_docrU   s      @r4   unary_func_fixturera     s    
 &%%%%%= = = = = I* 35 5I ) )!("(** 5 "

	, , ,
 9$$rW   c                      ddl m}  d }d}ddd}|                     |||dt          j                    it          j                               ||fS )	z4
    Register a unary aggregate function (mean)
    r   rL   c                 N    t          j        t          j        |                    S )N)r]   scalarnpnanmean)rS   rT   s     r4   funcz$unary_agg_func_fixture.<locals>.func$  s    yA'''rW   mean_udfzy=avg(x)zfind mean of xrX   rT   )r3   rM   register_aggregate_functionr]   float64rU   rg   r_   func_docs       r4   unary_agg_func_fixturerm     s    
 &%%%%%( ( ( I%/1 1H ""4#,#+'*BJLL$% $&:<<% % % ?rW   c                      ddl m}  d }d}ddd}|                     |||t          j                    t          j                    dt          j                               ||fS )z-
    Register a unary aggregate function
    r   rL   c                 f    d}|D ]}|t          j        |          z  }t          j        |          S )Ng        )re   rf   r]   rd   )rS   argssumargs       r4   rg   z&varargs_agg_func_fixture.<locals>.func=  s8     	# 	#C2:c??"CCy~~rW   sum_meanzVarargs aggregaterX   )rT   y)r3   rM   ri   r]   r^   rj   rk   s       r4   varargs_agg_func_fixtureru   6  s    
 &%%%%%   I.24 4H ""4#,#+,.HJJ,.JLL$% $% $&:<<% % % ?rW   )!pytestr3   r]   r   r   numpyre   groupsis_availabler*   r   r,   r   pyarrow.gandivapyarrow.aceropyarrow.datasetpyarrow.orcr   pyarrow.parquetpyarrow.parquet.encryptionpyarrow.flightr-   r    r"   r#   pyarrow.substraitr5   fixturer>   rH   ra   rm   ru    rW   r4   <module>r      s  $                     
 
 
<U e ** 
5e$$ e	
 u 5 e u 
5 
4 Ev&& E % E 
5e$$  5!" #$ !
 e **Ev&&7  <	MMMHX 	 	 	D		"H] 	 	 	D		HY 	 	 	D		HW 	 	 	D		HY 	 	 	D		HUOO 	 	 	D	 MMMHX      HZ 	HY 	 	 	D		%%%%%)H!"" 	 	 	D		HX 	 	 	D		((((((HUOO 	 	 	D		''''''HTNN 	 	 	D		++++++HV 	 	 	D		 H[ 	 	 	D	
, , ,`   ( 
 
 
, i   % % ! %( i     ! 0 i     !   s   4	B> >CC
	C CC 	C* *C21C26	D   DD	D DD"	D, ,D43D48	E 
EE	E E%$E%)	E3 3E;:E;?	F	 	FFF! !F)(F)-F9 9G GG GG	G' 'G/.G/