3
Xd                 @   s    d dl mZ dd Zdd ZdS )   )	ImageMathc              C   s   t t dddt ddt ddddt dddt dddt dd ddd
t dddt ddd	dt dd ddgdddd	} tj }xLt| j D ]<\}}x2t|j D ]"\}}t|j | ||kstqW qW d S )Nz%s   )argstrpositionT)
usedefaultz%d)r   r   r   )Znohashr      )r   
extensions	mandatoryr   r   )r   r	   r   op1z%s_maths)r   r   Zkeep_extensionZname_sourceZname_templater   )	argsZcopy_headerZ	dimensionenvironnum_threadsr
   Zop2Z	operationoutput_image)dictr   Z
input_speclistitemsgetattrtraitsAssertionError)Z	input_mapinputskeymetadatametakeyvalue r   b/var/www/html/virt/lib/python3.6/site-packages/nipype/interfaces/ants/tests/test_auto_ImageMath.pytest_ImageMath_inputs   sN    r   c              C   sj   t t d dd} tj }xLt| j D ]<\}}x2t|j D ]"\}}t|j | ||ks<tq<W q&W d S )N)r   )r   )r   r   Zoutput_specr   r   r   r   r   )Z
output_mapoutputsr   r   r   r   r   r   r   test_ImageMath_outputs:   s    r!   N)utilsr   r   r!   r   r   r   r   <module>   s   5