o
    Xεi                     @   sB   d 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 GraphemesToPhonemes class    N)GraphemesToPhonemesc                   @   s    e Zd ZdZdd Zdd ZdS )GraphemesToPhonemesTestCasez(Test cases for GraphemesToPhonemes 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_g2p.pytest_encode_decode   s   z.GraphemesToPhonemesTestCase.test_encode_decodec                 C   sZ   d}t j|dddd}ddddd	ddd
dddd
ddddddd
dg}| || dS )zTest word featurestest   F)chars_forwardchars_backwardencodeg      ?tTe)biasgraphemebegin
grapheme+1r   )r   r   r   
grapheme-1)r   r   endr   N)r   word2featuresr   )r   wordactual_featuresexpected_featuresr	   r	   r
   test_features   s   z)GraphemesToPhonemesTestCase.test_featuresN)__name__
__module____qualname____doc__r   r   r	   r	   r	   r
   r      s    r   __main__)r!   unittest	gruut.g2pr   TestCaser   r   mainr	   r	   r	   r
   <module>   s   