`]c@sdZddlZddlmZddlmZddlmZddlmZddlm Z dej fd YZ dS( s( Tests for IBM Model 2 training methods iN(t defaultdict(t AlignedSent(tIBMModel(t IBMModel2(t AlignmentInfot TestIBMModel2cBs#eZdZdZdZRS(cCstddgdddgtddddgddgg}t|d}|j||j|jdddd d d |j|jd d d d d ddS( Nthamteggstschinkenteiertspamiiiig?i(RRtset_uniform_probabilitiest assertEqualtalignment_table(tselftcorpustmodel2((sAlib/python2.7/site-packages/nltk/test/unit/translate/test_ibm2.pyt(test_set_uniform_alignment_probabilitiess $ 'cCstddgdddgtddddgddgg}t|d}|j||j|jddd d tj|j|jd dd d tjdS( NRRRR R iiciiii(RRR R R RtMIN_PROB(RRR((sAlib/python2.7/site-packages/nltk/test/unit/translate/test_ibm2.pyt=test_set_uniform_alignment_probabilities_of_non_domain_values s $ &c Csdddddg}dddd d d g}t||g}tddg|dg|d}td}d|dd<tg\(\?cSs tdS(NcSs tdS(NcSs ttS(N(RR(((sAlib/python2.7/site-packages/nltk/test/unit/translate/test_ibm2.pyRER (R(((sAlib/python2.7/site-packages/nltk/test/unit/translate/test_ibm2.pyRER (R(((sAlib/python2.7/site-packages/nltk/test/unit/translate/test_ibm2.pyRER g ףp= ?iigQ?(iiiiiiig@ǘ?g7h>?gSKB; ?gčN?gX?gQ?g&4?gJQT?gW92?gh\?( RRtNoneRRttranslation_tableR tprob_t_a_given_sR tround( Rt src_sentencet trg_sentenceRtalignment_infoR"R Rt probabilitytlexical_translationt alignmenttexpected_probability((sAlib/python2.7/site-packages/nltk/test/unit/translate/test_ibm2.pyttest_prob_t_a_given_s0s>       (t__name__t __module__RRR,(((sAlib/python2.7/site-packages/nltk/test/unit/translate/test_ibm2.pyRs  ( t__doc__tunittestt collectionsRtnltk.translateRRRtnltk.translate.ibm_modelRtTestCaseR(((sAlib/python2.7/site-packages/nltk/test/unit/translate/test_ibm2.pyts