a 4ja6@sTddlmZddlZddlZddlmZddlZddlZddlZe ej Z ej dddddZ d d d d d Zej dddddZej dddddZej dddddddZdddddZej ddddddd Zd!dd"d#d$Zd!dd"d%d&ZeZd!d d"d'd(Zejd)d*d+d d,d-d.Zejd)d*d+d d,d/d0ZdS)1) annotationsN)Literalz pytest.Configint)configreturnc s,tjd}|r8z t|WSty6tdYn0z ddl}WntyVYn*0|j j dk}|j |dpv| }|r|Sz"ddlm ddfd d }WnXtytjd d krYd Szddlm }Wntyddl m }Yn0Yn0z |}WntyYdS0|r(|SdS)NZPYTEST_XDIST_AUTO_NUM_WORKERSzHPYTEST_XDIST_AUTO_NUM_WORKERS is not a number: {env_var!r}. Ignoring it.rlogical)rsched_getaffinityr)rcs tdS)Nr)lenrr 2E:\soccer\ml_env\Lib\site-packages\xdist/plugin.py cpu_count&sz0pytest_xdist_auto_num_workers..cpu_countZTRAVIStrue)r )osenvirongetr ValueErrorwarningswarnpsutil ImportErroroption numprocessesr r multiprocessingNotImplementedError)renv_varrZ use_logicalcountr nr rr pytest_xdist_auto_num_workerss>           r strz int | Literal['auto', 'logical'])srcCs |dvr |S|durt|SdS)Nautor)r)r"r r r parse_numprocesses8sr%z pytest.ParserNone)parserrc CsN|dd}|jdddddtdd|jd d d dtd d|jd ddd dd|jdddgddddd|jdddddd|jddddd |jd!d"d#gd$d%d&|jd'd(d#gd$d)d&|jd*dd+d,d-d.|jd/d#gd0d1d2|jd3d#gd4d5d2|jd6dd7d8|jd9dtd:d;|jdd?|jd@dAd>d?|jdBd>dCdDdS)ENxdistz"distributed and subprocess testingz-nz--numprocessesrstorea$Shortcut for '--dist=load --tx=NUM*popen'. With 'logical', attempt to detect logical CPU count (requires psutil, falls back to 'auto'). With 'auto', attempt to detect physical CPU count. If physical CPU count cannot be determined, falls back to 1. Forced to 0 (disabled) when used with --pdb.)destmetavaractiontypehelpz--maxprocesses maxprocesseszkLimit the maximum number of workers to process the tests when using --numprocesses with 'auto' or 'logical'z--max-worker-restartZmaxworkerrestartzbMaximum number of workers that can be restarted when crashed (set to zero to disable this feature))r,defaultr*r.z--distZdistmode)ZeachloadZ loadscopeZloadfileZ loadgroupZ workstealnodistr2a{Set mode for distributing tests to exec environments. each: Send each test to all available environments. load: Load balance by sending any pending test to any available environment. loadscope: Load balance by sending pending groups of tests in the same scope to any available environment. loadfile: Load balance by sending test grouped by file to any available environment. loadgroup: Like 'load', but sends tests marked with 'xdist_group' to the same worker. worksteal: Split the test suite between available environments, then re-balance when any worker runs out of tests. (default) no: Run tests inprocess, don't distribute.)r+r,choicesr*r0r.z--loadscope-reorderZloadscopereorder store_trueTa Pytest-xdist will default reorder tests by number of tests per scope when used in conjunction with loadscope. This option will enable loadscope reorder which will improve the parallelism of the test suite. However, the partial order of tests might not be retained. )r*r,r0r.z--no-loadscope-reorder store_falsea6Pytest-xdist will default reorder tests by number of tests per scope when used in conjunction with loadscope. This option will disable loadscope reorder, and the partial order of tests can be retained. This is useful when pytest-xdist is used together with other plugins that specify tests in a specific order.)r*r,r.z--txtxappendZxspeczAdd a test execution environment. Some examples: --tx popen//python=python2.5 --tx socket=192.168.1.102:8888 --tx ssh=user@codespeak.net//chdir=testcache)r*r,r0r+r.z--pxZpxzAdd a proxy gateway to pass to test execution environments using `via`. Example: --px id=my_proxy//socket=192.168.1.102:8888 --tx 5*popen//via=my_proxyz-ddistloadFz/Load-balance tests. Shortcut for '--dist=load'.)r,r*r0r.z --rsyncdirDIRz-Add directory for rsyncing to remote tx nodes)r,r0r+r.z --rsyncignoreZGLOBz;Add expression for ignores when rsyncing to remote tx nodesz --testrunuidzProvide an identifier shared amongst all workers as the value of the 'testrun_uid' fixture. If not provided, 'testrun_uid' is filled with a new unique string on every test run.)r,r.z--maxschedchunkaMaximum number of tests scheduled in one step for --dist=load. Setting it to 1 will force pytest to send tests to workers one by one - might be useful for a small number of slow tests. Larger numbers will allow the scheduler to submit consecutive chunks of tests to workers - allows reusing fixtures. Due to implementation reasons, at least 2 tests are scheduled per worker at the start. Only later tests can be scheduled one by one. Unlimited if not set.)r,r-r. rsyncdirszFlist of (relative) paths to be rsynced for remote distributed testing.paths)r-Z rsyncignorez?list of (relative) glob-style paths to be ignored for rsyncing.looponfailrootsz=directories to check for changes. Default: current directory.)r-r.)ZgetgroupZ _addoptionr%Z addoptionrZaddini)r'groupr r r pytest_addoption?s         r?zpytest.PytestPluginManager) pluginmanagerrcCsddlm}||dS)Nr)newhooks)r(rAZ add_hookspecs)r@rAr r r pytest_addhookss rBT)ZtrylastcCsd}|d||drdSt|r^ddlm}||}|j|d|jd}|r^d|_| d dst| d rt d }| |d | d ds| drt d}| |d dS)Nzuxdist_group: specify group for tests should run in same session.in relation to one another. Provided by pytest-xdist.markers collectonlyr)DSessionZdsessionZterminalreporterFZ looponfailr=zThe --looponfail command line argument and looponfailroots config variable are deprecated. The loop-on-fail feature will be removed in pytest-xdist 4.0.rZrsyncdirr;zThe --rsyncdir command line argument and rsyncdirs config variable are deprecated. The rsync feature will be removed in pytest-xdist 4.0.) Zaddinivalue_linegetvalue_is_distribution_modeZxdist.dsessionrEr@registerZ getpluginZ showfspath getoptionZgetiniDeprecationWarningZissue_config_time_warning)rZ config_linerEsessiontrwarningr r r pytest_configures,     rNboolcCs|ddkot|dS)z Whether distribution mode is on.r3r2r7)rIrOrr r r rG+srG)ZtryfirstcCs|jjrd|j_|dd}|jjdvrT|r>d|j_d|j_n|jj|d}||j_|jjr|jjdkrpd|j_|jj}|jjrt||jj}dg||j_ |jjdkrd|j_g|j_ |j }|d st |r|rt d dS) Nr1usepdbFr#rr2rPpopenrDzG--pdb is incompatible with distributing tests; try using -n0 or -nauto.)rr9r3rIrhookr r/minr7rFrGpytestZ UsageError)rrQZ auto_num_cpusrvalr r r pytest_cmdline_main0s0     rWz&pytest.FixtureRequest | pytest.Session)request_or_sessionrcCs t|jdS)zReturn `True` if this is an xdist worker, `False` otherwise. :param request_or_session: the `pytest` `request` or `session` object workerinput)hasattrrrXr r r is_xdist_workerVsr\cCst| o|jjjdkS)zReturn `True` if this is the xdist controller, `False` otherwise. Note: this method also returns `False` when distribution has not been activated at all. :param request_or_session: the `pytest` `request` or `session` object r2)r\rrr3r[r r r is_xdist_controller`s  r]cCs$t|jdr|jjd}|SdSdS)a*Return the id of the current worker ('gw0', 'gw1', etc) or 'master' if running on the controller node. If not distributing tests (for example passing `-n0` or not passing `-n` at all) also return 'master'. :param request_or_session: the `pytest` `request` or `session` object rYworkeridZmasterN)rZrrY)rXr^r r r get_xdist_worker_idts  r_rK)scopezpytest.FixtureRequest)requestrcCst|S)zkReturn the id of the current worker ('gw0', 'gw1', etc) or 'master' if running on the master node. )r_)rar r r worker_idsrbcCs*t|jdr|jjd}|StjSdS)z)Return the unique id of the current test.rYZ testrunuidN)rZrrYuuiduuid4hex)raZ testrunidr r r testrun_uids  rf) __future__rrsystypingrrcrrUlistpathZ _sys_pathZhookimplr r%r?rBrNrGrWr\r]Zis_xdist_masterr_Zfixturerbrfr r r r s8   (9 & %