3
Xd                 @   s    d dl mZ dd Zdd ZdS )   )CreateMatrixc              C   s   t t ddt d ddt d ddt d ddt d ddt d ddt d ddt d ddt d ddt d ddt d ddd} tj }xLt| j D ]<\}}x2t|j D ]"\}}t|j | ||kstqW qW d S )NT)
usedefault)
extensionsZgenfile)r   r   )r   	mandatory)Zcount_region_intersectionsZout_endpoint_array_nameZ$out_fiber_length_std_matrix_mat_fileZ out_intersection_matrix_mat_fileZout_matrix_fileZout_matrix_mat_fileZ%out_mean_fiber_length_matrix_mat_fileZ'out_median_fiber_length_matrix_mat_fileZresolution_network_fileZroi_fileZ
tract_file)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/cmtk/tests/test_auto_CreateMatrix.pytest_CreateMatrix_inputs   sJ    r   c              C   s   t t d dt d dt d dt d dt d dt d dt  t d dt d dt d dt d dt  t d dt  t d dt d dt d dt d dd} tj }xLt| j D ]<\}}x2t|j D ]"\}}t|j | ||kstqW qW d S )N)r   )Zendpoint_fileZendpoint_file_mmZfiber_label_fileZfiber_labels_noorphansZfiber_length_fileZ fiber_length_std_matrix_mat_fileZfiltered_tractographiesZfiltered_tractographyZ&filtered_tractography_by_intersectionsZintersection_matrix_fileZintersection_matrix_mat_fileZmatlab_matrix_filesZmatrix_fileZmatrix_filesZmatrix_mat_fileZ!mean_fiber_length_matrix_mat_fileZ#median_fiber_length_matrix_mat_fileZ
stats_file)r   r   Zoutput_specr   r   r	   r
   r   )Z
output_mapoutputsr   r   r   r   r   r   r   test_CreateMatrix_outputs:   sL    r   N)Zcmtkr   r   r   r   r   r   r   <module>   s   5