U
    ºmœd

  ã                   @   sL   d dl Z d dlZd dlmZmZ dd„ Zdd„ Zdd„ Zd	d
„ Zdd„ Z	dS )é    N)ÚraisesÚcheck_subprocess_callc                  C   s:   d  ddddg¡} ttjd| gƒ ttjd| gdd� d S )	NÚ
zresult = 1 + 2 * 3zprint(result)zmy_list = [1, 2, 3]zprint(my_list)ú-cz7\s{1,2}\[1, 2, 3\]©Zstdout_regex)Újoinr   ÚsysÚ
executable)Úcode© r   úQ/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/joblib/test/test_testing.pyÚtest_check_subprocess_call   s    ýþr   c               	   C   sD   d} d}t tƒ�}ttjd| g|d� W 5 Q R X | d |¡¡ d S )NZ42Z_no_way_this_matches_anything_r   r   zUnexpected stdout.+{})r   Ú
ValueErrorr   r   r	   ÚmatchÚformat)r
   Znon_matching_patternÚexcinfor   r   r   Ú-test_check_subprocess_call_non_matching_regex   s    
ÿr   c               	   C   s&   d} t tƒ� t| gƒ W 5 Q R X d S )NZ_a_command_that_does_not_exist_)r   ÚOSErrorr   )Zwrong_commandr   r   r   Ú(test_check_subprocess_call_wrong_command   s    
r   c               	   C   sR   d  ddddg¡} t dtj¡}ttƒ�}ttjd| gƒ W 5 Q R X | 	|¡ d S )Nr   ú
import syszprint("writing on stdout")z%sys.stderr.write("writing on stderr")zsys.exit(123)zONon-zero return code: 123.+Stdout:
writing on stdout.+Stderr:
writing on stderrr   ©
r   ÚreÚcompileÚDOTALLr   r   r   r   r	   r   )Zcode_with_non_zero_exitÚpatternr   r   r   r   Ú/test_check_subprocess_call_non_zero_return_code%   s    üþ
r   c                  C   s`   d  dddddddd	dg	¡} t d
tj¡}ttƒ�}ttjd| gdd� W 5 Q R X | 	|¡ d S )Nr   zimport timer   zprint("before sleep on stdout")zsys.stdout.flush()z*sys.stderr.write("before sleep on stderr")zsys.stderr.flush()ztime.sleep(10)z-print("process should have be killed before")zVNon-zero return code:.+Stdout:
before sleep on stdout\s+Stderr:
before sleep on stderrr   é   )Útimeoutr   )Zcode_timing_outr   r   r   r   r   Ú"test_check_subprocess_call_timeout5   s&    õý
ÿr   )
r   r   Zjoblib.testingr   r   r   r   r   r   r   r   r   r   r   Ú<module>   s   
