o
    ¶N iŸ  ã                   @   sz   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 d dl	m
Z
 G dd	„ d	ejƒZed
kr;e ¡  dS dS )é    )Úabsolute_import)Údivision)Úprint_functionN)Útesting)Údataset_utils)Úucf101c                       s>   e Zd ZejZdddœZdddœZddgZ‡ fd	d
„Z	‡  Z
S )Ú
Ucf101Testé   é   )ÚtrainÚtestÚvideosÚsplits)r   r   Úucf101_1_256Úucf101_2c                    sž   t t| ƒ |¡ i }|jjd }| ¡ }tjD ]$}t t	¡||< t
 || ¡D ]}| |d ¡}|| |  d7  < q'q|  t|ƒdddœdddœdœ¡ dS )z,Check the label distribution for each split.Úlabelé   )ÚArcheryÚ	Nunchucksr
   )r   r   N)Úsuperr   Ú_assertAsDatasetÚinfoÚfeaturesÚ
as_datasetÚSPLITSÚcollectionsÚdefaultdictÚintr   Úas_numpyÚint2strÚassertEqualÚdict)ÚselfÚbuilderÚlabel_frequnciesÚlabel_featureÚdatasetÚ
split_namer   Ú
label_name©Ú	__class__© úY/home/ubuntu/.local/lib/python3.10/site-packages/tensorflow_datasets/video/ucf101_test.pyr   +   s   
þ
ÿÿzUcf101Test._assertAsDataset)Ú__name__Ú
__module__Ú__qualname__r   ÚUcf101ÚDATASET_CLASSr   ÚDL_EXTRACT_RESULTÚBUILDER_CONFIG_NAMES_TO_TESTr   Ú__classcell__r+   r+   r)   r,   r      s    þþr   Ú__main__)Ú
__future__r   r   r   r   Útensorflow_datasetsr   Útensorflow_datasets.corer   Útensorflow_datasets.videor   ÚDatasetBuilderTestCaser   r-   Ú	test_mainr+   r+   r+   r,   Ú<module>   s   ÿ