From b7658187bfb3c0c406e7ce9336abcab24d575f9e Mon Sep 17 00:00:00 2001 From: Massimiliano Lincetto Date: Fri, 21 Oct 2022 17:11:58 +0200 Subject: [PATCH] add test for dataset index (WIP) --- tests/test_dataset_index.py | 26 ++++++++++++++++++++++++++ 1 file changed, 26 insertions(+) create mode 100644 tests/test_dataset_index.py diff --git a/tests/test_dataset_index.py b/tests/test_dataset_index.py new file mode 100644 index 00000000..da6665e8 --- /dev/null +++ b/tests/test_dataset_index.py @@ -0,0 +1,26 @@ +import unittest +import logging + +from flarestack.data.dataset_index import dataset_index + +logging.basicConfig(level=logging.INFO) +logger = logging.getLogger(__name__) + + +class TestDatasetIndex(unittest.TestCase): + def setUp(self) -> None: + pass + + def test_index_content(self) -> None: + n = len(dataset_index.get_dataset_list()) + logger.info(f"There are {n} datasets in the index.") + self.assertGreater(n, 0) + + def test_index_consistency(self) -> None: + for key in dataset_index.get_dataset_list(): + dataset = dataset_index.get_dataset(key) + self.assertEqual(key, dataset.name) + + +if __name__ == "__main__": + unittest.main()