§
    Øõc’  ã                   ó  — d Z ddlZddlZddlmZmZ ddlmZ ddlm	Z	 ddl
mZmZ ddlZ ee¦  «        Zd„ eD ¦   «         Z G d„ d	ej        e	j        ¦  «        Zed
k    r( ej        dej        ¬¦  «          ej        ¦   «          dS dS )zN
Automated tests for checking transformation algorithms (the models package).
é    N)ÚmmcorpusÚ
Dictionary)Úhdpmodel)Úbasetmtests)ÚdatapathÚcommon_textsc                 óB   — g | ]}t                                |¦  «        ‘ŒS © )Ú
dictionaryÚdoc2bow)Ú.0Útexts     ú9lib/python3.11/site-packages/gensim/test/test_hdpmodel.pyú
<listcomp>r      s&   € Ð	<Ð	<Ð	< t*×
Ò
˜TÑ
"Ô
"Ð	<Ð	<Ð	<ó    c                   ó    — e Zd Zd„ Zd„ Zd„ ZdS )ÚTestHdpModelc                 óü   — t          j        t          d¦  «        ¦  «        | _        t          j        | _        |                      t          t          t          j	         
                    d¦  «        ¬¦  «        | _        d S )Nztestcorpus.mmr   )Úid2wordÚrandom_state)r   ÚMmCorpusr   Úcorpusr   ÚHdpModelÚclass_r   ÚnpÚrandomÚseedÚmodel)Úselfs    r   ÚsetUpzTestHdpModel.setUp   sR   € ÝÔ'­°Ñ(AÔ(AÑBÔBˆŒÝÔ'ˆŒØ—[’[¥µÍ"Ì)Ï.Ê.ÐYZÑJ[ÔJ[[Ñ\Ô\ˆŒ
ˆ
ˆ
r   c                 óL  — | j                              ¦   «         d         }d\  }}|d                              d¦  «        d                              d¦  «        \  }}|                      |d         d¦  «         |                      ||¦  «         |                      ||¦  «         dS )z*
        Check show topics method
        r   )z0.264ztrees é   ú+Ú*N)r   Úshow_topicsÚsplitÚassertEqual)r   ÚresultsÚexpected_probÚexpected_wordÚprobÚwords         r   Útest_topic_valueszTestHdpModel.test_topic_values    sŸ   € ð ”*×(Ò(Ñ*Ô*¨1Ô-ˆØ'8Ñ$ˆ}Ø˜Q”Z×%Ò% cÑ*Ô*¨1Ô-×3Ò3°CÑ8Ô8‰
ˆˆdØ×Ò˜ œ QÑ'Ô'Ð'Ø×Ò˜˜}Ñ-Ô-Ð-Ø×Ò˜˜}Ñ-Ô-Ð-àˆr   c                 óš   — | j                              ¦   «         }|                      |j        d         | j         j        d         ¦  «         dS )zZ
        Create ldamodel object, and check if the corresponding alphas are equal.
        r   N)r   Úsuggested_lda_modelr'   ÚalphaÚ	lda_alpha)r   Úldams     r   Útest_ldamodelzTestHdpModel.test_ldamodel-   sC   € ð Œz×-Ò-Ñ/Ô/ˆØ×Ò˜œ Aœ¨¬
Ô(<¸QÔ(?Ñ@Ô@Ð@Ð@Ð@r   N)Ú__name__Ú
__module__Ú__qualname__r    r-   r3   r
   r   r   r   r      sI   € € € € € ð]ð ]ð ]ð
ð ð ðAð Að Að Að Ar   r   Ú__main__z)%(asctime)s : %(levelname)s : %(message)s)ÚformatÚlevel)Ú__doc__ÚloggingÚunittestÚgensim.corporar   r   Úgensim.modelsr   Úgensim.testr   Úgensim.test.utilsr   r   Únumpyr   r   r   ÚTestCaseÚTestBaseTopicModelr   r4   ÚbasicConfigÚDEBUGÚmainr
   r   r   ú<module>rG      s&  ððð ð
 €€€Ø €€€à /Ð /Ð /Ð /Ð /Ð /Ð /Ð /Ø "Ð "Ð "Ð "Ð "Ð "Ø #Ð #Ð #Ð #Ð #Ð #Ø 4Ð 4Ð 4Ð 4Ð 4Ð 4Ð 4Ð 4à Ð Ð Ð àˆZ˜Ñ%Ô%€
Ø	<Ð	<¨|Ð	<Ñ	<Ô	<€ðAð Að Að Að A8Ô$ kÔ&Dñ Aô Að Að6 ˆzÒð Ø€GÔÐJÐRYÔR_Ð`Ñ`Ô`Ð`Ø€H„MO„O€O€O€Oðð r   