3
Xd              	   @   s  d dl Z d dljjZd dlmZ d dlmZmZ d dlZej	j
e dddd Zej	j
e dddd	 Zd
d Zej	j
e dddd Zej	j
e dddd Zej	j
e ddej	jdi ddifddiddifddddddfdddddifgdd ZdS )    N)InterfaceResult)	check_fslno_fslzfsl is not installed)reasonc              C   s&   t jj } | jddd j s"td S )N.   r   )fslInfoversionsplitisdigitAssertionError)ver r   W/var/www/html/virt/lib/python3.6/site-packages/nipype/interfaces/fsl/tests/test_base.pytest_fslversion   s    
r   c              C   s*   t tjjj } tjj }|| ks&td S )N)listr   r	   ftypeskeysoutput_typer   )typesZorig_out_typer   r   r   test_fsloutputtype   s    
r   c           	   C   sX   x0t jjj D ] \} }t jj| }||kstqW tjt t jjd W d Q R X d S )NZJUNK)	r   r	   r   itemsZoutput_type_to_extr   pytestZraisesKeyError)Zftypeextresr   r   r   test_outputtype_to_ext   s
    r   c              C   s(   t jdd} | j }t|tks$td S )NZls)command)r   
FSLCommandruntyper   r   )cmdr   r   r   r   test_FSLCommand#   s    r#   c              C   s   t jdd} | jt jj ks t| jjd | jks6t| jt jjksHtt j} t jdd}xFt jjD ]:}| j	| | j|kst|t jj krd|jj|ksdtqdW d S )Njunk)r   ZFSLOUTPUTTYPE)
r   r   Z_output_typer	   r   r   inputsenvironr   Zset_default_output_type)r"   ZcmdinstZout_typer   r   r   test_FSLCommand2,   s    
r'   zargs, desired_namefilez
foo.nii.gzsuffixZ_brainzfoo_brain.nii.gzz/data)r)   cwd)dirr(   z
_brain.matF)r)   Z
change_extzfoo_brain.matc             C   sh   t jddd}tj }|jd| }d|j krFtjj|d |d }ntjj||d }||ksdtd S )Nr$   ZNIFTI_GZ)r   r   
foo.nii.gzr+   r(   )r,   )	r   r   osgetcwdZ
_gen_fnamer   pathjoinr   )argsZdesired_namer"   pthfnameZdesiredr   r   r   test_gen_fname>   s    r4   )r-   Znipype.interfaces.fslZ
interfacesr   Znipype.interfaces.baser   r   r   r   markZskipifr   r   r   r#   r'   Zparametrizer4   r   r   r   r   <module>   s"   		

