o
    ¶N i†  ã                   @   sš   d Z ddlmZ ddlmZ ddlmZ ddlZddlmZ ddlm	Z	 ej
 ej
 ej
 e¡d ¡d	d
dd¡ZG dd„ dejƒZedkrKe ¡  dS dS )z!Tests for wiki40b dataset module.é    )Úabsolute_import)Údivision)Úprint_functionN)Útesting)Úwiki40bz/../r   Ú	test_dataÚfake_examplesr   c                       s8   e Zd Ze‡ fdd„ƒZejZdgZddddœZ	‡  Z
S )ÚWiki40bTestc                    s   t t| ƒ ¡  tt_d S )N)Úsuperr	   Ú
setUpClassÚ_EXAMPLE_DIRr   Ú_DATA_DIRECTORY)Úcls©Ú	__class__© úY/home/ubuntu/.local/lib/python3.10/site-packages/tensorflow_datasets/text/wiki40b_test.pyr   '   s   
zWiki40bTest.setUpClassÚené   é   é   )ÚtrainÚ
validationÚtest)Ú__name__Ú
__module__Ú__qualname__Úclassmethodr   r   ÚWiki40bÚDATASET_CLASSÚBUILDER_CONFIG_NAMES_TO_TESTÚSPLITSÚ__classcell__r   r   r   r   r	   %   s    ýr	   Ú__main__)Ú__doc__Ú
__future__r   r   r   ÚosÚtensorflow_datasetsr   Útensorflow_datasets.textr   ÚpathÚjoinÚnormpathÚdirnameÚ__file__r   ÚDatasetBuilderTestCaser	   r   Ú	test_mainr   r   r   r   Ú<module>   s$   û	ÿ