o
    ,wi                     @   s   d Z ddlmZ ddlmZ ddlZddlmZ ddl	Zddl
Z
dedee fdd	ZG d
d dejZedkr>e  dS dS )z"Tests for fiddle.extensions.seqio.    )List)absltestN)legacy_codegencodereturnc                 C   s   |    S )N)stripsplit)r    r	   ^/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/fiddle/_src/extensions/seqio_test.pytokens   s   r   c                   @   s   e Zd Zdd ZdS )	SeqioTestc                 C   sT   t jj  tjtjttjd}dt	
| }d}| t|t| d S )N)feature_converter
z
import fiddle as fdl
import seqio


def build_config():
  root = fdl.Config(seqio.Evaluator)

  root.feature_converter = fdl.Config(seqio.DecoderFeatureConverter)

  return root
    )fiddle
extensionsseqioenablefdlConfig	EvaluatorDecoderFeatureConverterjoinr   codegen_dot_syntaxlinesassertEqualr   )selfcfgr   expectedr	   r	   r
   test_codegen!   s   
zSeqioTest.test_codegenN)__name__
__module____qualname__r   r	   r	   r	   r
   r      s    r   __main__)__doc__typingr   absl.testingr   r   r   fiddle._src.codegenr   fiddle.extensions.seqior   strr   TestCaser   r   mainr	   r	   r	   r
   <module>   s   