def test_registered(self):
name = "empty_dataset_builder"
self.assertEqual(name, EmptyDatasetBuilder.name)
self.assertIs(EmptyDatasetBuilder, registered.builder(name))
self.assertIn(name, registered.list_builders())
self.assertNotIn("unregistered_builder", registered.list_builders())
nonexistent = "nonexistent_foobar_dataset"
with self.assertRaisesWithPredicateMatch(ValueError, "not found"):
registered.builder(nonexistent)
# Prints registered datasets
with self.assertRaisesWithPredicateMatch(ValueError, name):
registered.builder(nonexistent)
def test_abstract(self):
name = "unregistered_builder"
self.assertEqual(name, UnregisteredBuilder.name)
self.assertNotIn(name, registered.list_builders())
with self.assertRaisesWithPredicateMatch(ValueError, "an abstract class"):
registered.builder(name)
def test_skip_regitration(self):
"""Test `skip_registration()`."""
with registered.skip_registration():
@six.add_metaclass(registered.RegisteredDataset)
class SkipRegisteredDataset(object):
pass
name = "skip_registered_dataset"
self.assertEqual(name, SkipRegisteredDataset.name)
self.assertNotIn(name, registered.list_builders())
def test_registered(self):
self.assertIn("image_label_folder", registered.list_builders(),
"Dataset was not registered.")
def test_registered(self):
self.assertIn(self.builder.name, registered.list_builders(),
"Dataset was not registered.")
