
    /h                     f    S r SSKrSSKJr  SSKJrJrJr  SSKJ	r	   " S S\R                  5      rg)z(
Tests for IBM Model 2 training methods
    Ndefaultdict)AlignedSentIBMModel	IBMModel2)AlignmentInfoc                   &    \ rS rSrS rS rS rSrg)TestIBMModel2   c                    [        SS// SQ5      [        / SQSS/5      /n[        US5      nUR                  U5        U R                  UR                  S   S   S   S	   S
5        U R                  UR                  S	   S   S	   S   S5        g )Nhameggsschinkenr   eierspamr   r   r   r   r            g      ?   gUUUUUU?)r   r   set_uniform_probabilitiesassertEqualalignment_tableselfcorpusmodel2s      Z/var/www/auris/envauris/lib/python3.13/site-packages/nltk/test/unit/translate/test_ibm2.py(test_set_uniform_alignment_probabilities6TestIBMModel2.test_set_uniform_alignment_probabilities   s     )IJ866:JK
 61% 	((0 	//215a8;WE//215a8;WE    c                 V   [        SS// SQ5      [        / SQSS/5      /n[        US5      nUR                  U5        U R                  UR                  S   S   S	   S
   [
        R                  5        U R                  UR                  S
   S   S
   S   [
        R                  5        g )Nr   r   r   r   r   r   c   r   r   r   r   )r   r   r   r   r   r   MIN_PROBr   s      r   =test_set_uniform_alignment_probabilities_of_non_domain_valuesKTestIBMModel2.test_set_uniform_alignment_probabilities_of_non_domain_values   s     )IJ866:JK
 61% 	((0 	//3A6q9!<h>O>OP//226q9!<h>O>OPr"   c                 >   / SQn/ SQn[        X!5      /n[        SS /U-   S/U-   S 5      n[        S 5      nSUS   S'   SUS	   S
'   SUS   S '   SUS   S'   SUS   S'   SUS   S'   [        S 5      nSUS   S   S   S'   SUS   S   S   S'   SUS   S   S   S'   SUS   S   S   S'   SUS   S   S   S'   SUS   S   S   S'   [        US5      nXWl        Xgl        UR                  U5      nSn	Sn
X-  nU R                  [        US5      [        US5      5        g )N)ichessejagern   räucherschinken)ilovetoeatsmokedr   )r   r   r   r   r      r3   UNUSEDc                       [        [        5      $ Nr   float r"   r   <lambda>5TestIBMModel2.test_prob_t_a_given_s.<locals>.<lambda>9   s	    E0Br"   g\(\?r.   r)   r/   r,   r0   r1   r*   r2   r-   r   c                      [        S 5      $ )Nc                      [        S 5      $ )Nc                       [        [        5      $ r6   r7   r9   r"   r   r:   YTestIBMModel2.test_prob_t_a_given_s.<locals>.<lambda>.<locals>.<lambda>.<locals>.<lambda>B   s	    K<Nr"   r   r9   r"   r   r:   GTestIBMModel2.test_prob_t_a_given_s.<locals>.<lambda>.<locals>.<lambda>B   s
    4N(Or"   r   r9   r"   r   r:   r;   B   s
    K OPr"   g
ףp=
?r   r   r3      r   r   r   gQ?gX?gh\?)	r   r   r   r   translation_tabler   prob_t_a_given_sr   round)r   src_sentencetrg_sentencer   alignment_inforB   r   r   probabilitylexical_translation	alignmentexpected_probabilitys               r   test_prob_t_a_given_s#TestIBMModel2.test_prob_t_a_given_s-   s   HBl9:&!F\!J%	
 ((BC(,#u%,0&!&)(,$%+/% (:>(#$677;% !34%P
 '+1a #&*1a #&*1a #&*1a #&*1a #&*1a #61%#4 !0 --n= F;	2>{A.6JA0NOr"   r9   N)__name__
__module____qualname____firstlineno__r    r&   rL   __static_attributes__r9   r"   r   r
   r
      s    F Q )Pr"   r
   )__doc__unittestcollectionsr   nltk.translater   r   r   nltk.translate.ibm_modelr   TestCaser
   r9   r"   r   <module>rY      s1     # ; ; 2JPH%% JPr"   