JL i6ddlZddlmZdZdZdZdZdZy)N)hmmc ddg}ddg}ddgddgg}ddgd d gg}d d g}gd }tt|dgt|z}tj|||||}||||fS) Nrainzno rainumbrella no umbrellagffffff?g333333?g?g?g?g??)rrrrr)listziplenr_create_hmm_tagger)statessymbolsABpiseqmodels ]/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/nltk/test/unit/test_hmm.py_wikipedia_example_hmmrsi F=)G sc3Z A sc3Z A sB IC s3S)* +C  " "67Aq" =E &'3 &&cddlm}tj\}}}ddg}gdgdg}d|j |z}|||y)Nrassert_array_almost_equal)upN)gffffff?g{Gz?g ףp= ?)gǺ?g rh?g48EG?) numpy.testingrr_market_hmm_example_forward_probability)rrr rrexpectedfps rtest_forward_probabilityr!sP7!446E67  &C"$<=H e((- -Bb(+rcddlm}t\}}}}d|j|z}|j|j dz j}ddgdd gd d gd d gddgg}|||dy)NrrraxisgX2ı.?g48E?g0*D?g}bٽ?g qh?g?gmV}b?g&S:?gC?gHP?)rrrrTsum)rrr rrr wikipedia_resultss rtest_forward_probability2r*%s7"8":E67C e((- -B $$Q  " "B      /Q7rcddlm}t\}}}}d|j|z}|j|j dz j}ddgdd gd d gd d gddgg}|||dy)Nrrrr#r$g2w-!?g?g_L?g|гY?gAf?g|a2U0?gPs?g_Q?rr&)rrr_backward_probabilityr'r()rrr rrbpr)s rtest_backward_probabilityr.9s7"8":E67C e))#. .B $$Q  " "B     c  /Q7rc.tjdy)Nnumpy)pytest importorskip)modules r setup_moduler4Qs  r)r1nltk.tagrrr!r*r.r4rrr7s$ '$ ,8(80!r