3
Xd                 @   s    d dl mZ dd Zdd ZdS )   )ANTSc              C   s`  t t ddt ddt dgdt dddt ddd	t dd
t dgdt dd
t ddgddt dddt dddt ddd	t dddt dddt dgdt ddddt ddgdt ddt dgdt dgdt dddt dddt dgdt dddt dddd} tj }xRt| j D ]B\}}x6t|j D ]&\}}t|j | ||ks,tq,W qW d S )Nz%s)argstrnumber_of_time_steps)requiresz%d   )r   positionT)Znohash
usedefault)	mandatorytransformation_modelmetric)r	   r   r   z--MI-option %sx)r   sep)r   r	   z --number-of-affine-iterations %sz--number-of-iterations %sgradient_step_lengthz--output-naming %s)r   r	   r   )r	   r   regularizationz--gaussian-smoothing-sigmas %sz--subsampling-factors %s
delta_time)r   r   )Zaffine_gradient_descent_optionargsr   Z	dimensionenvironZfixed_imager   r   Zmetric_weightZ	mi_optionZmoving_imagenum_threadsZnumber_of_affine_iterationsZnumber_of_iterationsr   Zoutput_transform_prefixZradiusr   Z&regularization_deformation_field_sigmaZ#regularization_gradient_field_sigmaZsmoothing_sigmasZsubsampling_factorsZsymmetry_typer
   Zuse_histogram_matching)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/ants/tests/test_auto_ANTS.pytest_ANTS_inputs   s    r!   c              C   s   t 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s\tq\W qFW d S )N)
extensions)Zaffine_transformZinverse_warp_transformZ
metaheaderZmetaheader_rawZwarp_transform)r   r   Zoutput_specr   r   r   r   r   )Z
output_mapoutputsr   r   r   r   r   r   r    test_ANTS_outputsj   s    r$   N)Zregistrationr   r!   r$   r   r   r   r    <module>   s   e