
    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  G d dej                  Z	e
dk    r0ej                            ej                    ej                     dS dS )zI
Automated tests for segmentation algorithms in the segmentation module.
    N)segmentation)arrayc                   &    e Zd Zd Zd Zd Zd ZdS )TestSegmentationc                 t    t          g d          t          g d          t          g d          g| _        d S )N	         r	   
            r   )r   topics)selfs    =lib/python3.11/site-packages/gensim/test/test_segmentation.pysetUpzTestSegmentation.setUp   s<    )))***)))
    c                     t          j        | j                  }g dg dg dg}|                     t	          j        ||                     dS )zTest s_one_pre segmentation.)r
   r	   r   r	   r   r
   )r   r	   r   r	   r   r   )r   r   r   r   r   r   N)r   	s_one_prer   
assertTruenpallcloser   actualexpecteds      r   test_s_one_prezTestSegmentation.test_s_one_pre   s\    '44$$$&&&$$$

 	FH5566666r   c                     t          j        | j                  }g dg dg dg}|                     t	          j        ||                     dS )zTest s_one_one segmentation.))r	   r
   )r	   r   r   )r
   r   r   r   ))r	   r   )r	   r   r   )r   r   r   r   ))r   r   )r   r   r   )r   r   r   r    N)r   	s_one_oner   r"   r#   r$   r%   s      r   test_s_one_onezTestSegmentation.test_s_one_one'   s\    '44<<<@@@<<<

 	FH5566666r   c           	         t          j        | j                  }dt          g d          fdt          g d          fdt          g d          fgdt          g d          fdt          g d          fdt          g d          fgdt          g d	          fd
t          g d	          fdt          g d	          fgg}t	          t          |                    D ]}t	          t          ||                             D ]}|                     ||         |         d         ||         |         d                    |                     t          j	        ||         |         d         ||         |         d                              dS )zTest s_one_set segmentation.r	   r   r
   r   r   r   r   r   r   r   r      N)
r   	s_one_setr   r   rangelenassertEqualr"   r#   r$   )r   r&   r'   s_ijs        r   test_s_one_setzTestSegmentation.test_s_one_set1   s   '44yyy!!"Qiii(8(8$9AuYYY?O?O;PQzzz""#b%


*;*;%<q%


BSBS>TUyyy!!"Qiii(8(8$9AuYYY?O?O;PQ

 V%% 	U 	UC3vc{++,, U U  Q!2HSM!4DQ4GHHHF3KN1,=x}Q?OPQ?R S STTTTU	U 	Ur   N)__name__
__module____qualname__r   r(   r+   r4    r   r   r   r      sU        
 
 
7 7 77 7 7U U U U Ur   r   __main__)__doc__loggingunittestnumpyr#   gensim.topic_coherencer   r   TestCaser   r5   rootsetLevelWARNINGmainr8   r   r   <module>rD      s    
       / / / / / /      'U 'U 'U 'U 'Ux( 'U 'U 'UT z L'/***HMOOOOO r   