a
    Kb                     @   s6   d dl Z d dlmZmZmZ dZG dd de jZdS )    N)BigramAssocMeasuresQuadgramAssocMeasuresTrigramAssocMeasuresg:0yE>c                   @   s$   e Zd Zdd Zdd Zdd ZdS )TestLikelihoodRatioc                 C   sN   | j tddddtd | j tddddtd | jttjgd	R   d S )
N   )   r      gwBoP@Zdelta   )r
   r
           )r   )r   r   r   )assertAlmostEqualr   likelihood_ratio_DELTAassertRaises
ValueErrorself r   :lib/python3.9/site-packages/nltk/test/unit/test_metrics.pytest_lr_bigram   s    z"TestLikelihoodRatio.test_lr_bigramc                 C   sR   | j tdddddtd | j tdddddtd | jttjgdR   d S )Nr
   )r
   r
   r
   r   g9B.@r	   r   )r
   r
   r
   r   r   r   )r   r   r   r   r   r   r   r   r   r   test_lr_trigram   s     z#TestLikelihoodRatio.test_lr_trigramc                 C   sV   | j tddddddtd | j tddddddtd | jttjgdR   d S )	Nr
   r
   r
   r
   r
   )r
   r
   r
   r
   r
   r
   r   gs;? @r	   r   )r
   r   )r
   r
   r
   r
   r
   r   r   r
   )r   r   r   r   r   r   r   r   r   r   test_lr_quadgram/   s(    

z$TestLikelihoodRatio.test_lr_quadgramN)__name__
__module____qualname__r   r   r   r   r   r   r   r      s   r   )ZunittestZnltk.metricsr   r   r   r   ZTestCaser   r   r   r   r   <module>   s   