3
Yd                 @   sv   d dl Z d dlZd dlmZ d dljjjZd dl	m
Z
 d dlmZmZ d dl	mZ dd Zdd	 Zd
d Zdd ZdS )    N)example_data)	isdefined)split_filenamefname_presuffix)
TraitErrorc        
      C   s   t dd} t dd}t dd}tjdd}||j_|jjdksBt| |j_t|jj	 s\tt
| \}}}t
|\}}}tjj|d||f }t|dd	}|jd }	|jj	|kst|jj|kstd S )
Nzfunctional.nii)infilezT1.niiz	trans.matmymatlab)
matlab_cmdz%s_to_%s.matZinverse_)prefix)r   spmuZCalcCoregAffineinputstargetr	   AssertionErrormovingr   matr   ospathjoinr   _make_matlab_commandinvmat)
r   r   r   ZcoregpthZmov_Ztgtr   scrpt r   X/var/www/html/virt/lib/python3.6/site-packages/nipype/interfaces/spm/tests/test_utils.py
test_coreg   s    



r   c              C   sn   t dd} t dd}tjdd}|jjdks0t| |j_||j_|jd }d}||ksZtd}||ksjtd S )Nzfunctional.nii)r   z	trans.matr   )r	   z#[p n e v] = spm_fileparts(V.fname);zV.mat = transform.M * V.mat;)	r   r   ZApplyTransformr   r	   r   in_filer   r   )r   r   Zapplymatr   expectedr   r   r   test_apply_transform   s    


r   c              C   s   t dd} t dd}tjdd}|jjdks0t| |j_||j_|jjdksPtt	j
t |jjdd W d Q R X t	j
t |jjd	d W d Q R X d
|j_|jd }t| dd}|jj|kstd}||jddkstd}||kstd|kstd S )Nzfunctional.nii)r   zT1.niiZmymatlab_version)r	   r   Znearest)interp
      r)r
   z+
flags.mean=0;
flags.which=1;
flags.mask=0;  zflags.interp = 1;
zspm_reslice(invols, flags);)r   r   ZReslicer   r	   r   r   space_definingr   pytestraisesr   	trait_setr   r   out_filereplace)r   r%   Zreslicescriptoutfiler   Zexpected_interpr   r   r   test_reslice-   s(    


r-   c              C   s   t dd} tjdd}|jjdks&t|jjdks6t|jjdksFt|jjdksVt|jj	 sdtt
jt |jjdd	 W d Q R X t
jt |jjd
d W d Q R X t
jt |jjdgd W d Q R X | g|j_|jj| gkstd S )Nzdicomdir/123456-1-1.dcm)r   r   )r	   Zflatz./converted_dicomZniiZwrong)output_dir_structZFAT)formatzdoes_sfd_not_32fn_exist.dcm)in_files)r   r   ZDicomImportr   r	   r   r.   
output_dirr/   Zicedimsr&   r'   r   r(   r0   )ZdicomZdir   r   r   test_dicom_importD   s    

r2   )r   r&   Znipype.testingr   Znipype.interfaces.spm.utilsZ
interfacesZspmutilsr   Znipype.interfaces.baser   Znipype.utils.filemanipr   r   r   r   r   r-   r2   r   r   r   r   <module>   s   