U
    ej                     @   sH   d dl mZ d dlmZ dZd Zdd Zdd Zd	d
 Z	dd Z
dS )    )BigramCollocationFinder)BigramAssocMeasuresg:0yE>zthis this is is a a test testc                 C   s   t dd t| |D S )zbVerify that two sequences of n-gram association values are within
    _EPSILON of each other.
    c                 s   s*   | ]"\}}t |d  |d   tkV  qdS )   N)abs_EPSILON).0x1y1 r
   a/var/www/html/assets/scripts/venv/lib/python3.8/site-packages/nltk/test/unit/test_collocations.py	<genexpr>   s     zclose_enough.<locals>.<genexpr>)allzip)xyr
   r
   r   close_enough
   s    r   c               	   C   s   t t} t| j dddddddgks.tt| j dd	d
dgksLtttt	| j
   kr|t	| j
 d ksn ttt| tjdddddddgstd S )N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&   r   
from_wordsSENTsortedngram_fditemsAssertionErrorword_fdlensumvaluesr   score_ngramsr   pmibr
   r
   r   test_bigram2   s.    

	6r9   c               
   C   s   t jtdd} t| j tddddddd	gks6tt| j td
dddgksXtttt	| j
   krt	| j
 d d d ksn ttt| tjtdddddddgstd S )N   Zwindow_size)r   r:   )r   r:   )r   r:   r   r   r   r   r!   r#   r$   r%   r"   r   g       @)r   h\?)r   r<   )r   r<   )r           )r   r=   )r   r=   )r    r=   r*   r7   r
   r
   r   test_bigram3/   s@    


 r>   c                  C   s   t jtdd} t| j tddddddd	d
dg	ks:tt| j tddddgks\tt| j	 }t| j	 d d d d d }t
t|  kr|ksn ttt| tjtdddddddddg	std S )N   r;   )r      )r   r@   )r   r@   )r   r   r:   )r   r   r:   r   r   r   r   r!   r#   r$   r%   r@   r:   r"   r   g      @r'   r(   r)   )rA   z?G?)rB   rC   )r         )r   rD   )r   rD   )r    rD   )r   r+   r,   r-   r.   r/   r0   r1   r3   r4   r2   r   r5   r   r6   )r8   Z	n_word_fdZ
n_ngram_fdr
   r
   r   test_bigram5T   sD    


"rE   N)Znltk.collocationsr   Znltk.metricsr   r   splitr,   r   r9   r>   rE   r
   r
   r
   r   <module>   s   %