From 358fc6ddef2c824972b67adb0645ebda713232d7 Mon Sep 17 00:00:00 2001 From: Philip Meier Date: Mon, 7 Mar 2022 15:41:35 +0100 Subject: [PATCH] check that prototype datasets are serializable (#5557) --- test/test_prototype_builtin_datasets.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/test/test_prototype_builtin_datasets.py b/test/test_prototype_builtin_datasets.py index 2d18ac1a815..2855a490351 100644 --- a/test/test_prototype_builtin_datasets.py +++ b/test/test_prototype_builtin_datasets.py @@ -1,5 +1,6 @@ import functools import io +import pickle from pathlib import Path import pytest @@ -108,12 +109,12 @@ def test_transformable(self, test_home, dataset_mock, config): next(iter(dataset.map(transforms.Identity()))) @parametrize_dataset_mocks(DATASET_MOCKS) - def test_traversable(self, test_home, dataset_mock, config): + def test_serializable(self, test_home, dataset_mock, config): dataset_mock.prepare(test_home, config) dataset = datasets.load(dataset_mock.name, **config) - traverse(dataset) + pickle.dumps(dataset) @parametrize_dataset_mocks(DATASET_MOCKS) @pytest.mark.parametrize("annotation_dp_type", (Shuffler, ShardingFilter))