
    Pi&                        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 ]&}d| t          |           v rt          |         s dS 'd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defaultspyarrow.cudaImportError
pyarrow.fsr.   getattr)collection_pathconfigdoctest_groupsgrouppyarrowr.   s         d/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/pyarrow/conftest.pypytest_ignore_collectr@      s   }# #O 4 4444
 
 
 $ 	  	 E!%!!S%9%999  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(B0 0
B>=B>)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   r9   getfixturevalueas_cwd)request	doctest_m	doctest_crC   s       r?   _docdirrI     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   r9   r   LocalFileSystemopen_output_streamr4   write)doctest_namespacerF   tmp_pathrG   rH   rK   rM   streams           r?   add_fsrT   !  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>   rY   register_scalar_functionpaint64)rb   	func_name	unary_docra   s      @r?   unary_func_fixturerm   8  s    
 &%%%%%= = = = = I* 35 5I ) )!("(** 5 "

	, , ,
 9$$rc   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   rX   Nc                 R    t          j                            |                    S )N)ri   scalarnanmean)r_   r`   nps     r?   funcz$unary_agg_func_fixture.<locals>.funcU  s    yA'''rc   mean_udfzy=avg(x)zfind mean of xrd   r`   )r>   rY   r   register_aggregate_functionri   float64ra   rs   rk   func_docrr   s       @r?   unary_agg_func_fixturery   M  s    
 &%%%%%( ( ( ( ( I%/1 1H ""4#,#+'*BJLL$% $&:<<% % % ?rc   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   rX   Nc                 j    d}|D ]}|                     |          z  }t          j        |          S )Ng        )rq   ri   rp   )r_   argssumargrr   s       r?   rs   z&varargs_agg_func_fixture.<locals>.funco  s;     	# 	#C2::c??"CCy~~rc   sum_meanzVarargs aggregaterd   )r`   y)r>   rY   r   ru   ri   rj   rv   rw   s       @r?   varargs_agg_func_fixturer   g  s    
 &%%%%%     I.24 4H ""4#,#+,.HJJ,.JLL$% $% $&:<<% % % ?rc   )*pytestosr>   ri   r   r   pyarrow.libr   pyarrow.tests.utilr   sysgroupsis_availabler5   platformrM   existsr   r7   r   pyarrow.gandivapyarrow.aceropyarrow.datasetpyarrow.orcr   r   pyarrow.parquetpyarrow.parquet.encryptionpyarrow.flightr8   r+   r,   r.   r/   pyarrow.substraitr@   fixturerI   rT   rm   ry   r    rc   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