o Zh!@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$d9d,d-Z%ejddd.d/Z&ejddd0d1Z'ejd2d3gdd4d5d6Z(ejddd7d2Z)ejddd8d3Z*dS):N)groupsdefaults)set_timezone_db_pathfind_free_portcii) max_examplesdev2debug )r verbosityZHYPOTHESIS_PROFILEz /dev/nullZAWS_CONFIG_FILEwin32ZPYARROW_TZDATA_PATHcCshd dd}tD]*}|d|t|}|jd|d|d|d|jd|dd d |dqdS) NcSsLtj|}|s |S|}|dvrdS|dvrdStd||)N>true1onyyesT>nfalseno0offFz {}={} is not parsable as boolean)osenvirongetupperlower ValueErrorformat)namedefaultvaluer$M/var/www/html/lang_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 rZ addoption)parserr&groupr"r$r$r%pytest_addoption9s    r/c@s$eZdZddZddZddZdS) PyArrowConfigcCs i|_dSr,) is_enabled)selfr$r$r%__init__Ss zPyArrowConfig.__init__cCs |j}|tvr||dSdSr,)r!rrequires)r2markr.r$r$r% apply_markVszPyArrowConfig.apply_markcCs"|j|std|dSdS)Nz{} NOT enabled)r1pytestskipr )r2r.r$r$r%r4[s zPyArrowConfig.requiresN)__name__ __module__ __qualname__r3r6r4r$r$r$r%r0Rs r0cCsXt|_tD]#}|d|d|}d|}||o"|| }||jj|<qdS)Nmarkersr'r+)r0pyarrowrZaddinivalue_liner Z getoptionr1)configr5Z enable_flagZ disable_flagr1r$r$r%pytest_configure`s   r?cCs |D] }|jj|qdSr,)Z iter_markersr>r=r6)itemr5r$r$r%pytest_runtest_setupqs rAcCs t|jSr,)pathlibPathZstrpath)tmpdirr$r$r%tempdirws rEsession)scopecCsttjdS)Ndata)rBrC__file__parentr$r$r$r% base_datadir}srKT)ZautousecCs|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. ZAWS_EC2_METADATA_DISABLEDrN)Zsetenv)Z monkeypatchr$r$r%disable_aws_metadatasrLcCs8tjdd}ttjdd}tjdd}|||fS)NZARROW_HDFS_TEST_HOSTr"ZARROW_HDFS_TEST_PORTrZARROW_HDFS_TEST_USERZhdfs)rrrint)hostportuserr$r$r%hdfs_connections rQcCs dt}}d\}}||||fS)N localhost)arrowZ apachearrowr)rNrO access_key secret_keyr$r$r% s3_connections  rV?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|)NrrY) Exceptionmintimesleep)argskwargsZremaining_attemptsZ curr_delayerrZlast_exception)attemptsbackoffdelayfunc max_delayr$r%wrappers   z(retry..decorate..wrapper) functoolswraps)rdrfrarbrcre)rdr%decorateszretry..decorater$)rarcrerbrjr$rir%retrysrkc cstdddddd}|}|\}}}}d||}tj} | ||dd d d d d ||g} d} z4z tj| | d} Wnt yLt dYn w|||| |dVW| durg| | dSdS| durv| | ww)Ng?r)rarcrbcSs(tjd|d}|dksJdS)Nzhttp://z/minio/health/cluster)urllibrequesturlopengetcode)addressrespr$r$r%minio_server_health_checksz,s3_server..minio_server_health_checkz{}:{})ZMINIO_ACCESS_KEYZMINIO_SECRET_KEYZminioz--compatserverz--quietz --addressenvz!`minio` command cannot be located) connectionprocessrE)rkZ getbasetempr rrcopyupdate subprocessPopenOSErrorr7r8killwait) rVZtmpdir_factoryrtrDrNrOrTrUrrrwr^procr$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--portrrvzCommand z# did not start server successfully!z failed to execute: rR)rxry)rrrrzsys executablestrrr|r}pollr7r8ModuleNotFoundErrorr~rr)rOrwr^rrer$r$r% gcs_servers4  "   rbuiltin_pickle cloudpickle)paramsrGcCs ||jSr,)Zgetfixturevalueparam)ror$r$r% pickle_modules rcCs ddl}|S)Nrpicklerr$r$r%rscCs td}d|jvr|j|_|S)NrHIGHEST_PROTOCOL)r7Z importorskip__dict__DEFAULT_PROTOCOLr)cpr$r$r%rs  )rWrXNrY)+rgrrBr|rr\urllib.requestrnr7Z hypothesishZconftestrrr=rZ pyarrow.utilrsettingsZregister_profileZ VerbosityverboseZ load_profilerrplatformZtzdata_set_pathr/r0r?rAZfixturerErKrLrQrVrkrrrrrr$r$r$r%sj               #  &