o $if&@sddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl m Z m Z ddlmZddlmZe jjddde jjd d de jjd d e jjd e jejdd dejd<ejdkruejddZerueeddZGdddZddZddZejddZ ejdddd Z!ejd!d"d#d$Z"ejddd%d&Z#ejddd'd(Z$d;d,d-Z%ejddd.d/Z&ejddd0d1Z'ejddd2d3Z(ejd4d5gdd6d7d8Z)ejddd9d4Z*ejddd:d5Z+dS)<N)groupsdefaults)set_timezone_db_pathfind_free_portcii) max_examplesdev2debug )r verbosityHYPOTHESIS_PROFILEz /dev/nullAWS_CONFIG_FILEwin32PYARROW_TZDATA_PATHcCshd dd}tD]*}|d|t|}|jd|d|d|d|jd|dd d |dqdS) NcSsLtj|}|s |S|}|dvrdS|dvrdStd||)N>1yonyestrueT>0nnoofffalseFz {}={} is not parsable as boolean)osenvirongetupperlower ValueErrorformat)namedefaultvaluer'O/var/www/html/corbot_env/lib/python3.10/site-packages/pyarrow/tests/conftest.pybool_env;s z"pytest_addoption..bool_envzPYARROW_TEST_{} --enable-{} store_truezEnable the {} test group)actionr%help --disable-{}FzDisable the {} test groupN)rr#r addoption)parserr)groupr%r'r'r(pytest_addoption9s    r3c@s$eZdZddZddZddZdS) PyArrowConfigcCs i|_dSr/) is_enabled)selfr'r'r(__init__Ss zPyArrowConfig.__init__cCs |j}|tvr||dSdSr/)r$rrequires)r6markr2r'r'r( apply_markVszPyArrowConfig.apply_markcCs"|j|std|dSdS)Nz{} NOT enabled)r5pytestskipr#)r6r2r'r'r(r8[s zPyArrowConfig.requiresN)__name__ __module__ __qualname__r7r:r8r'r'r'r(r4Rs r4cCsXt|_tD]#}|d|d|}d|}||o"|| }||jj|<qdS)Nmarkersr*r.)r4pyarrowraddinivalue_liner# getoptionr5)configr9 enable_flag disable_flagr5r'r'r(pytest_configure`s   rGcCs |D] }|jj|qdSr/) iter_markersrDrAr:)itemr9r'r'r(pytest_runtest_setupqs rJcCs t|jSr/)pathlibPathstrpath)tmpdirr'r'r(tempdirws rOsession)scopecCsttjdS)Ndata)rKrL__file__parentr'r'r'r( base_datadir}srUT)autousecCs|dddS)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) monkeypatchr'r'r(disable_aws_metadatasrZcCs8tjdd}ttjdd}tjdd}|||fS)NARROW_HDFS_TEST_HOSTr%ARROW_HDFS_TEST_PORTrARROW_HDFS_TEST_USERhdfs)rrrint)hostportuserr'r'r(hdfs_connections rccCs dt}}d\}}||||fS)N localhost)arrow apachearrowr)r`ra access_key secret_keyr'r'r( s3_connections  ri?csfdd}|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"tfdd}|S)Nc s~}}|dkr=z|i|WSty8}z|d8}|}|9}r)t|}t|WYd}~nd}~ww|dks|)Nrrl) Exceptionmintimesleep)argskwargsremaining_attempts curr_delayerrlast_exception)attemptsbackoffdelayfunc max_delayr'r(wrappers   z(retry..decorate..wrapper) functoolswraps)rzr|rwrxryr{)rzr(decorateszretry..decorater')rwryr{rxrr'rr(retrysrc cstdddddd}|}|\}}}}d||}tj} | ||dd d d d d ||g} d} z4z tj| | d} Wnt yLt dYn w|||| |dVW| durg| | dSdS| durv| | ww)Ng?r)rwryrxcSs(tjd|d}|dksJdS)Nzhttp://z/minio/health/cluster)urllibrequesturlopengetcode)addressrespr'r'r(minio_server_health_checksz,s3_server..minio_server_health_checkz{}:{})MINIO_ACCESS_KEYMINIO_SECRET_KEYminioz--compatserverz--quietz --addressenvz!`minio` command cannot be located connectionprocessrO)r getbasetempr#rrcopyupdate subprocessPopenOSErrorr;r<killwait) ritmpdir_factoryrrNr`rargrhrrrqprocr'r'r( s3_serversB          rc cst}tj}tjdddt|g}d}zUzddl}tj ||d}| dur1t d|dWnt tfyQ}zt d|d|WYd}~n d}~wwd |f|d VW|duri||dSdS|durx||ww) Nz-m testbenchz--portrrCommand # did not start server successfully! failed to execute: rd)rr)rrrrsys executablestrrrrpollr;r<ModuleNotFoundErrorrrr)rarrqrrer'r'r( gcs_servers4  "   rc cst}tj}|}dd|dt|g}d}zTztj||d}|dur0t d|dWnt t fyP}zt d|d|WYd}~nd}~wwd|d d f||d VW|durk| |dSdS|durz| |ww) Nz azurite-blobz --locationz --blobPortrrrrz 127.0.0.1devstoreaccount1zXEby8vdM02xNOcqFlqUwJPLlmEtlCDXJ1OUzFT50uSRZ6IFsuFq2UVErCz4I6tq/K1SZFPTOtr/KBHBeksoGMGw==r)rrrrrrrrrr;r<rrrr)rrarrNrqrrr'r'r( azure_servers:  "    rbuiltin_pickle cloudpickle)paramsrQcCs ||jSr/)getfixturevalueparam)rr'r'r( pickle_module"s rcCs ddl}|S)Nrpicklerr'r'r(r-scCs td}d|jvr|j|_|S)NrHIGHEST_PROTOCOL)r; importorskip__dict__DEFAULT_PROTOCOLr)cpr'r'r(r3s  )rjrkNrl),r}rrKrrrourllib.requestrr; hypothesishconftestrrrAr pyarrow.utilrsettingsregister_profile Verbosityverbose load_profilerrplatformtzdata_set_pathr3r4rGrJfixturerOrUrZrcrirrrrrrrr'r'r'r(sn               #  &