U md @sdZddlZddlZddlZddlZddlZddlZddlZddl Z ej Z ej Z e j j ZejjZejZejjZejjZejjZejZddZdddZdS) z Helper for testing. Ncstjdfdd }|t_dS)z& Redirect all warnings to stdout. Nrcs||tj||tjdS)N)ospathbasenamesysstdout)msgcatfnamelnofilelineZshowwarning_origG/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/joblib/testing.py showwarning sz'warnings_to_stdout..showwarning)Nr)warningsr)rrr rwarnings_to_stdoutsrc stjtjtjdfdd}fdd}z|dk rft||}|td||}|\}} || }} j dkrd j || } t | |dk rt ||st d |||dk rt || st d || W5|dk r ||XdS) 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. )rstderrcstddS)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_process4sz0check_subprocess_call..terminate_processcstddS)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=sz+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}) subprocessPopenPIPEcancel threadingTimerstart communicatedecode returncodeformat ValueErrorresearch) rtimeoutZ stdout_regexZ stderr_regexrrZterminate_timerZ kill_timerrrmessagerrrcheck_subprocess_call'sT         r/)rNN)__doc__rros.pathrr+rr#ZpytestZ_pytestZraisesZwarnsrunnerZSkippedZSkipTestmarkZskipifZfixtureZ parametrizer-Zxfailparamrr/rrrrs*