3
Xd                 @   s   d dl mZ ddlmZ ddlZddlZddlZdZ	yed W n& e
k
rf Z z
dZ	W Y ddZ[X nX ej d	d
 Zejje	dddd Zejje	 dddd ZdS )   )NetworkBasedStatistic   )package_check    NTZcviewerFc             C   sp   g }dd t dD }xTt|D ]H\}}tjjdd}tj|}| j||  d }tj|| |j	| q W |S )Nc             S   s   g | ]}d t | qS )name)str).0i r
   W/var/www/html/virt/lib/python3.6/site-packages/nipype/interfaces/cmtk/tests/test_nbs.py
<listcomp>   s    z#creating_graphs.<locals>.<listcomp>   
   z.pck)
range	enumeratenprandomZrandnxZfrom_numpy_matrixZstrpathZwrite_gpickleappend)tmpdir	graphlistZ
graphnamesidxr   graphGout_filer
   r
   r   creating_graphs   s    
r   z)tests for import error, cviewer available)reasonc          	   C   sd   |j   | }|d d }|dd  }t }||j_||j_d|j_tjt}|j	  W d Q R X d S )N   Zweight)
chdirr   inputs	in_group1	in_group2edge_keypytestraisesImportErrorrun)r   r   r   group1group2nbser
   r
   r   test_importerror   s    r+   zcviewer has to be availablec          
   C   sn   | }|d d }|dd  }t  }||j_||j_d|j_tjt}|j  W d Q R X dt	|j
ksjtd S )Nr   Z	Your_edgez/the graph edges do not have Your_edge attribute)r   r   r    r!   r"   r#   r$   KeyErrorr&   r   valueAssertionError)r   r   r'   r(   r)   r*   r
   r
   r   test_keyerror,   s    r/   )r)   r   Z
utils.miscr   Znumpyr   Znetworkxr   r#   Zhave_cv	Exceptionr*   Zfixturer   markZskipifr+   r/   r
   r
   r
   r   <module>   s   