
    c                     ,    d Z ddlZ G d d          ZdS )zN
Automated tests for checking transformation algorithms (the models package).
    Nc                   ,    e Zd Zd Zd Zd Zd Zd ZdS )TestBaseTopicModelc                     | j                             d          }|D ]U\  }}|                     t          |t                               |                     t          |t
                               Vd S )NT	formatted)modelshow_topics
assertTrue
isinstanceintstrselftopicstopic_notopics       7lib/python3.11/site-packages/gensim/test/basetmtests.pytest_print_topicz#TestBaseTopicModel.test_print_topic   sr    ''$'77% 	4 	4OHeOOJx55666OOJuc223333	4 	4    c                     | j                                         }|D ]U\  }}|                     t          |t                               |                     t          |t
                               Vd S )N)r   print_topicsr
   r   r   r   r   s       r   test_print_topicsz$TestBaseTopicModel.test_print_topics   sm    ((**% 	4 	4OHeOOJx55666OOJuc223333	4 	4r   c                    | j                             d          }|D ]a\  }}|                     t          |t                               |                     t          |t
          j        t          f                     bd S )N   )r   
show_topicr
   r   r   npfloatingfloat)r   r   kvs       r   test_show_topicz"TestBaseTopicModel.test_show_topic   sz    
%%a(( 	A 	ADAqOOJq#..///OOJq2;*>??@@@@	A 	Ar   c           	         | j                             d          }|D ]\  }}|                     t          |t                               |                     t          |t
                               |D ]a\  }}|                     t          |t                               |                     t          |t          j        t          f                     bd S )NFr   )
r   r	   r
   r   r   listr   r   r   r   )r   r   r   r   r   r    s         r   test_show_topicsz#TestBaseTopicModel.test_show_topics#   s    ''%'88% 	E 	EOHeOOJx55666OOJud33444 E E1
1c 2 2333
1r{E.B C CDDDDE	E 	Er   c                 b   | j                                         }t          | j         j                  }|D ]y}|                     t          |t          j                             |                     ||j	        d                    | 
                    t          j        |          dd           zd S )Nr   g      ?   )r   
get_topicslenid2wordr
   r   r   ndarrayassertEqualshapeassertAlmostEqualsum)r   r   
vocab_sizer   s       r   test_get_topicsz"TestBaseTopicModel.test_get_topics-   s    &&((+,,
 	: 	:EOOJubj99::: ZQ888""26%==#q9999	: 	:r   N)__name__
__module____qualname__r   r   r!   r$   r0    r   r   r   r      se        4 4 44 4 4A A AE E E: : : : :r   r   )__doc__numpyr   r   r4   r   r   <module>r7      sO        ': ': ': ': ': ': ': ': ': ':r   