U
    Be                     @   sB   d dl Z d dlmZ d dlmZ dZdd ZG dd de jZdS )	    N)BigramCollocationFinder)BigramAssocMeasuresg:0yE>c                 C   sF   t | |D ]6\}}|d |d ks:t|d |d  tkr
 dS q
dS )zeVerify that two sequences of n-gram association values are within
       _EPSILON of each other.
    r      FT)zipabs_EPSILON)xyx1y1 r   |/var/www/nmhs-web.org.in/public_html/infoladakh/backend/venv/lib/python3.8/site-packages/nltk/test/unit/test_collocations.pyclose_enough   s    (r   c                   @   s$   e Zd Zdd Zdd Zdd ZdS )
TestBigramc                 C   s   d  }t|}| t|j tdddddddg | t|j td	d
ddg | t	|t
|j   kot
|j d kn   | tt|tjtdddddddg d S )Nthis this is is a a test testar   r   )r   testr   )isr   r   r   r   r   r   r   r   )thisr   r   r   r   r   r      r   r!   r   r!   r   r!   r   )r         ?r   r%   r   r%   )r   r%   )r   r%   r   r%   )r   r%   splitr   Z
from_wordsassertEqualsortedZngram_fditemsZword_fd
assertTruelensumvaluesr   Zscore_ngramsr   Zpmiselfsentbr   r   r   test_bigram2   sJ    
2zTestBigram.test_bigram2c                 C   s   d  }tj|dd}| t|j tdddddd	d
g | t|j tddddg | t	|t
|j   kot
|j d d d kn   | tt|tjtdddddddg d S )Nr      Zwindow_size)r   r7   )r   r7   )r   r7   r   r   r   r   r    r"   r#   r$   r!   r   g       @)r   h\?)r   r9   )r   r9   )r           )r   r:   )r   r:   )r   r:   r)   r2   r   r   r   test_bigram3D   sR    zTestBigram.test_bigram3c                 C   s   d  }tj|dd}| t|j tdddddd	d
ddg	 | t|j tddddg | t	|t
|j   kot
|j d d d d d kn   | tt|tjtdddddddddg	 d S )Nr      r8   )r      )r   r=   )r   r=   )r   r   r7   )r   r   r7   r   r   r   r   r    r"   r#   r$   r=   r7   r!   r   g      @r&   r'   r(   )r>   z?G?)r?   r@   )r         )r   rA   )r   rA   )r   rA   r)   r2   r   r   r   test_bigram5p   sZ     zTestBigram.test_bigram5N)__name__
__module____qualname__r6   r;   rB   r   r   r   r   r      s   ,,r   )	ZunittestZnltk.collocationsr   Znltk.metricsr   r   r   ZTestCaser   r   r   r   r   <module>   s
   