ÍÆV  ã@sádZddlZddlZddlZddlZddlZddlZddlm Z dd„Z yddl m Z WnJe k rÊyddl mZ Wn!e k rÅddd „Z YnXYnXd dd d „ZdS) z Helper for testing. éN)Ú PY3_OR_LATERcs.tj‰dd‡fdd†}|t_dS)z& Redirect all warnings to stdout. Nrcs)ˆ||tjj|ƒ|tjƒdS)N)ÚosÚpathÚbasenameÚsysÚstdout)ÚmsgÚcatÚfnameÚlnoÚfileÚline)Úshowwarning_orig©ú[/s/python-3.5.2/amd64_ubu14/lib/python3.5/site-packages/sklearn/externals/joblib/testing.pyÚ showwarningsz'warnings_to_stdout..showwarning)Úwarningsr)rr)rrÚwarnings_to_stdouts r)Úassert_raises_regex)Úassert_raises_regexpcOsŸd}y|||Žd}Wn_tk r~}z?t|ƒ}tj|ƒj|ƒsltd||fƒ‚WYdd}~XnX|r›td||ƒƒ‚dS)z;Helper function to check for message patterns in exceptionsFTz3Error message should match pattern %r. %r does not.NzShould have raised %r)Ú ExceptionÚstrÚreÚcompileÚsearchÚAssertionError)Úexpected_exceptionÚexpected_regexpÚ callable_objÚargsÚkwargsZ not_raisedÚeÚ error_messagerrrr#s   #réc stj|dtjdtjƒ‰‡fdd†}tj||ƒ}z©|jƒˆjƒ\}}trƒ|jƒ|jƒ}}ˆj dkr¶dj ˆj ||ƒ}t |ƒ‚|dk rít j ||ƒ rít dj ||ƒƒ‚Wd|jƒXdS) zƒRuns a command in a subprocess with timeout in seconds. Also checks returncode is zero and stdout if stdout_regex is set. rÚstderrcsˆjƒdS)N)Úkillr)ÚprocrrÚ kill_process>sz+check_subprocess_call..kill_processrz2Non-zero return code: {0}. Stdout: {1} Stderr: {2}Nz0Unexpected output: '{0!r}' does not match: {1!r})Ú subprocessÚPopenÚPIPEÚ threadingÚTimerÚstartÚ communicaterÚdecodeÚ returncodeÚformatÚ ValueErrorrrÚcancel)ÚcmdÚtimeoutZ stdout_regexr'Ztimerrr$Úmessager)r&rÚcheck_subprocess_call6s&    r7)Ú__doc__rrÚos.pathrrr(r+Z sklearn.externals.joblib._compatrrZ nose.toolsrÚ ImportErrorrr7rrrrÚs