o
    Ni                  	   @   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 de_	de_
G d	d
 d
ejZG dd deZG dd deZejjejddddddg G dd dejZedkrbe  dS dS )zTests for mnist dataset module.    )absolute_import)division)print_function)testing)mnist
      c                   @   s*   e Zd ZejZdddZdddddZd	S )
	MNISTTestr   r   traintestztrain-imageztrain-labelz
test-imagez
test-label)
train_datatrain_labels	test_datatest_labelsN)__name__
__module____qualname__r   MNISTDATASET_CLASSSPLITSDL_EXTRACT_RESULT r   r   g/home/ubuntu/.local/lib/python3.10/site-packages/tensorflow_datasets/image_classification/mnist_test.pyr	   !   s    
r	   c                   @      e Zd ZejZdS )FashionMNISTTestN)r   r   r   r   FashionMNISTr   r   r   r   r   r   /       
r   c                   @   r   )
KMNISTTestN)r   r   r   r   KMNISTr   r   r   r   r   r   3   r   r   r      zEMNIST test data config.)nameclass_numbertrain_examplestest_examplesdescriptionc                   @   s"   e Zd ZejZdddZdgZdS )
EMNISTTestr   r   r
   r   N)r   r   r   r   EMNISTr   r   BUILDER_CONFIG_NAMES_TO_TESTr   r   r   r   r&   B   s    
r&   __main__N)__doc__
__future__r   r   r   tensorflow_datasetsr   (tensorflow_datasets.image_classificationr   _TRAIN_EXAMPLES_TEST_EXAMPLESDatasetBuilderTestCaser	   r   r   r'   BUILDER_CONFIGSextendEMNISTConfigr&   r   	test_mainr   r   r   r   <module>   s0   	