From c574cc2012cde73033f134d65f73b980b828fc2e Mon Sep 17 00:00:00 2001 From: Philip Meier Date: Tue, 5 Apr 2022 09:26:27 +0200 Subject: [PATCH] add test to enforce __len__ is working on prototype datasets --- test/test_prototype_builtin_datasets.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/test/test_prototype_builtin_datasets.py b/test/test_prototype_builtin_datasets.py index f8dc3a0542b..f5599623397 100644 --- a/test/test_prototype_builtin_datasets.py +++ b/test/test_prototype_builtin_datasets.py @@ -154,6 +154,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({})