U
    f»{j  ã                   @   sd   d dl Z d dlZd dlmZ d dlmZ d dlmZ ee j	 
ddddd	d
ddddg	¡dd„ ƒƒZdS )é    N)Úassert_array_equal)Úcombine_adjacency)Úrequires_sklearnÚshape©é   )é   )r   r   )r   r   )r   r   )é   é   )r   r   r   )r   r   r   )r	   r
   é   c                 C   sº   ddl m} t| ƒdkr| n| d }||Ž  ¡ }t| Ž }t | ¡fd }|j|j  kr`|ksfn t‚|j	dk 
¡ sxt‚| ¡ }t |ddddg¡ 
¡ sœt‚|j|jks¬t‚t||ƒ dS )	z1Test adjacency equivalence for lattice adjacency.r   )Úgrid_to_graphr   r   r   g      ð?r	   N)Zsklearn.feature_extractionr   ÚlenZtoarrayr   ÚnpÚprodr   ÚAssertionErrorÚdataÚallZin1dr   )r   r   Zsk_shapeZconn_skÚconnZ
want_shape© r   ú]/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mne/stats/tests/test_adjacency.pyÚtest_adjacency_equiv   s    r   )ZpytestÚnumpyr   Znumpy.testingr   Z	mne.statsr   Z	mne.utilsr   ÚmarkZparametrizer   r   r   r   r   Ú<module>   s"   ÷