
    LgS&                        S SK r S SKrS SKrS SKJr  S SKJr  S SKJr  S SKJ	r	  S SK
r
/ SQr0 SS_S	S_S
\R                  " S
5      _S\R                  " S5      _SS_SS_SS_SS_SS_SS_SS_S\R                  " S5      _SS_SS_SS_S\R                  " S5      _SS_0 SS_SS_SS_SS_SS_SS_S S_S!S_S"S_S#S_S$S_S%\R                  " S%5      _S&S_S'S_S(\" 5       _S)S_S*\R                  " S*5      _Er\
R                  S+:X  a  S\S'   S\S!'   S\S&'   \
R                  S,:X  a  \	" 5       \S)'   O.\
R                  S+:X  a  \R                  R!                  S-5      \S)'    S SKrS\S'    S SKrS\S'    S SKrS\S'    S SKrS\S'    S SKrS\S'    S SKr\
R                  S,:X  a  S\S'   O	\S)   \S'     S SKrS\S'    S SKrS\S'    S SKrS\S'    S SKrS\S '    S SKrS\S'    S S.KJr  S\S	'    S S/KJr  S\S'    S S0KJ r   S\S#'    S S1KJ!r!  S\S'    S SK"rS\S''   S2 r#\ RH                  " SS39S4 5       r%\ RH                  " SS39S5 5       r&\ RH                  " S6S79S8 5       r'\ RH                  " S6S79S9 5       r(\ RH                  " S6S79S: 5       r)g! \ a     GN3f = f! \ a     GN6f = f! \ a     GN9f = f! \ a     GN<f = f! \ a     GN?f = f! \ a     GN(f = f! \ a	    S\S'    GN0f = f! \ a	    S\S'    GN8f = f! \ a     GN;f = f! \ a     GN>f = f! \ a     GNAf = f! \ a     GNBf = f! \ a     GNCf = f! \ a     GNDf = f! \ a     GNEf = f! \ a     GNHf = f);    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                     UR                   R                  (       a  S[        U 5      ;   a  g/ SQnU H3  nSR                  U5      [        U 5      ;   d  M#  [        U   (       a  M3    g   S[        U 5      ;   a  [        S   (       d  gS[        U 5      ;   a   SS Kng	S
[        U 5      ;   a   SSKJn  g	[        UR                   SS	5      (       a   S[        U 5      ;   a  gS[        U 5      ;   a  gg	! [         a     gf = f! [         a     gf = f)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.12/site-packages/pyarrow/conftest.pypytest_ignore_collectrA      s    }}##D	)
 $E""5)SY6 $
 (3t9401SY&# 3t9$3 v}}.66D	))SY6!    s$   C0 ,D  0
C=<C= 
DD)autousec              #   2  #    U R                   R                  R                  n[        U R                   R                  SS5      nU(       d  U(       a/  U R	                  S5      nUR                  5          S v   S S S 5        g S v   g ! , (       d  f       g = f7f)Nr1   Ftmpdir)r<   r2   r3   r:   getfixturevalueas_cwd)request	doctest_m	doctest_crD   s       r@   _docdirrJ     sm      %%44I--/?GII ((2 ]]_ _ 		 _s   A1B3B8B
BBc              #     #    UR                   R                  R                  n[        UR                   R                  SS5      nU(       d  U(       aw  [        U S'   [        R
                  " 5       nUS-  nUR                  [        U5      5       nUR                  S5        S S S 5        XPS'   [        U5      U S'   [        U5      U S'   S v   g ! , (       d  f       N3= f7f)	Nr1   Fr   zpyarrow-fs-example.dats   datalocal
local_pathr;   )	r<   r2   r3   r:   r   LocalFileSystemopen_output_streamr4   write)doctest_namespacerG   tmp_pathrH   rI   rL   r;   streams           r@   add_fsrT   !  s      %%44I--/?GII"$$ ""$22%%c$i0FLL! 1%*'"*-h-,'$'I&!	 10s   BCC!-C
CCsession)scopec                     ^ SSK Jm  U4S jn SnSSS.nTR                  U UUS[        R                  " 5       0[        R                  " 5       5        X4$ )	z#
Register a unary scalar function.
r   computec                 >   > TR                  SUS/U R                  S9$ )Nadd   )memory_pool)call_functionr]   )ctxxpcs     r@   unary_function*unary_func_fixture.<locals>.unary_function?  s*    1v,/OO   = 	=    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_fixturern   8  sY    
 &= I* 35I ) )!("((* 5 "
	,
 $$rd   c                     ^ SSK Jn   SSKmU4S jnSnSSS.nU R                  UUUS	[        R
                  " 5       0[        R
                  " 5       5        X4$ )
z,
Register a unary aggregate function (mean)
r   rX   Nc                 N   > [         R                  " TR                  U5      5      $ )N)rj   scalarnanmean)r_   r`   nps     r@   func$unary_agg_func_fixture.<locals>.funcU  s    yyA''rd   mean_udfzy=avg(x)zfind mean of xre   r`   )r?   rY   r   register_aggregate_functionrj   float64ra   rt   rl   func_docrs   s       @r@   unary_agg_func_fixturer{   M  sa    
 &( I%/1H ""4#,#+'*BJJL$% $&::<% ?rd   c                     ^ SSK Jn   SSKmU4S jnSnSSS.nU R                  UUU[        R
                  " 5       [        R                  " 5       S.[        R                  " 5       5        X4$ )	z%
Register a unary aggregate function
r   rX   Nc                 n   > SnU H  nUTR                  U5      -  nM     [        R                  " U5      $ )Ng        )rr   rj   rq   )r_   argssumargrs   s       r@   rt   &varargs_agg_func_fixture.<locals>.funco  s2    C2::c?"C yy~rd   sum_meanzVarargs aggregatere   )r`   y)r?   rY   r   rw   rj   rk   rx   ry   s       @r@   varargs_agg_func_fixturer   g  sh    
 & I.24H ""4#,#+,.HHJ,.JJL$% $&::<% ?rd   )*pytestosr?   rj   r   r   pyarrow.libr   pyarrow.tests.utilr   sysgroupsis_availabler6   platformr;   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   rT   rn   r{   r    rd   r@   <module>r      s  $  	    , 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 4H_\\\! "/D EH_	HX	"H]	HY	HW	HY
	
||w
 #?3 HXHW	HY	%%)H!"	HX	*HW	(HUO	'HTN	+HV	 H[,`  ( 
 
, i % !%( i  !2 i  !c  		  		  		  		  		  		   HZ   HY  		  		  		  		  		  		  		  		s&  %	J /	J% 9	J2 	J? 	K K 1K ;	K& 	K8 	L
 	L #	L$ -L1 9L> M M 	M% J"!J"%J/.J/2J<;J<?K	K	KKK#"K#&K54K58LL
LLL! L!$L.-L.1L;:L;>MMMMM"!M"%M/.M/