3
Xd                 @   s    d dl mZ dd Zdd ZdS )   )DenoiseImagec              C   s   t t ddt ddt dddt dd ddt d ddd	gd
dt dddt dddt dd ddd	gddt dddgdt dddt ddd} tj }xLt| j D ]<\}}x2t|j D ]"\}}t|j | ||kstqW qW d S )Nz%s)argstrz-d %dT)Znohash
usedefaultz-i %s)r   
extensions	mandatoryFinput_imagez%s_noise)r   
hash_fileskeep_extensionname_sourcename_templatez-n %s)r   r   z-o %sz%s_noise_corrected)r   r   r   r	   r
   r   noise_image)r   r   xorz-s %sz-v)argsZ	dimensionenvironr   r   Znoise_modelnum_threadsoutput_imageZ
save_noiseZshrink_factorverbose)dictr   Z
input_speclistitemsgetattrtraitsAssertionError)Z	input_mapinputskeymetadatametakeyvalue r   e/var/www/html/virt/lib/python3.6/site-packages/nipype/interfaces/ants/tests/test_auto_DenoiseImage.pytest_DenoiseImage_inputs   sX    r    c              C   sr   t t d dt d dd} tj }xLt| j D ]<\}}x2t|j D ]"\}}t|j | ||ksDtqDW q.W d S )N)r   )r   r   )r   r   Zoutput_specr   r   r   r   r   )Z
output_mapoutputsr   r   r   r   r   r   r   test_DenoiseImage_outputsA   s    r"   N)Zsegmentationr   r    r"   r   r   r   r   <module>   s   <