
    c^                         d Z ddl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 )	za
Automated tests for indirect confirmation measures in the indirect_confirmation_measure module.
    N)
Dictionary)indirect_confirmation_measure)text_analysisc                        e Zd Zd Zd Zd ZdS )TestIndirectConfirmationc                 
   t          j        ddg          g| _        dt          j        ddg          fdt          j        ddg          fgg| _        d| _        d| _        t                      | _        ddd| j        _        d S )N      nlrfaketokens)r	   r
   )	nparraytopicssegmentationgammameasurer   
dictionaryid2token)selfs    Flib/python3.11/site-packages/gensim/test/test_indirect_confirmation.pysetUpzTestIndirectConfirmation.setUp   s     xA''( "(Aq6"2"23a1a&9I9I5JKL
$,,'-(#;#;       c                    t          j        ddh| j                  }h dddhd|_        d|_        t          j        | j        || j        | j	        | j
                  }d}|                     ||d         d	           t          j        | j        || j        | j	        | j
        d
          d         \  }}|                     ||d	           |                     d|d           dS )zTest cosine_similarity()r	   r
   >   r
         r      )r   r	   gV-?r   r   Twith_std        N)r   InvertedIndexAccumulatorr   _inverted_index	_num_docsr   cosine_similarityr   r   r   r   assertAlmostEqual)r   accumulatorobtainedexpectedmeanstds         r   test_cosine_similarityz/TestIndirectConfirmation.test_cosine_similarity"   s    #<aVT_UU*3))A&?&?# !0B{DKtzS S *x!a8881C{DKtz  	c 	xq111sC+++++r   c                    t          j        ddh| j                  }|                    ddgddggd           t	          j        | j        |d          d         \  }}|                     d	|           |                     d	|           d
S )z!Sanity check word2vec_similarity.r	   r
   r   r   r   Tr   r   r    N)r   WordVectorsAccumulatorr   
accumulater   word2vec_similarityr   assertNotEqual)r   r&   r)   r*   s       r   test_word2vec_similarityz1TestIndirectConfirmation.test_word2vec_similarity:   s    #:Aq64?SSXv 
 	 	 	
 2E{T; ; ;;<>	cC&&&C%%%%%r   N)__name__
__module____qualname__r   r+   r1    r   r   r   r      sA        
< 
< 
<, , ,0& & & & &r   r   __main__)__doc__loggingunittestnumpyr   gensim.corpora.dictionaryr   gensim.topic_coherencer   r   TestCaser   r2   rootsetLevelWARNINGmainr5   r   r   <module>rB      s           0 0 0 0 0 0 @ @ @ @ @ @ 0 0 0 0 0 00& 0& 0& 0& 0&x0 0& 0& 0&f z L'/***HMOOOOO r   