o `^h @sdZddlZddlZddlZddlZddlZddlZddlZddl Z ej Z ej Z e j j ZejjZejZejjZejjZejjZejZddZ  dddZdS) z Helper for testing. Ncstjdfdd }|t_dS)z& Redirect all warnings to stdout. Nrcs||tj||tjdS)N)ospathbasenamesysstdout)msgcatfnamelnofilelineshowwarning_origL/home/air/shanriGPT/back/venv/lib/python3.10/site-packages/joblib/testing.py showwarning sz'warnings_to_stdout..showwarning)Nr)warningsr)rrr rwarnings_to_stdouts rc s,tjtjtjdfdd}fdd}zo|dur3t||}|td||}|\}} || }} jdkrSd j|| } t | |duret ||set d |||durwt || swt d || W|dur| | dSdS|dur| | ww) aRuns a command in a subprocess with timeout in seconds. A SIGTERM is sent after `timeout` and if it does not terminate, a SIGKILL is sent after `2 * timeout`. Also checks returncode is zero, stdout if stdout_regex is set, and stderr if stderr_regex is set. )rstderrctddS)z Attempt to terminate a leftover process spawned during test execution: ideally this should not be needed but can help avoid clogging the CI workers in case of deadlocks. Timeout running N)rwarn terminatercmdprocrrterminate_process4 z0check_subprocess_call..terminate_processcr)z Kill a leftover process spawned during test execution: ideally this should not be needed but can help avoid clogging the CI workers in case of deadlocks. rN)rrkillrrrr kill_process=rz+check_subprocess_call..kill_processNrz/Non-zero return code: {}. Stdout: {} Stderr: {}z,Unexpected stdout: {!r} does not match: {!r}z,Unexpected stderr: {!r} does not match: {!r}) subprocessPopenPIPE threadingTimerstart communicatedecode returncodeformat ValueErrorresearchcancel) rtimeout stdout_regex stderr_regexrr terminate_timer kill_timerrrmessagerrrcheck_subprocess_call'sX           r6)rNN)__doc__rros.pathrr-r"r%pytest_pytestraiseswarnsrunnerSkippedSkipTestmarkskipiffixture parametrizer0xfailparamrr6rrrrs,