
    c                     $   d dl Z d dlZd dlZd dlmZ d dlZd dlZ ej	        ej
                            dd          dk    d           G d dej                              Zedk    r( e j        d	e j        
            ej                     dS dS )    NSKIP_NETWORK_TESTSF1z@Skip network-related tests (probably SSL problems on this CI/OS)c                   ,    e Zd Zd Zd Zd Zd Zd ZdS )TestApic                 n   t           j                            t          j                  rt          j        t          j                   t          j                     |                     t           j                            t          j                             t          j	        t          j                   d S )N)
ospathisdirapiBASE_DIRshutilrmtree_create_base_dir
assertTruermdir)selfs    4lib/python3.11/site-packages/gensim/test/test_api.pytest_base_dir_creationzTestApi.test_base_dir_creation   su    7==&& 	(M#,'''cl33444
    c           	      :   t           j                            t          j        dd          }t           j                            t          j                  rt          j        t          j                   |                     t          j	        dd          |           t          j        t          j                   |                     t          t          t          j	        d                              d           t          j        t          j                   d S )Nz__testing_matrix-synopsisz__testing_matrix-synopsis.gzTreturn_path   r   r	   joinr   r   r
   r   r   assertEqualloadlenlist)r   dataset_paths     r   test_load_datasetzTestApi.test_load_dataset   s    w||CL2MOmnn7==&& 	(M#,'''"=4PPPR^___cl###T#(+F"G"GHHII1MMMcl#####r   c                 t   t           j                            t          j                  rt          j        t          j                   t          j        g d          }t           j        	                    t          j        dd          }t          j
        d          }|j        d         }|                     t          j        ||                     t          j        t          j                   |                     t          j
        dd          |           t          j        t          j                   d S )N)2g/wF?g)Bg ]vgAkg_.g&2ggDp?gܹͧ?gٗ4Կg%N?g=vYD?gβĿg6ىys?g
u?g0 |gTO.fg`?gVVgxI9땪g8ƾ?gt{Ic?gP̂?g~_ ?g<gҷw?gpȿgW6maҿg*b8?gO!pg	|Ͽg3"(?gޚ#ʿg=)Ӡ?g<?ѿg031gІ:gE?gҟ:?g(N?gI Hg]o%?gYgt$5Ŀg!l(8?gt?g&lp?g-rg]@x?gSԲοgT?z"__testing_word2vec-matrix-synopsisz%__testing_word2vec-matrix-synopsis.gzdeadTr   )r   r	   r
   r   r   r   r   nparrayr   r   wvr   allcloser   )r   vector_deadr    modelvector_dead_calcs        r   test_load_modelzTestApi.test_load_model   s    7==&& 	(M#,'''h  
  
  
   w||L>@g
 
 =>> 8F+K1ABBCCCcl###"FTXYYY[ghhhcl#####r   c                    t           j                            t          j        dd          }t           j                            t          j                  rt          j        t          j                   |                     |t          j	        dd                     t          j        t          j                   t          j	        d          }|                     t          t          |                    d           d S )Nz#__testing_multipart-matrix-synopsisz&__testing_multipart-matrix-synopsis.gzTr   r   r   )r   r    datasets      r   test_multipart_loadzTestApi.test_multipart_load7   s    w||L?Ai
 
 7==&& 	(M#,'''sx0Ucg'h'h'hiiicl###(@AAT']]++Q/////r   c                    t          j        d          }|                     |d         d           |                     |d         d           t          j                    }|                     t          |                                          t          ddg                     |                     t          |d                              |                     t          |d                              t          j        d	          }|                     t          |                                          d
           |                     ddht          |          k               d S )Ntext8partsr   	file_nameztext8.gzmodelscorporaT)	name_only   )r   infor   sortedkeysr   r   set)r   dataname_only_datas      r   	test_infozTestApi.test_infoB   s&   x  g***k*J777xzz		,,fh	5J.K.KLLLDN++,,,DO,,---D111^002233Q7779-^1D1DDEEEEEr   N)__name__
__module____qualname__r   r!   r+   r.   r=    r   r   r   r   	   sd        
  $ $ $$ $ $2	0 	0 	0
F 
F 
F 
F 
Fr   r   __main__z)%(asctime)s : %(levelname)s : %(message)s)formatlevel)loggingunittestr   gensim.downloader
downloaderr   r   numpyr$   skipIfenvirongetTestCaser   r>   basicConfigINFOmainrA   r   r   <module>rQ      s      				            JNN'//36F ?F ?F ?F ?F ?Fh ?F ?F	 ?FD z GJRYR^____HMOOOOO r   