o
    ¶N i  ã                   @   s|   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 G dd„ dej	ƒZ
G d	d
„ d
ej	ƒZedkr<e ¡  dS dS )zTest for div2k dataset.é    )Úabsolute_import)Údivision)Úprint_function)Útesting)Údiv2kc                   @   ó0   e Zd ZejZdgZdddœZdddddœZdS )	ÚDiv2kTestBicubicXÚ
bicubic_x2é   ©ÚtrainÚ
validationÚ ÚDIV2K_train_LR_bicubic_X2ÚDIV2K_valid_LR_bicubic_X2©Útrain_hr_urlÚvalid_hr_urlÚtrain_lr_urlÚvalid_lr_urlN©	Ú__name__Ú
__module__Ú__qualname__r   ÚDiv2kÚDATASET_CLASSÚBUILDER_CONFIG_NAMES_TO_TESTÚSPLITSÚDL_EXTRACT_RESULT© r   r   úX/home/ubuntu/.local/lib/python3.10/site-packages/tensorflow_datasets/image/div2k_test.pyr      ó    þ
ür   c                   @   r   )	ÚDiv2kTestUnknownXÚ
unknown_x2r
   r   r   ÚDIV2K_train_LR_unknown_X2ÚDIV2K_valid_LR_unknown_X2r   Nr   r   r   r   r    r"   +   r!   r"   Ú__main__N)Ú__doc__Ú
__future__r   r   r   Útensorflow_datasetsr   Útensorflow_datasets.imager   ÚDatasetBuilderTestCaser   r"   r   Ú	test_mainr   r   r   r    Ú<module>   s   ÿ