def test_tfds_splits(self):
    self.assertSameElements(['train', 'validation'], dataset_providers.TfdsDataSource(tfds_name='fake:0.0.0').splits)
    self.assertSameElements(['validation'], dataset_providers.TfdsDataSource(tfds_name='fake:0.0.0', splits=['validation']).splits)
    self.assertSameElements(['validation'], dataset_providers.TfdsDataSource(tfds_name='fake:0.0.0', splits={'validation': 'train'}).splits)
def test_tfds_splits(self):
    self.assertSameElements(['train', 'validation'], dataset_providers.TfdsDataSource(tfds_name='fake:0.0.0').splits)
    self.assertSameElements(['validation'], dataset_providers.TfdsDataSource(tfds_name='fake:0.0.0', splits=['validation']).splits)
    self.assertSameElements(['validation'], dataset_providers.TfdsDataSource(tfds_name='fake:0.0.0', splits={'validation': 'train'}).splits)