
    y-Ph&                        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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	            ed)<   n(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 d0lm Z  ded#<   n# e$ r Y nw xY w	 d d1l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d2 Z# e j$        d3          d4             Z% e j$        d3          d5             Z& e j$        d67          d8             Z' e j$        d67          d9             Z( e j$        d67          d:             Z)dS );    N)Codec)fs)is_threading_enabled)windows_has_tzdata)"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(   
emscriptenwin32z/usr/share/zoneinfo)AzureFileSystem)GcsFileSystemS3FileSystem)HadoopFileSystemc                    |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)collection_pathconfigdoctest_groupsgrouppyarrowr.   s         P/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/pyarrow/conftest.pypytest_ignore_collectrA      s   }# #O 4 4444
 
 
 $ 	  	 E""5))S-A-AAA  44'3+?+???01 tS1111####u   tt 3////333333u   tt v}.66 O 4 4444)S-A-AAA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 )Nr1   Ftmpdir)r<   r2   r3   r:   getfixturevalueas_cwd)request	doctest_m	doctest_crD   s       r@   _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 )	Nr1   Fr   zpyarrow-fs-example.dats   datalocal
local_pathpath)	r<   r2   r3   r:   r   LocalFileSystemopen_output_streamr4   write)doctest_namespacerG   tmp_pathrH   rI   rL   rN   streams           r@   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     r@   unary_functionz*unary_func_fixture.<locals>.unary_function?  s-    1v,/O   = = 	=    zy=x+1zadd functionztest add functionsummarydescriptionarray)r?   rZ   register_scalar_functionpaint64)rc   	func_name	unary_docrb   s      @r@   unary_func_fixturern   8  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     r@   funcz$unary_agg_func_fixture.<locals>.funcU  s    yA'''rd   mean_udfzy=avg(x)zfind mean of xre   ra   )r?   rZ   r   register_aggregate_functionrj   float64rb   rt   rl   func_docrs   s       @r@   unary_agg_func_fixturerz   M  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       r@   rt   z&varargs_agg_func_fixture.<locals>.funco  s;     	# 	#C2::c??"CCy~~rd   sum_meanzVarargs aggregatere   )ra   y)r?   rZ   r   rv   rj   rk   rw   rx   s       @r@   varargs_agg_func_fixturer   g  s    
 &%%%%%     I.24 4H ""4#,#+,.HJJ,.JLL$% $% $&:<<% % % ?rd   )*pytestosr?   rj   r   r   pyarrow.libr   pyarrow.tests.utilr   sysgroupsis_availabler6   platformrN   existsr   r8   r   pyarrow.gandivapyarrow.aceropyarrow.datasetpyarrow.orcr   r   pyarrow.parquetpyarrow.parquet.encryptionpyarrow.flightr9   r+   r,   r.   r/   pyarrow.substraitrA   fixturerJ   rU   rn   rz   r    rd   r@   <module>r      s  $  				                 , , , , , , 1 1 1 1 1 1 



#
 #
 #
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<7 2 2 4 4H_\\!! "/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	
, , ,`   ( 
 
 
, i   % % ! %( i     ! 2 i     !   s   	E E! E!%	E/ /E76E7;	F FF	F F#"F#'	F1 1F98F9= G G&%G&*	G4 4
H H	H 
HH 	H* *H21H26	I   II	I II"I. .I65I6:J JJJ J&%J&*J6 6J>=J>	K KK