
`]c           @@ sW   d  d l  m Z m Z d  d l m Z d   Z d   Z d   Z d   Z d   Z	 d S(	   i    (   t   absolute_importt   unicode_literals(   t   hmmc          C@ s   d d g }  d d g } d d g d d g g } d d g d	 d
 g g } d d g } d d d d d g } t  t | d  g t |    } t j |  | | | |  } | |  | | f S(   Nu   rainu   no rainu   umbrellau   no umbrellagffffff?g333333?g?g?g?g?g      ?(   t   listt   zipt   Nonet   lenR   t   _create_hmm_tagger(   t   statest   symbolst   At   Bt   pit   seqt   model(    (    s6   lib/python2.7/site-packages/nltk/test/unit/test_hmm.pyt   _wikipedia_example_hmm   s    "c          C@ ss   d d l  m }  t j   \ } } } d d g } d d d g d d d	 g g } d
 | j |  } |  | |  d  S(   Ni    (   t   assert_array_almost_equalu   upgffffff?g{Gz?g
ףp=
?gǺ?g rh?g48EG?i   (   u   upN(   u   upN(   t   numpy.testingR   R   t   _market_hmm_exampleR   t   _forward_probability(   R   R   R   R	   R   t   expectedt   fp(    (    s6   lib/python2.7/site-packages/nltk/test/unit/test_hmm.pyt   test_forward_probability   s    c          C@ s   d d l  m }  t   \ } } } } d | j |  } | j | j d d  j } d d g d d	 g d
 d g d d g d d g g } |  | | d  d  S(   Ni    (   R   i   t   axisi   gX2ı.?g48E?g0*D?g}bٽ?gqh?g?gmV}b?g&S:?gC?gHP?i   (   R   R   R   R   t   Tt   sum(   R   R   R   R	   R   R   t   wikipedia_results(    (    s6   lib/python2.7/site-packages/nltk/test/unit/test_hmm.pyt   test_forward_probability2%   s    				c          C@ s   d d l  m }  t   \ } } } } d | j |  } | j | j d d  j } d d g d d	 g d
 d g d d g d d g g } |  | | d  d  S(   Ni    (   R   i   R   i   g2w-!?g?g_L?g|гY?gAf?g|a2U0?gPs?g_Q?g      ?i   (   R   R   R   t   _backward_probabilityR   R   (   R   R   R   R	   R   t   bpR   (    (    s6   lib/python2.7/site-packages/nltk/test/unit/test_hmm.pyt   test_backward_probability9   s    				c         C@ sD   d d l  m } y d d  l } Wn t k
 r? | d   n Xd  S(   Ni    (   t   SkipTestu(   numpy is required for nltk.test.test_hmm(   t   noseR   t   numpyt   ImportError(   t   moduleR   R!   (    (    s6   lib/python2.7/site-packages/nltk/test/unit/test_hmm.pyt   setup_moduleQ   s
    N(
   t
   __future__R    R   t   nltk.tagR   R   R   R   R   R$   (    (    (    s6   lib/python2.7/site-packages/nltk/test/unit/test_hmm.pyt   <module>   s   				