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"Tests for starcraft video dataset.é    )Úabsolute_import)Údivision)Úprint_function)Útesting)Ú	starcraftc                   @   s2   e Zd ZejZdgZdddddœZdddd	œZd
S )ÚStarcraftVideoDatasetTestÚbrawl_64zvalid.tfrecordsztest.tfrecordsztrain_0.tfrecordsztrain_1.tfrecords©ÚvalidÚtestÚtrain_0Útrain_1é   é   ©Útrainr   Ú
validationN)	Ú__name__Ú
__module__Ú__qualname__r   ÚStarcraftVideoÚDATASET_CLASSÚBUILDER_CONFIG_NAMES_TO_TESTÚDL_EXTRACT_RESULTÚSPLITS© r   r   ú\/home/ubuntu/.local/lib/python3.10/site-packages/tensorflow_datasets/video/starcraft_test.pyr      s    ü
ýr   c                   @   s6   e Zd ZdZejZdgZdddddœZdd	d	d
œZ	dS )ÚStarcraftVideoDataset128Testz5Separate test to cover the 128x128 resolution videos.Ú	brawl_128z128_valid.tfrecordsz128_test.tfrecordsz128_train_0.tfrecordsz128_train_1.tfrecordsr	   r   r   r   N)
r   r   r   Ú__doc__r   r   r   r   r   r   r   r   r   r   r   -   s    ü
ýr   Ú__main__N)r   Ú
__future__r   r   r   Útensorflow_datasetsr   Útensorflow_datasets.videor   ÚDatasetBuilderTestCaser   r   r   Ú	test_mainr   r   r   r   Ú<module>   s   ÿ