o Gh(@s\ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl m Z m Z ddlmZddlmZe jjddde jjd d de jjd d e jjd e jejdd dejd<ejdkruejddZerueeejdkrejdddurddlmZzejeddejd<Wn e ye!dYnwddZ"GdddZ#ddZ$ddZ%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@d1d2Z,ej&d"d#d3d4Z-ej&d"d#d5d6Z.ej&d"d#d7d8Z/ej&d9d:gd"d;dd9Z1ej&d"d#d?d:Z2dS)AN)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.cCsld dd}tD],}|d|t|}|jd|d|d|dd|jd |dd d |ddqdS) NcSsPtj|}|s |S|}|dvrdS|dvrdSt|d|d)N>1yonyestrueT>0nnoofffalseF=z is not parsable as boolean)osenvirongetupperlower ValueError)namedefaultvaluer+W/var/www/html/Persson_Maskin/env/lib/python3.10/site-packages/pyarrow/tests/conftest.pybool_envLsz"pytest_addoption..bool_env PYARROW_TEST_ --enable- store_truez Enable the z test group)actionr)help --disable-Fz Disable the N)rr addoption)parserr-groupr)r+r+r,pytest_addoptionJs      r8c@s$eZdZddZddZddZdS) PyArrowConfigcCs i|_dSr4) is_enabled)selfr+r+r,__init__cs zPyArrowConfig.__init__cCs |j}|tvr||dSdSr4)r(rrequires)r;markr7r+r+r, apply_markfszPyArrowConfig.apply_markcCs"|j|st|ddSdS)Nz NOT enabled)r:pytestskip)r;r7r+r+r,r=ks zPyArrowConfig.requiresN)__name__ __module__ __qualname__r<r?r=r+r+r+r,r9bs r9cCsXt|_tD]#}|d|d|}d|}||o"|| }||jj|<qdS)Nmarkersr/r3)r9pyarrowraddinivalue_line getoptionr:)configr> enable_flag disable_flagr:r+r+r,pytest_configureps   rLcCs |D] }|jj|qdSr4) iter_markersrIrFr?)itemr>r+r+r,pytest_runtest_setups rOcCs t|jSr4)pathlibPathstrpath)tmpdirr+r+r,tempdirs rTsession)scopecCsttjdS)Ndata)rPrQ__file__parentr+r+r+r, base_datadirsrZT)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_metadatasr_cCs8tjdd}ttjdd}tjdd}|||fS)NARROW_HDFS_TEST_HOSTr)ARROW_HDFS_TEST_PORTrARROW_HDFS_TEST_USERhdfs)r"r#r$int)hostportuserr+r+r,hdfs_connections rhcCs dt}}d\}}||||fS)N 127.0.0.1)arrow apachearrowr)rerf access_key secret_keyr+r+r, s3_connections  rn?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|)Nrrq) Exceptionmintimesleep)argskwargsremaining_attempts curr_delayerrlast_exception)attemptsbackoffdelayfunc max_delayr+r,wrappers   z(retry..decorate..wrapper) functoolswraps)rrr|r}r~r)rr,decorateszretry..decorater+)r|r~rr}rr+rr,retrysrc cstdddddd}|}|\}}}}|d|}tj} | ||dd d d d d ||g} d} z4z tj| | d} WntyMt dYn w|||| |dVW| durh| | dSdS| durw| | ww)Nrqr)r|r~r}cSs(tjd|d}|dksJdS)Nzhttp://z/minio/health/live)urllibrequesturlopengetcode)addressrespr+r+r,minio_server_health_checksz,s3_server..minio_server_health_check:)MINIO_ACCESS_KEYMINIO_SECRET_KEYminioz--compatserverz--quietz --addressenvz!`minio` command cannot be located connectionprocessrT) r getbasetempr"r#copyupdate subprocessPopenOSErrorr@rAkillwait) rntmpdir_factoryrrSrerfrlrmrrrvprocr+r+r, s3_serversB         rc cst}tj}d}|dt|g}d}zOztj||d}|dur,t d|dWnt yJ}zt d|d|WYd}~n d}~wwd|f|dVW|durb| | dSdS|durq| | ww) Nzstorage-testbenchz--portrCommand # did not start server successfully! failed to execute: localhost)rr) rr"r#rstrrrpollr@rArrr)rfrexervrer+r+r, gcs_servers4  "   rc cst}tj}|}dd|dt|g}|dg7}d}zTztj||d}|dur5t d|dWnt t fyU}zt d|d|WYd}~nd}~wwd |d d f||d VW|durp| |dSdS|dur| |ww) Nz azurite-blobz --locationz --blobPortz--skipApiVersionCheckrrrrridevstoreaccount1zXEby8vdM02xNOcqFlqUwJPLlmEtlCDXJ1OUzFT50uSRZ6IFsuFq2UVErCz4I6tq/K1SZFPTOtr/KBHBeksoGMGw==r)rr"r#rrrrrrr@rAModuleNotFoundErrorrrr)rrfrrSrvrrr+r+r, azure_servers<   "    rbuiltin_pickle cloudpickle)paramsrVcCs ||jSr4)getfixturevalueparam)rr+r+r, pickle_module4s rcCs ddl}|S)Nrpicklerr+r+r,r?scCs td}d|jvr|j|_|S)NrHIGHEST_PROTOCOL)r@ importorskip__dict__DEFAULT_PROTOCOLr)cpr+r+r,rEs  )rorpNrq)3rr"rPrsysrturllib.requestrr@ hypothesishconftestrrrFr pyarrow.utilrsettingsregister_profile Verbosityverbose load_profiler#r$platformtzdata_set_path importlibrpathjoinfilesrprintr8r9rLrOfixturerTrZr_rhrnrrrrrrrr+r+r+r,s                  #  &   !