U -e@s^ddlZddlZddlZddlZddlmZddlZddlZddl m Z m Z ddl m Z ejjdddejjd d dejjd d ejjd ejejdd dejd<ddZGdddZddZddZejddZejddddZejdd d!d"Zejddd#d$Zejddd%d&Zejddd'd(Zejddd)d*Z dS)+N)TemporaryDirectory)groupsdefaultsfind_free_portcii) max_examplesdev2debug )r verbosityZHYPOTHESIS_PROFILEz /dev/nullZAWS_CONFIG_FILEcCshd dd}tD]T}|d|t|}|jd|d|d|d|jd|dd d |dqdS) NcSsPtj|}|s|S|}|dkr,dS|dkr8dStd||dS)N>trueyyes1onT>nofffalseno0Fz {}={} is not parsable as boolean)osenvirongetupperlower ValueErrorformat)namedefaultvaluer#W/var/www/html/Darija-Ai-Train/env/lib/python3.8/site-packages/pyarrow/tests/conftest.pybool_env2sz"pytest_addoption..bool_envzPYARROW_TEST_{} --enable-{} store_truezEnable the {} test group)actionr!help --disable-{}FzDisable the {} test group)N)rrrZ addoption)parserr%groupr!r#r#r$pytest_addoption0s   r-c@s$eZdZddZddZddZdS) PyArrowConfigcCs i|_dSN) is_enabled)selfr#r#r$__init__JszPyArrowConfig.__init__cCs|j}|tkr||dSr/)r rrequires)r1markr,r#r#r$ apply_markMszPyArrowConfig.apply_markcCs|j|std|dS)Nz{} NOT enabled)r0pytestskipr)r1r,r#r#r$r3Rs zPyArrowConfig.requiresN)__name__ __module__ __qualname__r2r5r3r#r#r#r$r.Isr.cCsXt|_tD]F}|d|d|}d|}||oD|| }||jj|<q dS)Nmarkersr&r*)r.pyarrowrZaddinivalue_linerZ getoptionr0)configr4Z enable_flagZ disable_flagr0r#r#r$pytest_configureWs   r>cCs |D]}|jj|qdSr/)Z iter_markersr=r<r5)itemr4r#r#r$pytest_runtest_setuphs r@cCs t|jSr/)pathlibPathZstrpath)Ztmpdirr#r#r$tempdirnsrCsession)scopecCsttjdS)Ndata)rArB__file__parentr#r#r#r$ base_datadirtsrIT)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_metadataysrJcCs8tjdd}ttjdd}tjdd}|||fS)NZARROW_HDFS_TEST_HOSTr!ZARROW_HDFS_TEST_PORTrZARROW_HDFS_TEST_USERZhdfs)rrrint)hostportuserr#r#r$hdfs_connectionsrOcCs dt}}d\}}||||fS)N localhost)arrowZ apachearrowr)rLrM access_key secret_keyr#r#r$ s3_connections rTc cs|\}}}}d||}tj}|||dt|}ddddd||g}d} zDztj ||d} Wnt k rt d YnX|| |d VW5| dk r| | XW5QRXdS) Nz{}:{})ZMINIO_ACCESS_KEYZMINIO_SECRET_KEYZminioz--compatserverz--quietz --addressenvz!`minio` command cannot be located) connectionprocessrC) rrrcopyupdaterkillwait subprocessPopenOSErrorr6r7) rTrLrMrRrSaddressrWrCargsprocr#r#r$ s3_servers0     rdc cst}tj}tjdddt|g}d}zz8ddl }t j ||d}| dk r`t d|dWn<ttfk r}zt d|d|W5d}~XYnXd |f|d VW5|dk r||XdS) Nz-m testbenchz--portrrVzCommand z# did not start server successfully!z failed to execute: rP)rXrY)rrrrZsys executablestrr\r]rer^r_pollr6r7ModuleNotFoundErrorr`)rMrWrbrcreer#r#r$ gcs_servers"  ( rl)!rrAr^rftempfilerr6Z hypothesishZconftestrrZ pyarrow.utilrsettingsZregister_profileZ VerbosityverboseZ load_profilerrr-r.r>r@ZfixturerCrIrJrOrTrdrlr#r#r#r$sB