3
Yd                 @   s   d dl Z d dlZd dlmZ d dljjZd dlmZ d dljj	Z
e
jje jdd dd Zdd	 Zd
d Zdd Zdd Zdd Zdd Zdd Zdd Zdd Zejje dddd Zejje dddd Zd d! Zd"d# Zd$d% ZdS )&    N)create_files_in_directory)no_spmZ	MATLABCMDmatlabc               C   s$   t jjdkstt jjdks td S )NZtemporalst)spmSliceTiming_jobtypeAssertionError_jobname r   r   ]/var/www/html/virt/lib/python3.6/site-packages/nipype/interfaces/spm/tests/test_preprocess.pytest_slicetiming   s    r   c             C   s8   | \}}t j|d d}|j d d d dks4td S )Nr   )in_filesZtimecorrected_filesa)r   r   _list_outputsr	   )r   filelistoutdirr   r   r   r   test_slicetiming_list_outputs   s    r   c               C   s8   t jjdkstt jjdks tt j jjdks4td S )NspatialZrealignestwrite)r   Realignr   r	   r
   inputsjobtyper   r   r   r   test_realign   s    r   c             C   sf   | \}}t j|d d}|j d d jds2t|j d d jdsLt|j d jdsbtd S )	Nr   )r   Zrealignment_parametersZrp_Zrealigned_filesrZ
mean_imageZmean)r   r   r   
startswithr	   )r   r   r   Zrlgnr   r   r   test_realign_list_outputs!   s
    r   c               C   s8   t jjdkstt jjdks tt j jjdks4td S )Nr   coregr   )r   
Coregisterr   r	   r
   r   r   r   r   r   r   test_coregister)   s    r   c             C   sf   | \}}t j|d d}|j d d jds2tt j|d |d d}|j d d jdsbtd S )Nr   )sourceZcoregistered_sourcer      )r    apply_to_filesZcoregistered_files)r   r   r   r   r	   )r   r   r   r   r   r   r   test_coregister_list_outputs/   s
    r#   c               C   s8   t jjdkstt jjdks tt j jjdks4td S )Nr   	normaliser   )r   	Normalizer   r	   r
   r   r   r   r   r   r   test_normalize7   s    r&   c             C   sf   | \}}t j|d d}|j d d jds2tt j|d |d d}|j d d jdsbtd S )Nr   )r    Znormalized_sourcewr!   )r    r"   normalized_files)r   r%   r   r   r	   )r   r   r   Znormr   r   r   test_normalize_list_outputs=   s
    r)   c               C   s8   t jjdkstt jjdks tt j jjdks4td S )Nr   r$   r   )r   Normalize12r   r	   r
   r   r   r   r   r   r   test_normalize12E   s    r+   c             C   sf   | \}}t j|d d}|j d d jds2tt j|d |d d}|j d d jdsbtd S )Nr   )image_to_alignZnormalized_imager'   r!   )r,   r"   r(   )r   r*   r   r   r	   )r   r   r   Znorm12r   r   r   test_normalize12_list_outputsK   s
    r-   zspm is not installed)reasonc               C   s\   t jj dkr4t j jdks tt j jdksXtn$t j jdksFtt j jdksXtd S )NSPM12toolsZoldsegr   preproc)r   InfonameZSegmentr   r	   r
   r   r   r   r   test_segmentS   s
    r4   c               C   s\   t jj dkr4t j jdks tt j jdksXtn$t j jdksFtt j jdksXtd S )Nr/   r   r1   r0   Zpreproc8)r   r2   r3   Z
NewSegmentr   r	   r
   r   r   r   r   test_newsegment]   s
    r5   c               C   s$   t jjdkstt jjdks td S )Nr   Zsmooth)r   ZSmoothr   r	   r
   r   r   r   r   test_smoothg   s    r6   c               C   s$   t jjdkstt jjdks td S )Nr0   dartel)r   ZDARTELr   r	   r
   r   r   r   r   test_dartell   s    r8   c               C   s$   t jjdkstt jjdks td S )Nr0   r7   )r   ZDARTELNorm2MNIr   r	   r
   r   r   r   r   test_dartelnorm2mniq   s    r9   )osZpytestZnipype.testing.fixturesr   Znipype.interfaces.spmZ
interfacesr   r   Znipype.interfaces.matlabr   ZmlabZMatlabCommandZset_default_matlab_cmdgetenvr   r   r   r   r   r#   r&   r)   r+   r-   markZskipifr4   r5   r6   r8   r9   r   r   r   r   <module>   s*   

