diff --git a/test/test_prototype_builtin_datasets.py b/test/test_prototype_builtin_datasets.py index 8a929b6907c..ea4f27f72a2 100644 --- a/test/test_prototype_builtin_datasets.py +++ b/test/test_prototype_builtin_datasets.py @@ -175,6 +175,13 @@ def test_infinite_buffer_size(self, test_home, dataset_mock, config): # resolved assert dp.buffer_size == INFINITE_BUFFER_SIZE + @parametrize_dataset_mocks(DATASET_MOCKS) + def test_has_length(self, test_home, dataset_mock, config): + dataset_mock.prepare(test_home, config) + dataset = datasets.load(dataset_mock.name, **config) + + assert len(dataset) > 0 + # FIXME: DATASET_MOCKS["qmnist"] @parametrize_dataset_mocks({})