
    g                     6    d dl Z d dlmZ d Zd Zd Zd Zd Zy)    N)hmmc            	          ddg} ddg}ddgddgg}ddgd	d
gg}ddg}g d}t        t        |d gt        |      z              }t        j                  | ||||      }|| ||fS )Nrainzno rainumbrellano umbrellagffffff?g333333?g?g?g?g?      ?)r   r   r   r   r   )listziplenr   _create_hmm_tagger)statessymbolsABpiseqmodels          L/var/www/openai/venv/lib/python3.12/site-packages/nltk/test/unit/test_hmm.py_wikipedia_example_hmmr      s     i F=)G
sc3Z A
sc3Z A
sB
IC
s3S)*
+C""67Aq"=E&'3&&    c                      ddl m}  t        j                         \  }}}ddg}g dg dg}d|j	                  |      z  } | ||       y )Nr   assert_array_almost_equal)upN)gffffff?g{Gz?g
ףp=
?)gǺ?g rh?g48EG?   )numpy.testingr   r   _market_hmm_example_forward_probability)r   r   r   r   r   expectedfps          r   test_forward_probabilityr!      sP    7 !446E67
&C"$<=H	
e((-	-Bb(+r   c                      ddl m}  t               \  }}}}d|j                  |      z  }|j                  |j                  d      z  j                  }ddgdd	gd
dgddgddgg} | ||d       y )Nr   r   r      axisgX2ı.?g48E?g0*D?g}bٽ?gqh?g?gmV}b?g&S:?gC?gHP?   )r   r   r   r   Tsum)r   r   r   r   r   r    wikipedia_resultss          r   test_forward_probability2r*   %   s    7"8":E67C	
e((-	-B $$Q
	"	"B 
				 /Q7r   c                      ddl m}  t               \  }}}}d|j                  |      z  }|j                  |j                  d      z  j                  }ddgdd	gd
dgddgddgg} | ||d       y )Nr   r   r   r#   r$   g2w-!?g?g_L?g|гY?gAf?g|a2U0?gPs?g_Q?r   r&   )r   r   r   _backward_probabilityr'   r(   )r   r   r   r   r   bpr)   s          r   test_backward_probabilityr.   9   s    7"8":E67C	
e))#.	.B $$Q
	"	"B 
				c
	 /Q7r   c                 .    t        j                  d       y )Nnumpy)pytestimportorskip)modules    r   setup_moduler4   Q   s    
 r   )r1   nltk.tagr   r   r!   r*   r.   r4    r   r   <module>r7      s$     '$
,8(80!r   