o Ûñh)ã@s\ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddlmZm Z ddlmZddlmZe jjddde jjd d de jjdde jjd e j ej dd ¡¡dejd<ejdkruej dd¡Zerueeƒejdkr¦ej dd¡dur¦ddlmZzej e d¡d¡ejd<Wn e y¥e!dƒYnwdd„Z"Gdd„dƒZ#dd„Z$dd„Z%ej&d d!„ƒZ'ej&d"d#d$d%„ƒZ(ej&d&d'd(d)„ƒZ)ej&d"d#d*d+„ƒZ*ej&d"d#d,d-„ƒZ+d@d1d2„Z,ej&d"d#d3d4„ƒZ-ej&d"d#d5d6„ƒZ.ej&d"d#d7d8„ƒZ/ej&d9d:gd"d;dd9„ƒZ1ej&d"d#d?d:„ƒZ2dS)AéNé)ÚgroupsÚdefaults)Úset_timezone_db_path©Úfind_free_portÚciiè)Úmax_examplesÚdevé2Údebugé )r Ú verbosityÚHYPOTHESIS_PROFILEz /dev/nullÚAWS_CONFIG_FILEÚwin32ÚPYARROW_TZDATA_PATHÚTZDIR)Ú resourcesÚtzdataÚzoneinfozCPackage "tzdata" not found. Not setting TZDIR environment variable.cCshddd„}tD]*}|d |¡t|ƒ}|jd |¡d|d |¡d|jd |¡dd d |¡dqdS)NcSsLtj | ¡¡}|s|S| ¡}|dvrdS|dvrdStd | ¡|¡ƒ‚)N>Ú1ÚyÚonÚyesÚtrueT>Ú0ÚnÚnoÚoffÚfalseFz {}={} is not parsable as boolean)ÚosÚenvironÚgetÚupperÚlowerÚ ValueErrorÚformat)ÚnameÚdefaultÚvalue©r+úJ/var/www/vscode/kcb/lib/python3.10/site-packages/pyarrow/tests/conftest.pyÚbool_envLsÿz"pytest_addoption..bool_envzPYARROW_TEST_{}ú--enable-{}Ú store_truezEnable the {} test group)Úactionr)Úhelpú--disable-{}FzDisable the {} test group©N)rr'rÚ addoption)Úparserr-Úgroupr)r+r+r,Úpytest_addoptionJs þþûr7c@s$eZdZdd„Zdd„Zdd„ZdS)Ú PyArrowConfigcCs i|_dSr3)Ú is_enabled)Úselfr+r+r,Ú__init__ds zPyArrowConfig.__init__cCs |j}|tvr| |¡dSdSr3)r(rÚrequires)r:Úmarkr6r+r+r,Ú apply_markgsÿzPyArrowConfig.apply_markcCs"|j|st d |¡¡dSdS)Nz{} NOT enabled)r9ÚpytestÚskipr')r:r6r+r+r,r<ls ÿzPyArrowConfig.requiresN)Ú__name__Ú __module__Ú__qualname__r;r>r<r+r+r+r,r8csr8cCsXtƒ|_tD]#}| d|¡d |¡}d |¡}| |¡o"| |¡}||jj|<qdS)NÚmarkersr.r2)r8ÚpyarrowrÚaddinivalue_liner'Ú getoptionr9)Úconfigr=Úenable_flagÚdisable_flagr9r+r+r,Úpytest_configureqsÿ ÿörKcCs | ¡D] }|jj |¡qdSr3)Úiter_markersrHrEr>)Úitemr=r+r+r,Úpytest_runtest_setup‚sÿrNcCst |j¡Sr3)ÚpathlibÚPathÚstrpath)Útmpdirr+r+r,ÚtempdirˆsrSÚsession)ÚscopecCst t¡jdS)NÚdata)rOrPÚ__file__Úparentr+r+r+r,Úbase_datadirŽsrYT)ÚautousecCs| dd¡dS)z Stop 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)Úmonkeypatchr+r+r,Údisable_aws_metadata“sr^cCs8tj dd¡}ttj dd¡ƒ}tj dd¡}|||fS)NÚARROW_HDFS_TEST_HOSTr)ÚARROW_HDFS_TEST_PORTrÚARROW_HDFS_TEST_USERÚhdfs)r!r"r#Úint)ÚhostÚportÚuserr+r+r,Úhdfs_connection s rgcCs dtƒ}}d\}}||||fS)Nú 127.0.0.1)ÚarrowÚapachearrowr)rdreÚ access_keyÚ secret_keyr+r+r,Ú s3_connection¨srméçð?écs‡‡‡‡fdd„}|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. cs"t ˆ¡‡‡‡‡‡fdd„ƒ}|S)Nc s~ˆ}ˆ}|dkr=zˆ|i|¤ŽWSty8}z|d8}|}|ˆ9}ˆr)t|ˆƒ}t |¡WYd}~nd}~ww|dks|‚)Nrrp)Ú ExceptionÚminÚtimeÚsleep)ÚargsÚkwargsÚremaining_attemptsÚ curr_delayÚerrÚlast_exception)ÚattemptsÚbackoffÚdelayÚfuncÚ max_delayr+r,Úwrapper¿s €úý z(retry..decorate..wrapper)Ú functoolsÚwraps)r~r€©r{r|r}r)r~r,Údecorate¾szretry..decorater+)r{r}rr|r„r+rƒr,Úretry¯sr…ccsîtdddddd„ƒ}| ¡}|\}}}}d ||¡}tj ¡} | ||dœ¡d d ddd ||g} d}z4z tj| | d}Wnt yLt d¡Ynw||ƒ|||dœVW|durg| ¡| ¡dSdS|durv| ¡| ¡ww)Nérpr)r{r}r|cSs(tj d|›d¡}| ¡dksJ‚dS)Nzhttp://z/minio/health/liveéÈ)ÚurllibÚrequestÚurlopenÚgetcode)ÚaddressÚrespr+r+r,Úminio_server_health_checkÔsz,s3_server..minio_server_health_checkz{}:{})ÚMINIO_ACCESS_KEYÚMINIO_SECRET_KEYÚminioz--compatÚserverz--quietz --address©Úenvz!`minio` command cannot be located©Ú connectionÚprocessrS)r…Úgetbasetempr'r!r"ÚcopyÚupdateÚ subprocessÚPopenÚOSErrorr?r@ÚkillÚwait)rmÚtmpdir_factoryrŽrRrdrerkrlrŒr”ruÚprocr+r+r,Ú s3_serverÒsB€ þ ÿÿý€þ þr¢c csätƒ}tj ¡}d}|dt|ƒg}d}zOztj||d}| ¡dur,t d|›d¡Wnt yJ}zt d|›d|›¡WYd}~n d}~wwd|f|dœVW|durb| ¡| ¡dSdS|durq| ¡| ¡ww) Nzstorage-testbenchz--portr“úCommand ú# did not start server successfully!ú failed to execute: Ú localhost)r–r—) rr!r"r™Ústrr›rœÚpollr?r@rržrŸ)rer”Úexerur¡Úer+r+r,Ú gcs_serverùs4€ €"€ÿþ€þ þr«c cstƒ}tj ¡}| ¡}dd|dt|ƒg}|dg7}d}zTztj||d}| ¡dur5t d|›d¡WnttfyU}zt d|›d|›¡WYd}~nd}~wwd |d df||dœVW|durp| ¡| ¡dSdS|dur| ¡| ¡ww) Nzazurite-blobz --locationz --blobPortz--skipApiVersionCheckr“r£r¤r¥rhÚdevstoreaccount1zXEby8vdM02xNOcqFlqUwJPLlmEtlCDXJ1OUzFT50uSRZ6IFsuFq2UVErCz4I6tq/K1SZFPTOtr/KBHBeksoGMGw==r•)rr!r"r™r˜r§r›rœr¨r?r@ÚModuleNotFoundErrorrržrŸ)r rer”rRrur¡rªr+r+r,Úazure_servers<€ €"€ÿÿù€ þ þr®Úbuiltin_pickleÚcloudpickle)ÚparamsrUcCs| |j¡Sr3)ÚgetfixturevalueÚparam)r‰r+r+r,Ú pickle_module5sr´cCsddl}|S)Nr©Úpicklerµr+r+r,r¯@scCs t d¡}d|jvr|j|_|S)Nr°ÚHIGHEST_PROTOCOL)r?ÚimportorskipÚ__dict__ÚDEFAULT_PROTOCOLr·)Úcpr+r+r,r°Fs )rnroNrp)3rr!rOr›ÚsysrsÚurllib.requestrˆr?Ú hypothesisÚhÚconftestrrrErÚpyarrow.utilrÚsettingsÚregister_profileÚ VerbosityÚverboseÚload_profiler"r#ÚplatformÚtzdata_set_pathÚ importlibrÚpathÚjoinÚfilesrÚprintr7r8rKrNÚfixturerSrYr^rgrmr…r¢r«r®r´r¯r°r+r+r+r,Ús‚ ÿ ÿÿ # & !þû