a xdj @sdZddlZddlZddlZddlZddlZddlZddlZda da ddZ ddZ ej ddZd d Zd d Zd dZddZdS)a! The functions in this module are meant to run on a separate worker process. Exception: in single process mode _execute is called directly. For efficiency, we copy all data needed to execute all tests into each worker and store it in global variables. This reduces the cost of each task. NcCs|a|attjtjdS)z=Copy data shared by all test executions into worker processesN) _lit_config_parallelism_semaphoressignalSIGINTSIG_IGN) lit_configZparallelism_semaphoresrS/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/lit/worker.py initializesr cCs@t|t|t}Wdn1s(0Y|||S)zRun one test in a multiprocessing.Pool Side effects in this function and functions it calls are not visible in the main lit process. Arguments and results of this function are pickled, so they should be cheap to copy. N)_get_parallelism_semaphore_executerZ setResult)testresultrrr execute#s ( rccs dVdSNrrrrr NopSemaphore4srcCs&|jj}t|r||}t|tSr)configZparallelism_groupcallablergetr)r Zpgrrr r 9sr cCs4t}t||}t||_||_t|_|Sr)time_execute_test_handle_errorselapsedstartosgetpidpid)r rrrrrr r As   r cCs^z|jj||}t|WS|jr*d}|t7}|d7}tj tjj |YS0dS)Nz#Exception during script execution:  ) rZ test_formatr _adapt_resultdebug traceback format_exclitTestResultZ UNRESOLVED)r rroutputrrr rJs  rcCs6t|tjjr|St|ts J|\}}tj||Sr) isinstancer!r"r#tuple)rcoder$rrr rYs r)__doc__ contextlibrrrrZlit.Testr!Zlit.utilrrr rcontextmanagerrr r rrrrrr s"