
    ck                         d dl Z d dlZd dlZd dlmZ d dlmZmZ  G d dej	                  Z
edk    r( e j        de j                    ej                     dS dS )	    N)LdaModel)common_dictionarycommon_corpusc                   &    e Zd Zd Zd Zd Zd ZdS )TestLdaDiffc                     t           | _        t          | _        d| _        d| _        t          | j        | j        | j        d          | _        d S )N   
   )corpusid2word
num_topicspasses)r   
dictionaryr   r   r   n_ann_termsr   modelselfs    7lib/python3.11/site-packages/gensim/test/test_tmdiff.pysetUpzTestLdaDiff.setUp   sD    +#T[$/VZVenpqqq


    c                 8   | j                             | j         | j                  \  }}|                     |j        | j        | j        f           |                     t          |          | j                   |                     t          |d                   | j                   | j                             | j         | j        d          \  }}|                     |j        | j        f           |                     t          |          | j                   d S )N)r   r   T)r   diagonal)r   diffr   assertEqualshaper   len)r   mdiff
annotations      r   
test_basiczTestLdaDiff.test_basic   s     JOODJDDTOUUzt&HIIIZ$/:::Z]++T_=== !JOODJDDT_cOddzt&8999Z$/:::::r   c           
      X   dD ]%}| j                             | j         | j        |          \  }}|D ]H}|D ]C\  }}|                     |g            |                     t	          |          | j                   DI|                     t          j        t          j        |          t          j	        |j
        d         |j                                       |dk    rF|                     t          j        |t          j	        |j
        |j                                       | j                             | j         | j        |d          \  }}|D ]C\  }}|                     |g            |                     t	          |          | j                   D|                     t          j        |t          j	        |j
        |j                                       |dk    rF|                     t          j        |t          j	        |j
        |j                                       'd S )N)	hellingerkullback_leiblerjaccardr   distancer   )dtyper#   T)r   r%   r   )r   r   r   r   r   
assertTruenpallclosediagzerosr   r&   )r   	dist_namer   r   row
int_tokensdiff_tokenss          r   test_identityzTestLdaDiff.test_identity%   s   E 	^ 	^I $

HXcl m mE:! H H14 H H-Z$$["555$$S__d6FGGGGH OOBKQW\Wb8c8c8cddeeeI% ^E28EKu{3[3[3[ \ \]]] 

8HS\gkll E: .8 D D)[  b111  Z$2BCCCCOOBKrx5;/W/W/WXXYYYI% ^E28EKu{3[3[3[ \ \]]]3	^ 	^r   c                     |                      t          | j        j        | j        | j        d           |                      t          | j        j        g | j        d           d S )N	somethingr$   )assertRaises
ValueErrorr   r   r   r   s    r   
test_inputzTestLdaDiff.test_inputA   sX    *djotztO_juvvv*djortGWbmnnnnnr   N)__name__
__module____qualname__r   r   r0   r5    r   r   r   r      s[        r r r; ; ;^ ^ ^8o o o o or   r   __main__z)%(asctime)s : %(levelname)s : %(message)s)formatlevel)loggingunittestnumpyr(   gensim.modelsr   gensim.test.utilsr   r   TestCaser   r6   basicConfigDEBUGmainr9   r   r   <module>rF      s          " " " " " " > > > > > > > >4o 4o 4o 4o 4o(# 4o 4o 4on z GJRYR_````HMOOOOO r   