o
    Ni                     @   sx   d Z ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ	 ddl
mZ G dd	 d	e	jjZed
kr:e  dS dS )z(Tests for OPUS translate dataset module.    )absolute_import)division)print_function)testingN)opusc                       s>   e Zd Ze fddZe fddZejZddiZ	  Z
S )OpusTestCustomConfigTestc                    s8   t t|   tjtjddddgd}|gtj_	d S )Nz0.1.0)deenTanzilEMEA)versionlanguage_pairsubsets)
superr   
setUpClassr   
OpusConfigtfdscoreVersionOpusBUILDER_CONFIGS)clsconfig	__class__ [/home/ubuntu/.local/lib/python3.10/site-packages/tensorflow_datasets/translate/opus_test.pyr      s   
z#OpusTestCustomConfigTest.setUpClassc                    s   t t|   tjj  d S )N)r   r   tearDownClassr   r   r   pop)r   r   r   r   r   *   s   z&OpusTestCustomConfigTest.tearDownClasstrain   )__name__
__module____qualname__classmethodr   r   r   r   DATASET_CLASSSPLITS__classcell__r   r   r   r   r      s    
r   __main__)__doc__
__future__r   r   r   tensorflow_datasetsr   tensorflow_datasets.public_api
public_apir   tensorflow_datasets.translater   DatasetBuilderTestCaser   r!   	test_mainr   r   r   r   <module>   s   