3
Xdw                 @   s    d dl mZ dd Zdd ZdS )   )DTIc              C   s   t t ddt d ddt d ddt d ddt d dt  d} tj }xLt| j D ]<\}}x2t|j D ]"\}}t|j | ||ksftqfW qPW d S )NT)Z
usedefault)
extensions	mandatory)r   )Zb0_thresZin_bvalZin_bvecin_fileZ	mask_fileZ
out_prefix)dictr   Z
input_speclistitemsgetattrtraitsAssertionError)Z	input_mapinputskeymetadatametakeyvalue r   \/var/www/html/virt/lib/python3.6/site-packages/nipype/interfaces/dipy/tests/test_auto_DTI.pytest_DTI_inputs   s&    
r   c           	   C   s   t t d dt d dt d dt d dt d dt d dd} tj }xLt| j D ]<\}}x2t|j D ]"\}}t|j | ||ksdtqdW qNW d S )N)r   )Zad_fileZcolor_fa_fileZfa_fileZmd_fileout_fileZrd_file)r   r   Zoutput_specr   r   r	   r
   r   )Z
output_mapoutputsr   r   r   r   r   r   r   test_DTI_outputs"   s"    r   N)Ztensorsr   r   r   r   r   r   r   <module>   s   