o
    ¶N iš  ã                   @   sj   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
ed	kr3e ¡  d
S d
S )zTests for Caltech data loading.é    )Úabsolute_import)Údivision)Úprint_function)Útesting)Úcaltechc                       s,   e Zd ZejZdddœZ‡ fdd„Z‡  ZS )ÚCaltech101Testé   )ÚtrainÚtestc                    s   t t| ƒ ¡  dt_d S )Né   )Úsuperr   ÚsetUpr   Ú_TRAIN_POINTS_PER_CLASS)Úself©Ú	__class__© úi/home/ubuntu/.local/lib/python3.10/site-packages/tensorflow_datasets/image_classification/caltech_test.pyr   #   s   
zCaltech101Test.setUp)	Ú__name__Ú
__module__Ú__qualname__r   Ú
Caltech101ÚDATASET_CLASSÚSPLITSr   Ú__classcell__r   r   r   r   r      s    þr   Ú__main__N)Ú__doc__Ú
__future__r   r   r   Útensorflow_datasetsr   Ú(tensorflow_datasets.image_classificationr   ÚDatasetBuilderTestCaser   r   Ú	test_mainr   r   r   r   Ú<module>   s   ÿ