o
    Xεi}
                     @   sJ   d Z ddlZddlZddlmZ G dd dejZedkr#e  dS dS )z"Tests for PartOfSpeechTagger class    N)PartOfSpeechTaggerc                   @   s    e Zd ZdZdd Zdd ZdS )PartOfSpeechTaggerTestCasez'Test cases for PartOfSpeechTagger classc                 C   s    d}|  tt|| dS )z4Test encode/decode functions for pycrfsuite featuresu5   ði ıntəˈnæʃənəl fəˈnɛtık əsoʊsiˈeıʃnN)assertEqualr   decode_stringencode_string)selfs r	   B/home/ubuntu/.local/lib/python3.10/site-packages/tests/test_pos.pytest_encode_decode   s   z-PartOfSpeechTaggerTestCase.test_encode_decodec              
   C   s
  d  }ddddddddddd	dd
dddddddddddd}dd }t|}|d ||d d |d ||d d |d ||d d |d ||d d d|d d< d|d d< |d |d |d g}tj|dddddd}| || dS )zTest sentence featuresz1 test .g      ?1   FT)biasword	len(word)word.ispunctuationword.isdigit()word[:2]	word[-2:]test   stte)r   r   r   r   r   r   r   .)r   r   r   r   r   r   r   )r   r   r   c                    s    fdd|   D S )Nc                    s   i | ]\}}  | |qS r	   r	   ).0kvprefixr	   r
   
<dictcomp>5   s    zPPartOfSpeechTaggerTestCase.test_features.<locals>.add_prefix.<locals>.<dictcomp>)items)dr   r	   r   r
   
add_prefix4   s   z<PartOfSpeechTaggerTestCase.test_features.<locals>.add_prefixz+1:z-1:BOSEOS   )words_forwardwords_backwardchars_front
chars_backencodeN)splitcopydeepcopyupdater   sent2featuresr   )r   sentenceword_featuresr"   context_featuresexpected_featuresactual_featuresr	   r	   r
   test_features   s^   


	z(PartOfSpeechTaggerTestCase.test_featuresN)__name__
__module____qualname____doc__r   r5   r	   r	   r	   r
   r   	   s    r   __main__)	r9   r,   unittest	gruut.posr   TestCaser   r6   mainr	   r	   r	   r
   <module>   s   O