σ
ω`]c           @@ sn   d  d l  m Z m Z d  d l Z d  d l m Z d  d l m Z d Z d   Z	 d e j
 f d     YZ d S(	   i    (   t   absolute_importt   unicode_literalsN(   t   BigramCollocationFinder(   t   BigramAssocMeasuresg:0βyE>c         C@ sZ   xS t  |  |  D]B \ } } | d | d k sN t | d | d  t k r t Sq Wt S(   ue   Verify that two sequences of n-gram association values are within
       _EPSILON of each other.
    i    i   (   t   zipt   abst   _EPSILONt   Falset   True(   t   xt   yt   x1t   y1(    (    s?   lib/python2.7/site-packages/nltk/test/unit/test_collocations.pyt   close_enough   s    2t
   TestBigramc           B@ s#   e  Z d    Z d   Z d   Z RS(   c         C@ s  d j    } t j |  } |  j t | j j    t d
 d d d d d d g   |  j t | j j    t d d d d g   |  j t	 |  t
 | j j    k oΛ t
 | j j    d k n  |  j t t | j t j   t d d d  d" d$ d& d( g    d  S()   Nu   this this is is a a test testu   ai   u   testu   isu   thisi   g      π?(   u   au   a(   (   u   au   ai   (   u   au   test(   (   u   au   testi   (   u   isu   a(   (   u   isu   ai   (   u   isu   is(   (   u   isu   isi   (   u   testu   test(   (   u   testu   testi   (   u   thisu   is(   (   u   thisu   isi   (   u   thisu   this(   (   u   thisu   thisi   (   u   ai   (   u   isi   (   u   testi   (   u   thisi   (   u   au   a(   (   u   au   ag      π?(   u   au   test(   (   u   au   testg      π?(   u   isu   a(   (   u   isu   ag      π?(   u   isu   is(   (   u   isu   isg      π?(   u   testu   test(   (   u   testu   testg      π?(   u   thisu   is(   (   u   thisu   isg      π?(   u   thisu   this(   (   u   thisu   thisg      π?(   t   splitR   t
   from_wordst   assertEqualt   sortedt   ngram_fdt   itemst   word_fdt
   assertTruet   lent   sumt   valuesR   t   score_ngramsR   t   pmi(   t   selft   sentt   b(    (    s?   lib/python2.7/site-packages/nltk/test/unit/test_collocations.pyt   test_bigram2   s8    Ec         C@ s&  d j    } t j | d d } |  j t | j j    t d d d d d d d g   |  j t | j j    t d d d d g   |  j t	 |  t
 | j j    k oΩ t
 | j j    d	 d d
 k n  |  j t t | j t j   t d  d" d$ d& d( d* d, g    d  S(-   Nu   this this is is a a test testt   window_sizei   u   au   testu   isu   thisi   i   g       @gh½£\ω?g        (   u   au   test(   (   u   au   testi   (   u   isu   a(   (   u   isu   ai   (   u   thisu   is(   (   u   thisu   isi   (   u   au   a(   (   u   au   ai   (   u   isu   is(   (   u   isu   isi   (   u   testu   test(   (   u   testu   testi   (   u   thisu   this(   (   u   thisu   thisi   (   u   ai   (   u   isi   (   u   testi   (   u   thisi   (   u   au   test(   (   u   au   testgh½£\ω?(   u   isu   a(   (   u   isu   agh½£\ω?(   u   thisu   is(   (   u   thisu   isgh½£\ω?(   u   au   a(   (   u   au   ag        (   u   isu   is(   (   u   isu   isg        (   u   testu   test(   (   u   testu   testg        (   u   thisu   this(   (   u   thisu   thisg        (   R   R   R   R   R   R   R   R   R   R   R   R   R   R   R   R   (   R   R   R   (    (    s?   lib/python2.7/site-packages/nltk/test/unit/test_collocations.pyt   test_bigram3E   s<    	*c         C@ s:  d j    } t j | d d } |  j t | j j    t d d d d d d d d d! g	   |  j t | j j    t d" d# d$ d% g   |  j t	 |  t
 | j j    k oη t
 | j j    d d	 d d
 d k n  |  j t t | j t j   t d' d) d+ d- d/ d1 d3 d5 d7 g	    d  S(8   Nu   this this is is a a test testR    i   u   au   testi   u   isu   thisi   i   i   g      @g      π?gΡz?GΈβ?g      πΏ(   u   au   test(   (   u   au   testi   (   u   isu   a(   (   u   isu   ai   (   u   thisu   is(   (   u   thisu   isi   (   u   isu   test(   (   u   isu   testi   (   u   thisu   a(   (   u   thisu   ai   (   u   au   a(   (   u   au   ai   (   u   isu   is(   (   u   isu   isi   (   u   testu   test(   (   u   testu   testi   (   u   thisu   this(   (   u   thisu   thisi   (   u   ai   (   u   isi   (   u   testi   (   u   thisi   (   u   au   test(   (   u   au   testg      π?(   u   isu   a(   (   u   isu   ag      π?(   u   thisu   is(   (   u   thisu   isg      π?(   u   isu   test(   (   u   isu   testgΡz?GΈβ?(   u   thisu   a(   (   u   thisu   agΡz?GΈβ?(   u   au   a(   (   u   au   ag      πΏ(   u   isu   is(   (   u   isu   isg      πΏ(   u   testu   test(   (   u   testu   testg      πΏ(   u   thisu   this(   (   u   thisu   thisg      πΏ(   R   R   R   R   R   R   R   R   R   R   R   R   R   R   R   R   (   R   R   R   (    (    s?   lib/python2.7/site-packages/nltk/test/unit/test_collocations.pyt   test_bigram5q   sD    	2(   t   __name__t
   __module__R   R!   R"   (    (    (    s?   lib/python2.7/site-packages/nltk/test/unit/test_collocations.pyR      s   	,	,(   t
   __future__R    R   t   unittestt   nltk.collocationsR   t   nltk.metricsR   R   R   t   TestCaseR   (    (    (    s?   lib/python2.7/site-packages/nltk/test/unit/test_collocations.pyt   <module>   s   	