
    c                     
   d Z ddlZddlZddlZddlmZ ddlmZmZ ddl	m
Z
 ddlmZ 	 ddlmZ dZn# e$ r d	ZY nw xY w ej        ed	u d
           G d dej                              Zedk    r ej                     dS dS )z)
Automated tests for checking visdom API
    N)LdaModel)datapathcommon_dictionary)MmCorpus)CoherenceMetric)VisdomTFzVisdom not installedc                       e Zd Zd Zd ZdS )TestLdaCallbackc                     t          t          d                    | _        t          | j        ddd          | _        | j        g| _        t          t          dd| j                  | _        d	| _	        d
| _
        d S )Nztestcorpus.mmu_massvisdom	Coherence)corpus	coherenceloggertitle   
   )id2word
num_topicspasses	callbackszhttp://localhosti  )r   r   r   r   ch_umasscallbackr   r   modelhostport)selfs    =lib/python3.11/site-packages/gensim/test/test_lda_callback.pysetUpzTestLdaCallback.setUp   so    x8899't{hW_grsss&7Ab\`\ijjj
&				    c           	         t          j        ddddt          | j                  g          5 }t	          | j        | j                  }t          d          D ],}t          j        d           |	                                r n-|	                                sJ |
                                 | j                            | j                   |                                 d d d            d S # 1 swxY w Y   d S )Npythonz-mzvisdom.serverz-port)serverr      g      ?)
subprocessPopenstrr   r   r   rangetimesleepcheck_connectioncloser   updater   kill)r   procvizattempts       r   test_callback_update_graphz*TestLdaCallback.test_callback_update_graph&   s(   xTYXYY 
	]a		:::C 88  
3'')) E'')))))IIKKKJdk***IIKKK
	 
	 
	 
	 
	 
	 
	 
	 
	 
	 
	 
	 
	 
	 
	 
	 
	 
	s   B5C..C25C2N)__name__
__module____qualname__r    r3    r!   r   r
   r
      s2              r!   r
   __main__)__doc__unittestr&   r*   gensim.modelsr   gensim.test.utilsr   r   gensim.corporar   gensim.models.callbacksr   r   r   VISDOM_INSTALLEDImportErrorskipIfTestCaser
   r4   mainr7   r!   r   <module>rD      s>          " " " " " " 9 9 9 9 9 9 9 9 # # # # # # 3 3 3 3 3 3    !U*,BCC    h'   DC2 z HMOOOOO s   3 ==