o
    rri^                     @   sX   d dl Z d dlZd dlZd dlZd dlZddlmZ dd Zdd Z	dd	 Z
d
d ZdS )    N   )get_mainc                 K   s   t jg d|  fddi|S )N)doraz-Pzdora.tests.integcheckT)sprun)argvkwargs r
   O/home/ubuntu/.local/lib/python3.10/site-packages/dora/tests/test_integration.pyrun_cmd   s   r   c                 C   s   t | tjd< ttj tg d W d    n1 sw   Y  tdg tdg tg d tg d tg d t| }|	dg}td	|j
gd
d}t|j |j r_J tdg|jd |j snJ d S )N_DORA_TEST_TMPDIR)info--za=32r   r   )gridtestz	--dry_runz--initz--no_monitor)r   r   z--a=32)z--main_moduleother_trainr   z--b=4exportT)capture_outputimport)input)strosenvironpytestraisesr   SubprocessErrorr   r   get_xpsigshutilrmtreefolderexistsstdout)tmpdirmainxpprocr
   r
   r   test_integration   s    

r(   c              	   C   s   t | tjd< dtjd< z0t| }|g }|j}tdg | r$J tddg | s0J tddg W dtjd< ndtjd< w |  tddg | sSJ d S )Nr   1_DORA_GIT_SAVEr   z
--git_save0)	r   r   r   r   r   code_folderr   r"   unlink)r$   r%   r&   coder
   r
   r   test_git_save(   s   


r/   c                 C   s@   t | tjd< ddlm} d|_tt|}||u sJ d S )Nr   r   )r%   zdora.tests.integ.train.main)	r   r   r   integ.trainr%   
_full_namepickleloadsdumps)r$   r%   otherr
   r
   r   test_pickle?   s
   r6   )r   r2   
subprocessr   r   r   	test_mainr   r   r(   r/   r6   r
   r
   r
   r   <module>   s   