K i(ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl m Z m Z ddlmZddlmZe j$j'dde j$j'd d e j$j'd d e j(j* e j$j-ej.j1dd dej.d<ej2dk(r&ej.j1ddZereeej2dk(r\ej.j1dd@ddlmZ ej:j=ej>ddej.d<dZ"GddZ#dZ$dZ%ejLdZ'ejLdd Z(ejLd!"d#Z)ejLdd$Z*ejLdd%Z+d0d&Z,ejLdd'Z-ejLdd(Z.ejLdd)Z/ejLd*d+gd,d-Z0ejLdd.Z1ejLdd/Z2y#e $r e!dY7wxYw)1N)groupsdefaults)set_timezone_db_pathfind_free_portcii) max_examplesdev2debug )r verbosityHYPOTHESIS_PROFILEz /dev/nullAWS_CONFIG_FILEwin32PYARROW_TZDATA_PATHTZDIR) resourcestzdatazoneinfozCPackage "tzdata" not found. Not setting TZDIR environment variable.c d d}tD]M}|d|t|}|jd|d|d|d|jd|dd d |dOy) Nctjj|j}|s|S|j }|dvry|dvryt |jd|d)N>1yonyestrueT>0nnoofffalseF=z is not parsable as boolean)osenvirongetupperlower ValueError)namedefaultvalues \/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/pyarrow/tests/conftest.pybool_envz"pytest_addoption..bool_envLsd tzz|,N  3 3 6 6 ~Qug5PQR R PYARROW_TEST_ --enable- store_truez Enable the z test group)actionr,help --disable-Fz Disable the N)rr addoption)parserr/groupr,s r.pytest_addoptionr;Js SC]5'2HUOD9UG, ,g!,UG;?  B :eW- ,e!-eWK@  C Cr0ceZdZdZdZdZy) PyArrowConfigci|_yr7) is_enabled)selfs r.__init__zPyArrowConfig.__init__cs r0cP|j}|tvr|j|yyr7)r+rrequires)r@markr:s r. apply_markzPyArrowConfig.apply_markfs#  F? MM%  r0cT|j|stj|dyy)Nz NOT enabled)r?pytestskip)r@r:s r.rCzPyArrowConfig.requiresks&u% KK5'. /&r0N)__name__ __module__ __qualname__rArErCr0r.r=r=bs! 0r0r=ct|_tD]\}|jd|d|}d|}|j |xr|j | }||jj |<^y)Nmarkersr2r6)r=pyarrowraddinivalue_line getoptionr?)configrD enable_flag disable_flagr?s r.pytest_configurerUps"_FN 5 t "$( #D6* &&{35&&|495 *4!!$' 5r0cx|jD]'}|jjj|)yr7) iter_markersrRrOrE)itemrDs r.pytest_runtest_setuprYs2!!#- &&t,-r0c@tj|jSr7)pathlibPathstrpath)tmpdirs r.tempdirr_s << ''r0session)scopecNtjtjdz S)Ndata)r[r\__file__parentrLr0r. base_datadirrfs << ! ( (6 11r0T)autousec(|jddy)zStop the AWS SDK from trying to contact the EC2 metadata server. Otherwise, this causes a 5 second delay in tests that exercise the S3 filesystem. AWS_EC2_METADATA_DISABLEDrN)setenv) monkeypatchs r.disable_aws_metadatarls2F;r0ctjjdd}ttjjdd}tjjdd}|||fS)NARROW_HDFS_TEST_HOSTr,ARROW_HDFS_TEST_PORTrARROW_HDFS_TEST_USERhdfs)r%r&r'int)hostportusers r.hdfs_connectionrvsS ::>>0) >0& 9D t r0c0dt}}d\}}||||fS)N 127.0.0.1)arrow apachearrowr)rsrt access_key secret_keys r. s3_connectionr}s'n.$D3J z: --r0cfd}|S)aP Retry decorator Parameters ---------- attempts : int, default 3 The number of attempts. delay : float, default 1 Initial delay in seconds. max_delay : float, optional The max delay between attempts. backoff : float, default 1 The multiplier to delay after each attempt. cLtjfd}|S)Nc}}|dkDr  |i|S#t$r9}|dz}|}|z} r t| }tj|Yd}~nd}~wwxYw|dkDrVN)Nr) Exceptionmintimesleep) argskwargsremaining_attempts curr_delayerrlast_exceptionattemptsbackoffdelayfunc max_delays r.wrapperz(retry..decorate..wrappers!) J$q(+000! !+&!+&%(N')J %(Y%? JJz** +%q(s A/AA) functoolswraps)rrrrrrs` r.decoratezretry..decorates%   !  !r0rL)rrrrrs```` r.retryrs" Or0c#Ktdddd}|j}|\}}}}|d|}tjj } | j ||ddd d d d ||g} d} t j| | } |||| |d| !| j| jyy#t$rtjdYDwxYw#| !| j| jwwxYww)Nrr)rrrcttjjd|d}|jdk(sJy)Nzhttp://z/minio/health/live)urllibrequesturlopengetcode)addressresps r.minio_server_health_checkz,s3_server..minio_server_health_checks4~~%%y8J&KL||~$$$r0:)MINIO_ACCESS_KEYMINIO_SECRET_KEYminioz--compatserverz--quietz --addressenv connectionprocessr_z!`minio` command cannot be located) r getbasetempr%r&copyupdate subprocessPopenOSErrorrGrHkillwait) r}tmpdir_factoryrr^rsrtr{r|rrrprocs r. s3_serverrs& AQ*%+% ' ' )F)6&D$ JavG **// CJJ&& Z9k V D D#. "'*(    IIK IIK  9 789   IIK IIK s<A0D 3B> C"$D >CC"CC""%DD c# Kt}tjj}d}|dt |g}d} t j ||}|jtjd|dd|f|d|!|j|jyy#t$r%}tjd|d|Yd}~Md}~wwxYw#|!|j|jwwxYww) Nzstorage-testbenchz--portrCommand # did not start server successfully! localhost)rr failed to execute: ) rr%r&rstrrrpollrGrHrrr)rtrexerres r. gcs_serverrs  D **// C C 3t9 %D D#. 99; " KK(4&(KL M '-    IIK IIK  > htf$8<==>   IIK IIK sA;DAB+> C$D+ C4CCCC%DDc#JKt}tjj}|j }dd|dt |g}|dgz }d} t j||}|jtjd|dd|d d f||d |!|j|jyy#ttf$r%}tjd|d |Yd}~Sd}~wwxYw#|!|j|jwwxYww) Nz azurite-blobz --locationz --blobPortz--skipApiVersionCheckrrrrxdevstoreaccount1zXEby8vdM02xNOcqFlqUwJPLlmEtlCDXJ1OUzFT50uSRZ6IFsuFq2UVErCz4I6tq/K1SZFPTOtr/KBHBeksoGMGw==rr)rr%r&rrrrrrrGrHModuleNotFoundErrorrrr)rrtrr^rrrs r. azure_serverrs(  D **// C  ' ' )F L&,D JD $ %%D D#. 99; " KK(4&(KL M'.@CD     IIK IIK   )> htf$8<==>   IIK IIK sCAD#AC C; $D#C8C3.C;3C88C;;%D  D#builtin_pickle cloudpickle)paramsrac8|j|jSr7)getfixturevalueparam)rs r. pickle_moduler4s  " "7== 11r0cddl}|S)Nrpicklers r.rr?s  Mr0cntjd}d|jvr|j|_|S)NrHIGHEST_PROTOCOL)rG importorskip__dict__DEFAULT_PROTOCOLr)cps r.rrEs1   ] +B, 11 Ir0)g?Nr)3rr%r[rsysrurllib.requestrrG hypothesishconftestrrrOr pyarrow.utilrsettingsregister_profile Verbosityverbose load_profiler&r'platformtzdata_set_path importlibrpathjoinfilesrprintr;r=rUrYfixturer_rfrlrvr}rrrrrrrrLr0r.rs$   '(' Dt4 E3 G"&'kk&9&9;  ';UCD!, <<7jjnn%:DAO_-<<7 zz~~gt$,' "$'',,yx/H*"UBJJw C0 0 05"- (( i 2!2<<i !i .!. Fi #!#Li !2i !B   22i ! i !E#  U  s9J00 KK