def test_tfds_source_splits(self):
    default_splits_src = dataset_providers.TfdsDataSource('fake:0.0.0')
    self.assertSameElements(['train', 'validation'], default_splits_src.splits)
    validation_split_src = dataset_providers.TfdsDataSource('fake:0.0.0', splits=['validation'])
    self.assertSameElements(['validation'], validation_split_src.splits)
    sliced_split_src = dataset_providers.TfdsDataSource('fake:0.0.0', splits={'validation': 'train[0:1%]'})
    self.assertSameElements(['validation'], sliced_split_src.splits)
def test_tfds_source_splits(self):
    default_splits_src = dataset_providers.TfdsDataSource('fake:0.0.0')
    self.assertSameElements(['train', 'validation'], default_splits_src.splits)
    validation_split_src = dataset_providers.TfdsDataSource('fake:0.0.0', splits=['validation'])
    self.assertSameElements(['validation'], validation_split_src.splits)
    sliced_split_src = dataset_providers.TfdsDataSource('fake:0.0.0', splits={'validation': 'train[0:1%]'})
    self.assertSameElements(['validation'], sliced_split_src.splits)