
    c1	                         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
 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 )zN
Automated tests for checking transformation algorithms (the models package).
    N)MmCorpus)rpmodel)matutils)datapathget_tmpfilec                   &    e Zd Zd Zd Zd Zd ZdS )TestRpModelc                 H    t          t          d                    | _        d S )Nztestcorpus.mm)r   r   corpus)selfs    8lib/python3.11/site-packages/gensim/test/test_rpmodel.pysetUpzTestRpModel.setUp   s    x8899    c                 d   t           j                            d           t          j        | j        d          }t          | j                  d         }||         }t          j        |d          }t          j	        ddg          }| 
                    t          j        ||                     d S )N      
num_topicsr   g&}`g&}`?)nprandomseedr   RpModelr   listr   sparse2fullarray
assertTrueallclose)r   modeldoctransformedvecexpecteds         r   test_transformzTestRpModel.test_transform   s     		r::: 4;"Cj";228[*566C2233333r   c                    t          d          }t          j        | j        d          }|                    |           t          j                            |          }|                     |j        |j                   |                     t          j
        |j        |j                             g }|                     t          j
        ||         ||                              d S )Nzgensim_models.tstr   r   r   r   r   r   saveloadassertEqualr   r   r   r   
projectionr   fnamer   model2tstvecs        r   test_persistencezTestRpModel.test_persistence)   s    /00:::

5%%e,,)6+<===E$4f6GHHIIIE&M6&>BBCCCCCr   c                    t          d          }t          j        | j        d          }|                    |           t          j                            |d           }|                     |j        |j                   |                     t          j
        |j        |j                             g }|                     t          j
        ||         ||                              d S )Nzgensim_models.tst.gzr   r   )mmapr%   r*   s        r   test_persistence_compressedz'TestRpModel.test_persistence_compressed3   s    233:::

5%%e$%77)6+<===E$4f6GHHIIIE&M6&>BBCCCCCr   N)__name__
__module____qualname__r   r#   r.   r1    r   r   r	   r	      sX        : : :4 4 4D D DD D D D Dr   r	   __main__z)%(asctime)s : %(levelname)s : %(message)s)formatlevel)__doc__loggingunittestnumpyr   gensim.corpora.mmcorpusr   gensim.modelsr   gensimr   gensim.test.utilsr   r   TestCaser	   r2   basicConfigDEBUGmainr5   r   r   <module>rE      s    
       , , , , , , ! ! ! ! ! !       3 3 3 3 3 3 3 3$D $D $D $D $D(# $D $D $DN z GJRYR_````HMOOOOO r   