YjtHddlZddlmcmZddlZddlZddlm Z ddl Z ddl m Z m Z ddlmZddlmZddlmZe rddlmZddl mZdd lmZdd lmZd ZddZe j dddZddZGddZGddeZ GddeZ!dS) N) TYPE_CHECKING)create_terminal_writerhookimpl) TestReport) algorithms)ensure_ipynb_compatibility)nodes)Config)Parser)ExitCodeiXparserr returnc R|d}|dddd|ddd tjtjd  |d d t d|ddt d|ddtdtj dtj |dddddS)z) Declare pytest-split's options. zSplit tests into groups which execution time is about the same. Run with --store-durations to store information about test execution times.z--store-durationsstore_durations store_truez(Store durations into '--durations-path'.)destactionhelpz--durations-pathdurations_pathzsPath to the file in which durations are (to be) stored, default is .test_durations in the current working directoryz.test_durations)rrdefaultz--splitssplitsz,The number of groups to split the tests into)rtyperz--groupgroupz;The group of tests that should be executed (first one is 1)z--splitting-algorithmsplitting_algorithmz,Algorithm used to split the tests. Choices: duration_based_chunks)rrrrchoicesz--clean-durationsclean_durationszpRemoves the test duration info for tests which are not present while running the suite with '--store-durations'.N) getgroup addoptionospathjoingetcwdintstrr Algorithmsnames)r rs Z/home/ubuntu/.hermes/hermes-agent/venv/lib/python3.11/site-packages/pytest_split/plugin.pypytest_addoptionr)sh OO V  E OO  7   OO  J RY[[*;<< OO   ;   OO   J   OO "  [J= 1z'argument `--group` must be >= 1 and <= ) getoptionpytest UsageError)r,rrs r(pytest_cmdline_mainr3Ms   W % %E   h ' 'F ~%-t B%- @AAA C ABBB zz BCCC qyyEFNN R& R RSSS 4r*c|jjr4|jjr(|jt |d|jjr*|jt|ddSdS)z% Enable the plugins we need. pytestsplitpluginpytestsplitcachepluginN)optionrr pluginmanagerregisterPytestSplitPluginrPytestSplitCachePlugin)r,s r(pytest_configurer<gs}V 3V%%&7&?&?ATUUU }$ %% "6 * *,D       r*ceZdZddZdS)Baser,r rNc||_t|j|_ t|jj5}t j||_ dddn #1swxYwYn#t$r i|_ YnwxYwt|j trt|j |_ dSdS)z Load durations and set up a terminal writer. This logic is shared for both the split- and cache plugin. N)r,rwriteropenr7rjsonloadsreadcached_durationsFileNotFoundError isinstancelistdict)selfr,fs r(__init__z Base.__init__us  ,T[99  'fm233 =q(, 16688(<(<% = = = = = = = = = = = = = = =  ' ' '$&D ! ! ! ' d+T 2 2 @$()>$?$?D ! ! ! @ @s4A?,A3' A?3A77A?:A7;A??BBr,r rN)__name__ __module__ __qualname__rLr*r(r>r>ts.@@@@@@r*r>cXeZdZd fd Zed d d ZxZS) r:r,r ct||js6|jd}|j|dSdS)Nz [pytest-split] No test durations found. Pytest-split will split tests evenly when no durations are found. [pytest-split] You can expect better results in consequent runs, when test timings have been documented. )superrLrEr@markupline)rJr,message __class__s r(rLzPytestSplitPlugin.__init__sj    $ &k((<G K  W % % % % % & &r*T)trylastitemslist[nodes.Item]rNc ,|jj}|jj}tj|jjj}||||j}||dz }t|||j |dd<|j |j |j |j d|jj|j |j d|d|d|jdd dS) zU Collect and select the tests we want to run, and deselect the rest. r/N)rZz1 [pytest-split] Splitting tests with algorithm: z[pytest-split] Running group /z (estimated duration: z.2fzs) )r7rrrr&rvaluerErselectedhookpytest_deselected deselectedr@rVrUduration)rJr,rZr group_idxalgogroupsrs r(pytest_collection_modifyitemsz/PytestSplitPlugin.pytest_collection_modifyitemss+m*, $V]%FGMfeT%:;;y1}%"5%000>aaa %%E,<%===  K  ifmFgii      K  r rrFrrZ_Zhrrrr       r*)r,r )r,r rZr[rN)rNrOrPrLrrg __classcell__)rXs@r(r:r:s & & & & & &Xd  '9         r*r:ceZdZdZddZdS)r;zB The cache plugin writes durations to our durations file. rNc"|jjd}i}|jD]o}|D]j}t |t rS|jdkr#|jdvr|jtkr=|j |vr d||j <||j xx|jz cc<kp|jj j rt||_n$|D]\}}||j|<t!|jj jd5}t%j|j|dddddn #1swxYwY|jd |jj j}|j|dS) z Method is called by Pytest after the test-suite has run. https://github.com/pytest-dev/pytest/blob/main/src/_pytest/main.py#L308 terminalreporterr)teardownsetupwT) sort_keysindentNz* [pytest-split] Stored test durations in )r,r8 get_pluginstatsvaluesrGrrcwhen,STORE_DURATIONS_SETUP_AND_TEARDOWN_THRESHOLDnodeidr7rrIrErZrArrBdumpr@rUrV) rJterminal_reportertest_durations test_reports test_reportkvrKrWs r(pytest_sessionfinishz+PytestSplitCachePlugin.pytest_sessionfinishs !K5@@ASTT+--3::<< O OL+ O O k:66O"+a// #(,AAA'0FGG!#)??=>{'9:";#5666+:NN666! O$ ;  - -$($8$8D ! !&,,.. - -1+,%a(( $+$3S 9 9 JQ Id+Q$q I I I I J J J J J J J J J J J J J J J+$$ ^4;;M;\ ^ ^   !!!!!sD>>EE)rN)rNrOrP__doc__rrQr*r(r;r;s2'"'"'"'"'"'"r*r;)r r rN)r,r rr-rM)"builtins @py_builtins_pytest.assertion.rewrite assertionrewrite @pytest_arrBr typingrr1_pytest.configrr_pytest.reportsr pytest_splitr pytest_split.ipynb_compatibilityr_pytestr r _pytest.config.argparsingr _pytest.mainr rvr)r3r<r>r:r;rQr*r(rs ;;;;;;;;&&&&&&######GGGGGG&%%%%%%000000%%%%%%07,3333l$ 2     @@@@@@@@.) ) ) ) ) ) ) ) X,",",",","T,",",",","r*