o
    –h  ã                   @   sD   d Z ddlZddlZddlmZ ejjejjgZ	dd„ Z
dd„ ZdS )za
Test some use cases of the chuliu_edmonds algorithm

(currently just the tarjan implementation)
é    N)Útarjanc                  C   sH   t  g d¢¡} t| ƒ}|g ksJ ‚t  g d¢¡} t| ƒ}|g ks"J ‚d S )N)r   é   r   r   r   )r   é   r   r   r   r   )ÚnpÚarrayr   )ÚsimpleÚresult© r	   úb/var/www/html/env_mimamsha/lib/python3.10/site-packages/stanza/tests/common/test_chuliu_edmonds.pyÚtest_tarjan_basic   s   r   c                  C   s®   t  g d¢¡} t| ƒ}t  g d¢¡}t|ƒdksJ ‚t j |d |¡ t  g d¢¡} t| ƒ}t|ƒdks6J ‚t  g d¢¡t  g d¢¡g}t||ƒD ]\}}t j ||¡ qId S )	N)r   é   é   r   )FTTTr   r   )r   r   r   r   é   é   r   r   )FTTTFFF)FFFFTTT)r   r   r   ÚlenÚtestingÚassert_array_equalÚzip)Úcycle_graphr   ÚexpectedÚrÚer	   r	   r
   Útest_tarjan_cycle   s   ÿÿr   )Ú__doc__Únumpyr   ÚpytestÚ#stanza.models.common.chuliu_edmondsr   ÚmarkÚtravisÚpipelineÚ
pytestmarkr   r   r	   r	   r	   r
   Ú<module>   s    	