
    '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	<   	 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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 dlZde	d#<   d$ Z e j6                  d%      d&        Z e j6                  d%      d'        Z e j6                  d()      d*        Z e j6                  d()      d+        Z e j6                  d()      d,        Z y# e$ r Y w xY w# e$ r Y w xY w# e$ r Y w xY w# e$ r Y w xY w# e$ r Y w xY w# e$ r Y w xY w# e$ r	 de	d<   Y "w xY w# e$ r Y $w xY w# e$ r Y &w xY w# e$ r Y (w xY w# e$ r Y (w xY w# e$ r Y (w xY w# e$ r Y (w xY w# e$ r Y *w xY w)-    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yg d}|D ]*  }dj                  |      t        |       v s t        |   r* y dt        |       v r
t        d   sydt        |       v r	 dd l}y	d
t        |       v r	 ddlm} y	t        |j                   dd	      rdt        |       v rydt        |       v ryy	# t        $ r Y yw xY w# t        $ r Y yw xY w)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         0lib/python3.11/site-packages/pyarrow/conftest.pypytest_ignore_collectr5      s    }}##D	)
 $ 	 E""5)SY6	 
 (3t9401SY&# 3t9$3 v}}.6D	))SY6!    s$   =C C 	C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)Nr%   Ftmpdir)r0   r&   r'   r.   getfixturevalueas_cwd)request	doctest_m	doctest_cr8   s       r4   _docdirr>      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)	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      %%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_functionrQ   )ctxxpcs     r4   unary_functionz*unary_func_fixture.<locals>.unary_function  s*    1v,/OO   = 	=    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5I ) )!("((* 5 "
	,
 9$$rW   c                      ddl m}  d }d}ddd}| j                  |||dt        j                         it        j                                ||fS )	z4
    Register a unary aggregate function (mean)
    r   rL   c                 R    t        j                  t        j                  |            S )N)r]   scalarnpnanmean)rS   rT   s     r4   funcz$unary_agg_func_fixture.<locals>.func$  s    yy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%/1H ""4#,#+'*BJJL$% $&::<% ?rW   c                      ddl m}  d }d}ddd}| j                  |||t        j                         t        j
                         dt        j
                                ||fS )z-
    Register a unary aggregate function
    r   rL   c                 n    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=  s6     	#C2::c?"C	#yy~rW   sum_meanzVarargs aggregaterX   )rT   y)r3   rM   ri   r]   r^   rj   rk   s       r4   varargs_agg_func_fixtureru   6  sf    
 & I.24H ""4#,#+,.HHJ,.JJL$% $&::<% ?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<	HX	"H]	HY	HW	HY	HUO HX	HY	%%)H!"	HX	(HUO
	'HTN	+HV	 H[,`  ( 
 
, i % !%( i  !0 i  !  		  		  		  		  		  		   HZ   		  		  		  		  		  		  		s    	G 
	G 	G* 	G6 (	H 2	H <	H 	H+ 	H7 	I $I 0I <I' 	I3 GGG'&G'*G32G36G?>G?H
HHH
H('H(+H43H47I ?I IIIII$#I$'I0/I03I<;I<