
    c                         d Z ddlZddlZddlmZ ddlmZ ddlmZ  G d dej                  Z	e
dk    r0ej                            ej                    ej                     dS dS )	z]
Automated tests for direct confirmation measures in the direct_confirmation_measure module.
    N)
namedtuple)direct_confirmation_measure)text_analysisc                   &    e Zd Zd Zd Zd Zd ZdS )TestDirectConfirmationMeasurec                 <   dgg| _         h dddhd| _        d| _        ddd}d |                                D             } t	          dd	          ||          }t          j        d
dh|          | _        h dddhd| j        _        | j        | j        _	        d S )N)      >   r
         r      testdocc                     i | ]\  }}||	S  r   ).0kvs      Dlib/python3.11/site-packages/gensim/test/test_direct_confirmation.py
<dictcomp>z7TestDirectConfirmationMeasure.setUp.<locals>.<dictcomp>   s    666TQAq666    
Dictionaryztoken2id, id2tokenr	   r
   )r   r	   )
segmentationposting_listnum_docsitemsr   r   InvertedIndexAccumulatoraccumulator_inverted_index	_num_docs)selfid2tokentoken2id
dictionarys       r   setUpz#TestDirectConfirmationMeasure.setUp   s     %XJ )		q!f55%((66X^^%5%5666CZ.BCCHhWW
(A1a&*UU/8yyaV+D+D(%)]"""r   c                 ,   t          j        | j        | j                  d         }d}|                     ||           t          j        | j        | j        d          d         \  }}|                     ||           |                     d|           dS )z"Test log_conditional_probability()r   g6B.Twith_std        N)r   log_conditional_probabilityr   r   assertAlmostEqualassertEqualr!   obtainedexpectedmeanstds        r    test_log_conditional_probabilityz>TestDirectConfirmationMeasure.test_log_conditional_probability"   s    .Jt/1 1124  x222/Kt/$@ @ @@AC	cx...c"""""r   c                 ,   t          j        | j        | j                  d         }d}|                     ||           t          j        | j        | j        d          d         \  }}|                     ||           |                     d|           dS )zTest log_ratio_measure()r   gUPVǿTr'   r)   Nr   log_ratio_measurer   r   r+   r,   r-   s        r   test_log_ratio_measurez4TestDirectConfirmationMeasure.test_log_ratio_measure/   s    .@t/1 1124  x222/At/$@ @ @@AC	cx...c"""""r   c                 2   t          j        | j        | j        d          d         }d}|                     ||           t          j        | j        | j        dd          d         \  }}|                     ||           |                     d|           dS )z#Test normalized_log_ratio_measure()T)	normalizer   gU7 )r8   r(   r)   Nr4   r-   s        r   !test_normalized_log_ratio_measurez?TestDirectConfirmationMeasure.test_normalized_log_ratio_measure<   s    .@t/4A A AABD  x222/At/4$P P PPQS	cx...c"""""r   N)__name__
__module____qualname__r%   r2   r6   r9   r   r   r   r   r      sP        3 3 3# # ## # ## # # # #r   r   __main__)__doc__loggingunittestcollectionsr   gensim.topic_coherencer   r   TestCaser   r:   rootsetLevelWARNINGmainr   r   r   <module>rH      s       " " " " " " > > > > > > 0 0 0 0 0 04# 4# 4# 4# 4#H$5 4# 4# 4#n z L'/***HMOOOOO r   