View source on GitHub

Testing utilities.

Classes

class DatasetBuilderTestCase: Inherit this class to test your DatasetBuilder class.

class DummyBeamDataset: Minimal beam DatasetBuilder.

class DummyDataset: Minimal DatasetBuilder.

class DummyDatasetCollection: Minimal Dataset Collection builder.

class DummyDatasetSharedGenerator: Test DatasetBuilder.

class DummyMnist: Test DatasetBuilder.

class DummyParser: To mock example_parser.ExampleParser.

class DummySerializer: To mock example_serializer.ExampleSerializer.

class FeatureExpectationItem: Test item of a FeatureExpectation.

class FeatureExpectationsTestCase: Tests FeatureExpectations with full encode-decode.

class MockFs: This util wraps mock for the tf.io.gfile / epath.Path API.

class MockPolicy: Strategy to use with tfds.testing.mock_data to mock the dataset.

class PickableDataSourceMock: Makes MagicMock pickable in order to work with multiprocessing in Grain.

class RaggedConstant: Container of tf.ragged.constant values.

class SubTestCase: Adds subTest() context manager to the TestCase if supported.

class TestCase: Base TestCase to be used for all tests.

Functions

assert_features_equal(...): Asserts that the 2 nested FeatureConnector structure match.

fake_examples_dir(...)

make_tmp_dir(...): Make a temporary directory.

mock_data(...): Mock tfds to generate random data.

mock_kaggle_api(...): Mock out the kaggle CLI.

rm_tmp_dir(...): Rm temporary directory.

run_in_graph_and_eager_modes(...): Execute the decorated test in both graph mode and eager mode.

test_main(...): Entrypoint for tests.

tmp_dir(...): Context manager for a temporary directory.